Spaces:
Sleeping
Sleeping
Add GoogleST, VisitWebPage and some cleaning
Browse files
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 |
-
|
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
|