Update ParseFeedbackAtomicFlow.py

ParseFeedbackAtomicFlow.py

@@ -1,5 +1,7 @@
 from typing import Dict, Any
 from flows.base_flows.atomic import AtomicFlow
+import time
+import os
 
 
 class ParseFeedbackAtomicFlow(AtomicFlow):
@@ -56,6 +58,13 @@ class ParseFeedbackAtomicFlow(AtomicFlow):
         elif code_file_exists and plan_file_exists:
             raise AssertionError("Both code file and plan file are passed to ParseFeedbackAtomic, which one to parse?")
 
+    def _wait_for_file_update(self, file_location, last_check_time, check_interval=1):
+        while True:
+            time.sleep(check_interval)
+            current_time = os.path.getmtime(file_location)
+            if current_time != last_check_time:
+                break
+
     def run(
             self,
             input_data: Dict[str, Any]
@@ -63,13 +72,17 @@ class ParseFeedbackAtomicFlow(AtomicFlow):
         response = {}
         self._check_input(input_data)
         if "temp_code_file_location" in input_data:
-            temp_code_file_location = input_data['plan_file_location']
+            temp_code_file_location = input_data['temp_code_file_location']
+            last_check_time = input_data["temp_code_file_written_timestamp"]
+            self._wait_for_file_update(temp_code_file_location, last_check_time)
             code_content = self._read_content(temp_code_file_location, "code")
             user_thoughts = self._parse_user_thoughts(temp_code_file_location)
             response = {"code": code_content, "human_feedback": user_thoughts}
 
         elif "plan_file_location" in input_data:
             plan_file_location = input_data['plan_file_location']
+            last_check_time = input_data["plan_file_written_timestamp"]
+            self._wait_for_file_update(plan_file_location, last_check_time)
             plan_content = self._read_content(plan_file_location, "plan")
             user_thoughts = self._parse_user_thoughts(plan_file_location)
             response = {"plan": plan_content, "human_feedback": user_thoughts}