Update ControllerAtomicFlow.py

ControllerAtomicFlow.py

@@ -74,6 +74,12 @@ class ControllerAtomicFlow(OpenAIChatAtomicFlow):
         ]
         kwargs.update({"commands": commands})
 
+        # ~~~ Set up file location for plan and code ~~~
+        plan_file_location = flow_config["plan_file_location"]
+        code_file_location = flow_config["code_file_location"]
+        kwargs.update({"plan_file_location": plan_file_location})
+        kwargs.update({"code_file_location": code_file_location})
+
         # ~~~ Instantiate flow ~~~
         return cls(**kwargs)