RCaz commited on
Commit
b861e24
·
verified ·
1 Parent(s): 3bc88d4

added wikipedia retreiver tool

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ import yaml
7
  from smolagents import DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, GoogleSearchTool, FinalAnswerTool, InferenceClientModel
8
 
9
  from tools import translator, image_qa
10
-
11
 
12
  # (Keep Constants as is)
13
  # --- Constants ---
@@ -27,7 +27,7 @@ class BasicAgent:
27
  )
28
  self.agent = CodeAgent(
29
  model=model,
30
- tools=[DuckDuckGoSearchTool(), WikipediaSearchTool(),FinalAnswerTool(),translator, image_qa],
31
  prompt_templates=prompt_templates,
32
  additional_authorized_imports=['datetime','numpy','pandas'],
33
  max_steps=3
 
7
  from smolagents import DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, GoogleSearchTool, FinalAnswerTool, InferenceClientModel
8
 
9
  from tools import translator, image_qa
10
+ from retreiver import retriever_tool
11
 
12
  # (Keep Constants as is)
13
  # --- Constants ---
 
27
  )
28
  self.agent = CodeAgent(
29
  model=model,
30
+ tools=[DuckDuckGoSearchTool(), WikipediaSearchTool(),FinalAnswerTool(),translator, image_qa, retriever_tool],
31
  prompt_templates=prompt_templates,
32
  additional_authorized_imports=['datetime','numpy','pandas'],
33
  max_steps=3