name: "CodeGeneratorAtomicFlow"
description: "Writes code given instruction"
enable_cache: True

input_interface_non_initialized: # initial input keys
  - "goal"

input_interface_initialized: # input_keys
  - "goal"

#######################################################
# Output keys
#######################################################

output_interface:
  - 'language_of_code'
  - 'code'

#######################################################
system_message_prompt_template:
  _target_: flows.prompt_template.JinjaPrompt
  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.
    
    Your task is to write functions for the given goal, you do not need to worry abot the execution part.
     
    You can access the Internet, using APIs could be useful.
    
    **It's important that all code you write must be functions to achieve certain goals.**
    
    **It's important that you make imports within functions.**
    
    **Make sure the function names, parameter names are self-explanatory.**
    
    You are capable of **any** task.
    

    Performance Evaluation:
    1. The priporty language of choice 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.
    
    **It's important that you should only respond in JSON format as described below:**
    Response Format:
    {
    "language_of_code": "language of the code",
    "code": "String of the code corresponding to the goal",
    }
    Ensure your responses can be parsed by Python json.loads

human_message_prompt_template:
  _target_: flows.prompt_template.JinjaPrompt
  template: |2-
    Here is the response to your last action:
    {{goal}}
  input_variables:
    - "goal"

init_human_message_prompt_template:
  _target_: flows.prompt_template.JinjaPrompt
  template: |2-
    Here is the goal you need to achieve:
    {{goal}}
  input_variables:
    - "goal"