lcapriles commited on
Commit
f6b5bcc
·
verified ·
1 Parent(s): 8a22b9b
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,9 +24,10 @@ class BasicAgent:
24
  api_key=os.environ.get("api_key"),
25
  model_id=os.environ.get("model")
26
  )
 
27
  agent = CodeAgent(
28
  model=model,
29
- tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
30
  additional_authorized_imports=["pandas"],
31
  max_steps=20,
32
  )
@@ -34,7 +35,6 @@ class BasicAgent:
34
 
35
  def __call__(self, question: str) -> str:
36
  print(f"Agent received question (first 50 chars): {question[:50]}...")
37
- fixed_answer = "This is a default answer."
38
  answer = agent.run(question)
39
  print(f"Agent returning answer: {answer}")
40
  return answer
 
24
  api_key=os.environ.get("api_key"),
25
  model_id=os.environ.get("model")
26
  )
27
+
28
  agent = CodeAgent(
29
  model=model,
30
+ tools=[DuckDuckGoSearchTool(),VisitWebpageTool(),FinalAnswerTool()],
31
  additional_authorized_imports=["pandas"],
32
  max_steps=20,
33
  )
 
35
 
36
  def __call__(self, question: str) -> str:
37
  print(f"Agent received question (first 50 chars): {question[:50]}...")
 
38
  answer = agent.run(question)
39
  print(f"Agent returning answer: {answer}")
40
  return answer