Update ControllerAtomicFlow.py

ControllerAtomicFlow.py

@@ -114,10 +114,11 @@ class ControllerAtomicFlow(OpenAIChatAtomicFlow):
         plan_to_append = self._get_plan()
         function_signatures_to_append = self._get_library_function_signatures()
         self.system_message_prompt_template.template = \
-            self.original_system_template + "\n" + f"Here are the available functions at {self.code_file_location}\n" \
+            self.original_system_template + "\n\n" + f"Here are the available functions at {self.code_file_location}\n" \
             + function_signatures_to_append + "\n" \
             + f"Here is the step-by-step plan to achieve the goal:\n" \
-            + plan_to_append + "\n"
+            + plan_to_append + "\n\n" + f"Make sure the plan your write is at {self.plan_file_location}\n" \
+            + f"Make sure the code you call the code writer to write is at {self.code_file_location}"
 
         api_output = super().run(input_data)["api_output"].strip()