workaround until inference fix
Browse files
agent.py
CHANGED
|
@@ -7,7 +7,7 @@ Facultative
|
|
| 7 |
- arxiv_search
|
| 8 |
'''
|
| 9 |
|
| 10 |
-
from smolagents import CodeAgent, HfApiModel, tool, DuckDuckGoSearchTool
|
| 11 |
|
| 12 |
|
| 13 |
@tool
|
|
@@ -92,4 +92,15 @@ def rounder(a:float, n:int) -> float:
|
|
| 92 |
def get_agent() -> CodeAgent:
|
| 93 |
search_tool = DuckDuckGoSearchTool()
|
| 94 |
|
| 95 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
- arxiv_search
|
| 8 |
'''
|
| 9 |
|
| 10 |
+
from smolagents import CodeAgent, HfApiModel, tool, DuckDuckGoSearchTool, MLXModel
|
| 11 |
|
| 12 |
|
| 13 |
@tool
|
|
|
|
| 92 |
def get_agent() -> CodeAgent:
|
| 93 |
search_tool = DuckDuckGoSearchTool()
|
| 94 |
|
| 95 |
+
model = MLXModel(
|
| 96 |
+
"mlx-community/Qwen2.5-Coder-32B-Instruct-4bit",
|
| 97 |
+
{
|
| 98 |
+
"temperature": 0.7,
|
| 99 |
+
"top_k": 20,
|
| 100 |
+
"top_p": 0.8,
|
| 101 |
+
"min_p": 0.05,
|
| 102 |
+
"num_ctx": 32768,
|
| 103 |
+
},
|
| 104 |
+
)
|
| 105 |
+
|
| 106 |
+
return CodeAgent(tools=[add, subtract, multiply, divide, modulus, rounder, search_tool], model=model)
|