Update agent.py
Browse files
agent.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
from smolagents import tool
|
| 2 |
-
from smolagents import CodeAgent,
|
| 3 |
|
| 4 |
@tool
|
| 5 |
def multiply(a: int, b: int) -> int:
|
|
@@ -59,7 +59,9 @@ def return_agent(question):
|
|
| 59 |
add,
|
| 60 |
subtract,
|
| 61 |
divide,
|
| 62 |
-
modulus
|
|
|
|
|
|
|
| 63 |
],
|
| 64 |
model=InferenceClientModel()
|
| 65 |
)
|
|
|
|
| 1 |
from smolagents import tool
|
| 2 |
+
from smolagents import CodeAgent, InferenceClientModel, VisitWebpageTool, DuckDuckGoSearchTool
|
| 3 |
|
| 4 |
@tool
|
| 5 |
def multiply(a: int, b: int) -> int:
|
|
|
|
| 59 |
add,
|
| 60 |
subtract,
|
| 61 |
divide,
|
| 62 |
+
modulus,
|
| 63 |
+
DuckDuckGoSearchTool,
|
| 64 |
+
VisitWebpageTool
|
| 65 |
],
|
| 66 |
model=InferenceClientModel()
|
| 67 |
)
|