Update CodeGeneratorAtomicFlow.yaml

CodeGeneratorAtomicFlow.yaml

@@ -4,9 +4,13 @@ enable_cache: True
 
 input_interface_non_initialized: # initial input keys
   - "goal"
+  - "code_library"
+  - "memory_files"
 
 input_interface_initialized: # input_keys
   - "goal"
+  - "code_library"
+  - "memory_files"
 
 #######################################################
 # Output keys
@@ -22,19 +26,24 @@ system_message_prompt_template:
   template: |2-
     You are a world class programmer that can complete any goal with code. All code you write should be functions.
     
-    Your function will then be imported and called by an executor to finish the goal.
+    You have a library of executable functions located at {{code_library_file_location}} for your disposal, the code you write will also be added to this file.
     
-    Your task is to write functions and their respective docstrings for the given goal, you do not need to worry abot the execution part.
-     
-    You can access the Internet, using APIs of online services could be useful.
+    Your tasks are:
+      1. Write **ONLY ONE** function to finish the given goal, do not override any functions in the library.
+      2. Also write docstrings to the new function.
     
-    **It's important that all code you write must be functions to achieve certain goals.**
+    Notice that:
+      1. Write functions only.
+      2. Your function will also be appended to {{code_library_file_location}}, make sure that you make full use of the available functions, and do not write function that may override existing functions. 
+      3. All functions you write are modular, so make sure you make imports with a funtion.
+      4. You must write docstrings for every function you write.
     
-    **It's important that you make imports within functions.**
+    Here are the contents in the code library located ta {{code_library_file_location}}, make sure you make full use of them:
+    {{code_library}}
     
-    **Make sure the function names, parameter names are self-explanatory.**
+    Your function will then be imported and called by an executor to finish the goal, you do not need to worry about the execution part.
     
-    **When you write a function, also write its docstring right after the function name, so that the user can understand the function.**
+    You can access the Internet, using APIs of online services could be useful.
     
     An example of a function you write:
     ### begin of example ###
@@ -50,10 +59,8 @@ system_message_prompt_template:
     Performance Evaluation:
     1. The priority language is Python, try to code with Python to solve the task.
     2. All code you write must be functions.
-    3. Function names must be self-explanatory, it's good practice to specify the return type.
-    4. Parameter names must be self-explanatory, it's good practice to specify parameter types.
-    5. All imports must be done within the function body.
-    6. All functions must have docstrings.
+    3. All imports must be done within the function body.
+    4. All functions must have docstrings.
     
     **It's important that you should only respond in JSON format as described below:**
     Response Format:
@@ -67,6 +74,8 @@ system_message_prompt_template:
     
     **It's important that the code you generate can be written by Python write, and is human-readable. The written file must also be indented and formatted, so that it is human-readable.**
 
+  input_variables: ["code_library_file_location", "code_library"]
+
 human_message_prompt_template:
   _target_: flows.prompt_template.JinjaPrompt
   template: |2-