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
|