Update CodeFileEditAtomicFlow.py

CodeFileEditAtomicFlow.py

@@ -4,7 +4,7 @@ from typing import Dict, Any
 from flows.base_flows.atomic import AtomicFlow
 import os
 
-class CodeFileEditorAtomicFlow(AtomicFlow):
+class CodeFileEditAtomicFlow(AtomicFlow):
     def _write_code_to_temp_file(self, code_str, language_of_code, code_lib_location):
         directory = os.path.dirname(code_lib_location)
         if language_of_code.lower() == 'python':
@@ -26,10 +26,11 @@ class CodeFileEditorAtomicFlow(AtomicFlow):
             )
             with open(temp_file_location, "w") as file:
                 file.write(content)
-            return True, f"Code written to {temp_file_location}", temp_file_location
+            file_written_timestamp = os.path.getmtime(temp_file_location)
+            return True, f"Code written to {temp_file_location}", temp_file_location, file_written_timestamp
 
         except Exception as e:
-            return False, str(e), temp_file_location
+            return False, str(e), temp_file_location, 0
 
     def _check_input(self, input_data: Dict[str, Any]):
         assert "code" in input_data, "code is not passed to CodeFileEditAtomicFlow"
@@ -48,7 +49,7 @@ class CodeFileEditorAtomicFlow(AtomicFlow):
         code_str = input_data['code']
         language_of_code = input_data["language_of_code"]
         code_lib_location = input_data["memory_files"]["code_library"]
-        result, code_editor_output, temp_file_location = self._write_code_to_temp_file(code_str, language_of_code, code_lib_location)
+        result, code_editor_output, temp_file_location, file_written_timestamp = self._write_code_to_temp_file(code_str, language_of_code, code_lib_location)
         if result:
             try:
                 subprocess.run(["code", temp_file_location], timeout=10)
@@ -57,4 +58,5 @@ class CodeFileEditorAtomicFlow(AtomicFlow):
         response = {}
         response["code_editor_output"] = code_editor_output
         response["temp_code_file_location"] = temp_file_location
+        response["temp_code_file_written_timestamp"] = file_written_timestamp
         return response