Spaces:
Runtime error
Runtime error
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -255,11 +255,7 @@ sys_msg = SystemMessage(content=system_prompt)
|
|
| 255 |
# Map tool names to functions
|
| 256 |
# -----------------------------
|
| 257 |
tool_map = {
|
| 258 |
-
"
|
| 259 |
-
"add": add,
|
| 260 |
-
"subtract": subtract,
|
| 261 |
-
"divide": divide,
|
| 262 |
-
"modulus": modulus,
|
| 263 |
"wiki_search": wiki_search,
|
| 264 |
"web_search": web_search,
|
| 265 |
"arvix_search": arvix_search,
|
|
|
|
| 255 |
# Map tool names to functions
|
| 256 |
# -----------------------------
|
| 257 |
tool_map = {
|
| 258 |
+
"math": calculator,
|
|
|
|
|
|
|
|
|
|
|
|
|
| 259 |
"wiki_search": wiki_search,
|
| 260 |
"web_search": web_search,
|
| 261 |
"arvix_search": arvix_search,
|