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 |
|