Upload 3 files

ParseFeedbackAtomicFlow.py

@@ -52,7 +52,7 @@ class ParseFeedbackAtomicFlow(AtomicFlow):
 
     def _check_input(self, input_data: Dict[str, Any]):
         code_file_exists = "temp_code_file_location" in input_data
-        plan_file_exists = "plan_file_location" in input_data
+        plan_file_exists = "temp_plan_file_location" in input_data
         if code_file_exists == False and plan_file_exists == False:
             raise AssertionError("Neither code file nor plan file is passed to ParseFeedbackAtomicFlow")
         elif code_file_exists and plan_file_exists:
@@ -79,7 +79,7 @@ class ParseFeedbackAtomicFlow(AtomicFlow):
             user_thoughts = self._parse_user_thoughts(temp_code_file_location)
             response = {"code": code_content, "feedback": user_thoughts}
 
-        elif "plan_file_location" in input_data:
+        elif "temp_plan_file_location" in input_data:
             plan_file_location = input_data["temp_plan_file_location"]
             last_check_time = input_data["temp_plan_file_written_timestamp"]
             self._wait_for_file_update(plan_file_location, last_check_time)