iQuentin commited on
Commit
6f4227b
·
verified ·
1 Parent(s): 8d1ae36

Add GoogleST, VisitWebPage and some cleaning

Browse files
Files changed (1) hide show
  1. agent.py +3 -1
agent.py CHANGED
@@ -2,6 +2,7 @@ import os
2
  from smolagents import (
3
  CodeAgent,
4
  DuckDuckGoSearchTool,
 
5
  VisitWebpageTool,
6
  PythonInterpreterTool,
7
  InferenceClientModel,
@@ -102,6 +103,7 @@ class QAgent:
102
  self.tools = [
103
  DuckDuckGoSearchTool(),
104
  PythonInterpreterTool(),
 
105
  # save_and_read_file,
106
  # download_file_from_url,
107
  # analyze_csv_file,
@@ -117,7 +119,7 @@ class QAgent:
117
  tools=self.tools,
118
  model=self.model,
119
  instructions=extra,
120
- # additional_authorized_imports=self.imports,
121
  # executor_type=executor_type,
122
  # executor_kwargs=executor_kwargs,
123
  verbosity_level=2 if self.verbose else 0
 
2
  from smolagents import (
3
  CodeAgent,
4
  DuckDuckGoSearchTool,
5
+ GoogleSearchTool,
6
  VisitWebpageTool,
7
  PythonInterpreterTool,
8
  InferenceClientModel,
 
103
  self.tools = [
104
  DuckDuckGoSearchTool(),
105
  PythonInterpreterTool(),
106
+ VisitWebpageTool(),
107
  # save_and_read_file,
108
  # download_file_from_url,
109
  # analyze_csv_file,
 
119
  tools=self.tools,
120
  model=self.model,
121
  instructions=extra,
122
+ additional_authorized_imports=self.imports,
123
  # executor_type=executor_type,
124
  # executor_kwargs=executor_kwargs,
125
  verbosity_level=2 if self.verbose else 0