iQuentin commited on
Commit
1d7240f
·
verified ·
1 Parent(s): c3d68f4

Add wikipedia

Browse files
Files changed (1) hide show
  1. agent.py +3 -1
agent.py CHANGED
@@ -3,7 +3,8 @@ from smolagents import (
3
  CodeAgent,
4
  DuckDuckGoSearchTool,
5
  GoogleSearchTool,
6
- VisitWebpageTool,
 
7
  PythonInterpreterTool,
8
  InferenceClientModel,
9
  OpenAIServerModel,
@@ -104,6 +105,7 @@ class QAgent:
104
  DuckDuckGoSearchTool(),
105
  PythonInterpreterTool(),
106
  VisitWebpageTool(),
 
107
  # save_and_read_file,
108
  # download_file_from_url,
109
  # analyze_csv_file,
 
3
  CodeAgent,
4
  DuckDuckGoSearchTool,
5
  GoogleSearchTool,
6
+ VisitWebpageTool,
7
+ WikipediaSearchTool,
8
  PythonInterpreterTool,
9
  InferenceClientModel,
10
  OpenAIServerModel,
 
105
  DuckDuckGoSearchTool(),
106
  PythonInterpreterTool(),
107
  VisitWebpageTool(),
108
+ WikipediaSearchTool(),
109
  # save_and_read_file,
110
  # download_file_from_url,
111
  # analyze_csv_file,