Update Controller_CoderFlow.yaml

Controller_CoderFlow.yaml

@@ -15,6 +15,7 @@ input_interface_non_initialized: # initial input keys
   - "code_library"
 
 input_interface_initialized:
+  - "goal"
   - "result"
   - "memory_files"
   - "plan"
@@ -64,6 +65,9 @@ system_message_prompt_template:
   template: |2-
           Your department is in charge of achieving goals with writing and executing code. You are the leader of this department.
     
+          Here is the goal you need to achieve:
+            {{goal}}
+          
           You have a code library located at {{code_library_location}}, which contains functions that are already written and saved by your subordinates. 
           Here are the functions in the library:
             {{code_library}}
@@ -137,7 +141,7 @@ system_message_prompt_template:
               }
           Ensure your responses can be parsed by Python json.loads
 
-  input_variables: ["commands", "plan_file_location", "plan", "logs", "code_library_location", "code_library"]
+  input_variables: ["commands", "plan_file_location", "plan", "logs", "code_library_location", "code_library", "goal"]
   template_format: jinja2
 
 human_message_prompt_template:
@@ -160,5 +164,4 @@ init_human_message_prompt_template:
 
 
 previous_messages:
-  first_k: 2
-  last_k: 1
+  last_k: 3