Klass777 commited on
Commit
a06afb1
·
verified ·
1 Parent(s): af5a01e

Change from ToolCalling agent to Code agent

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -140,7 +140,7 @@ class BasicAgent:
140
  def __init__(self):
141
  model = OpenAIServerModel(api_key=os.environ.get("OPENAI_API_KEY"), model_id="gpt-4o")
142
 
143
- self.code_agent = ToolCallingAgent(
144
  tools=[PythonInterpreterTool(), DuckDuckGoSearchTool(), VisitWebpageTool(), transcribe_audio_file,
145
  get_youtube_transcript,
146
  FinalAnswerTool()],
@@ -148,7 +148,7 @@ class BasicAgent:
148
  max_steps=20,
149
  name="hf_agent_course_final_assignment_solver",
150
  prompt_templates=yaml.safe_load(
151
- importlib.resources.files("prompts").joinpath("agent.yaml").read_text()
152
  )
153
 
154
  )
 
140
  def __init__(self):
141
  model = OpenAIServerModel(api_key=os.environ.get("OPENAI_API_KEY"), model_id="gpt-4o")
142
 
143
+ self.code_agent = CodeAgent(
144
  tools=[PythonInterpreterTool(), DuckDuckGoSearchTool(), VisitWebpageTool(), transcribe_audio_file,
145
  get_youtube_transcript,
146
  FinalAnswerTool()],
 
148
  max_steps=20,
149
  name="hf_agent_course_final_assignment_solver",
150
  prompt_templates=yaml.safe_load(
151
+ importlib.resources.files("prompts").joinpath("code_agent.yaml").read_text()
152
  )
153
 
154
  )