jc7k commited on
Commit
67cd948
·
1 Parent(s): 8083d45

Removed reference to final_answer

Browse files
Files changed (1) hide show
  1. app.py +2 -9
app.py CHANGED
@@ -5,8 +5,6 @@ import inspect
5
  import yaml
6
  import pandas as pd
7
  from smolagents import OpenAIServerModel, ToolCallingAgent, CodeAgent, HfApiModel, DuckDuckGoSearchTool
8
- from tools.final_answer import FinalAnswerTool
9
-
10
 
11
  # (Keep Constants as is)
12
  # --- Constants ---
@@ -25,20 +23,15 @@ class BasicAgent:
25
  api_key=os.environ["OPENAI_API_KEY"],
26
  )
27
 
28
- final_answer = FinalAnswerTool()
29
- # with open("prompts.yaml", "r") as stream:
30
- # prompt_templates = yaml.safe_load(stream)
31
-
32
  # agent = ToolCallingAgent(
33
  agent = CodeAgent(
34
  name="Jeff's GAIA L1 Agent",
35
  verbosity_level=1,
36
- tools=[DuckDuckGoSearchTool(), final_answer],
37
  add_base_tools=True,
38
  max_steps=5,
39
  model=model,
40
- # prompt_templates=prompt_templates
41
- )
42
 
43
  self.agent = agent
44
  print(f"Agent initialized with model ID: {os.environ['MODEL_ID']}")
 
5
  import yaml
6
  import pandas as pd
7
  from smolagents import OpenAIServerModel, ToolCallingAgent, CodeAgent, HfApiModel, DuckDuckGoSearchTool
 
 
8
 
9
  # (Keep Constants as is)
10
  # --- Constants ---
 
23
  api_key=os.environ["OPENAI_API_KEY"],
24
  )
25
 
 
 
 
 
26
  # agent = ToolCallingAgent(
27
  agent = CodeAgent(
28
  name="Jeff's GAIA L1 Agent",
29
  verbosity_level=1,
30
+ tools=[DuckDuckGoSearchTool()],
31
  add_base_tools=True,
32
  max_steps=5,
33
  model=model,
34
+ )
 
35
 
36
  self.agent = agent
37
  print(f"Agent initialized with model ID: {os.environ['MODEL_ID']}")