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