First_agent_template

Sleeping

Ebraheem commited on Mar 16, 2025

Commit

2b58583

verified ·

1 Parent(s): 93adeb6

fix prompt

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
-print(prompt_templates)
 agent = CodeAgent(
     model=model,
     tools=[final_answer, get_current_time_in_timezone, web_search,
@@ -57,7 +57,7 @@ agent = CodeAgent(
     planning_interval=None,
     name=None,
     description=None,
-    prompt_templates=prompt_templates + vwb.helium_instructions,
 )
 agent.python_executor("from helium import *")

 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
+prompt_templates["system_prompt"] += "\n"  + vwb.helium_instructions
 agent = CodeAgent(
     model=model,
     tools=[final_answer, get_current_time_in_timezone, web_search,
     planning_interval=None,
     name=None,
     description=None,
+    prompt_templates=prompt_templates,
 )
 agent.python_executor("from helium import *")