Update PlanWriterCtrlFlow.py

PlanWriterCtrlFlow.py

@@ -73,6 +73,13 @@ class PlanWriterCtrlFlow(ChatAtomicFlow):
 
     def run(self, input_data: Dict[str, Any]) -> Dict[str, Any]:
         self._update_prompts_and_input(input_data)
+        
+        # ~~~when conversation is initialized, append the updated system prompts to the chat history ~~~
+        if self._is_conversation_initialized():
+            updated_system_message_content = self._get_message(self.system_message_prompt_template, input_data)
+            self._state_update_add_chat_message(content=updated_system_message_content,
+                                                role=self.flow_config["system_name"])
+        
         api_output = super().run(input_data)["api_output"].strip()
         try:
             response = json.loads(api_output)