Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ import hashlib
|
|
8 |
import json
|
9 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
10 |
import torch
|
11 |
-
|
12 |
|
13 |
# (Keep Constants as is)
|
14 |
# --- Constants ---
|
@@ -58,7 +58,7 @@ class BasicAgent:
|
|
58 |
self.agent = CodeAgent(
|
59 |
model=model,
|
60 |
tools=[web_search,python_interpreter,visit_webpage_tool,math_model_tool],
|
61 |
-
max_steps=
|
62 |
verbosity_level=1,
|
63 |
grammar=None,
|
64 |
planning_interval=3,
|
|
|
8 |
import json
|
9 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
10 |
import torch
|
11 |
+
from tooling import ModelMathTool
|
12 |
|
13 |
# (Keep Constants as is)
|
14 |
# --- Constants ---
|
|
|
58 |
self.agent = CodeAgent(
|
59 |
model=model,
|
60 |
tools=[web_search,python_interpreter,visit_webpage_tool,math_model_tool],
|
61 |
+
max_steps=1,
|
62 |
verbosity_level=1,
|
63 |
grammar=None,
|
64 |
planning_interval=3,
|