JerryGao commited on
Commit
cd9cc73
·
verified ·
1 Parent(s): ce1f94f

Update app.py

Browse files

make a complete agent implementation with image generation tool and web search tool and fact checking task guideline in prompt

Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -46,18 +46,18 @@ custom_role_conversions=None,
46
  # Import tool from Hub
47
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
48
 
49
- with open("prompts.yaml", 'r') as stream:
50
  prompt_templates = yaml.safe_load(stream)
51
 
52
  agent = CodeAgent(
53
  model=model,
54
- tools=[final_answer], ## add your tools here (don't remove final answer)
55
  max_steps=6,
56
  verbosity_level=1,
57
  grammar=None,
58
  planning_interval=None,
59
- name=None,
60
- description=None,
61
  prompt_templates=prompt_templates
62
  )
63
 
 
46
  # Import tool from Hub
47
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
48
 
49
+ with open("my_code_agent.yaml", 'r') as stream:
50
  prompt_templates = yaml.safe_load(stream)
51
 
52
  agent = CodeAgent(
53
  model=model,
54
+ tools=[image_generation_tool, DuckDuckGoSearchTool(), final_answer], ## add your tools here (don't remove final answer)
55
  max_steps=6,
56
  verbosity_level=1,
57
  grammar=None,
58
  planning_interval=None,
59
+ name="fact_checker_agent_with_image_generation_capability",
60
+ description="interpret user's intent and generate image or do web search based on user prompt or behavior as fact checker",
61
  prompt_templates=prompt_templates
62
  )
63