Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ from smolagents import DuckDuckGoSearchTool,HfApiModel,load_tool,PythonInterpret
|
|
| 7 |
import hashlib
|
| 8 |
import json
|
| 9 |
from transformers import AutoTokenizer, AutoModelForCausalLM , pipeline, TransformersEngine
|
| 10 |
-
|
| 11 |
import torch
|
| 12 |
from tooling import ModelMathTool
|
| 13 |
|
|
@@ -105,6 +105,8 @@ class BasicAgent:
|
|
| 105 |
verbosity_level=1,
|
| 106 |
grammar=None,
|
| 107 |
planning_interval=1,
|
|
|
|
|
|
|
| 108 |
|
| 109 |
)
|
| 110 |
|
|
|
|
| 7 |
import hashlib
|
| 8 |
import json
|
| 9 |
from transformers import AutoTokenizer, AutoModelForCausalLM , pipeline, TransformersEngine
|
| 10 |
+
import wikipedia
|
| 11 |
import torch
|
| 12 |
from tooling import ModelMathTool
|
| 13 |
|
|
|
|
| 105 |
verbosity_level=1,
|
| 106 |
grammar=None,
|
| 107 |
planning_interval=1,
|
| 108 |
+
add_base_tools=True,
|
| 109 |
+
additional_authorized_imports=['requests','wikipedia']
|
| 110 |
|
| 111 |
)
|
| 112 |
|