Spaces:
Running
Running
Kunal Pai commited on
Commit ·
2c0c391
1
Parent(s): 338633f
Add support for deepseek-r1 model in AgentManager and AgentCostManager
Browse files
src/manager/agent_manager.py
CHANGED
|
@@ -326,6 +326,8 @@ class AgentManager():
|
|
| 326 |
return "ollama"
|
| 327 |
elif base_model == "mistral":
|
| 328 |
return "ollama"
|
|
|
|
|
|
|
| 329 |
elif "gemini" in base_model:
|
| 330 |
return "gemini"
|
| 331 |
else:
|
|
|
|
| 326 |
return "ollama"
|
| 327 |
elif base_model == "mistral":
|
| 328 |
return "ollama"
|
| 329 |
+
elif base_model == "deepseek-r1":
|
| 330 |
+
return "ollama"
|
| 331 |
elif "gemini" in base_model:
|
| 332 |
return "gemini"
|
| 333 |
else:
|
src/tools/default_tools/agent_cost_manager.py
CHANGED
|
@@ -25,6 +25,11 @@ class AgentCostManager():
|
|
| 25 |
"create_resource_cost": 20,
|
| 26 |
"invoke_resource_cost": 50,
|
| 27 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
"gemini-2.5-flash-preview-04-17": {
|
| 29 |
"description": "Adaptive thinking, cost efficiency",
|
| 30 |
"create_expense_cost": 20,
|
|
|
|
| 25 |
"create_resource_cost": 20,
|
| 26 |
"invoke_resource_cost": 50,
|
| 27 |
},
|
| 28 |
+
"deepseek-r1": {
|
| 29 |
+
"description": "7 Billion reasoning model",
|
| 30 |
+
"create_resource_cost": 20,
|
| 31 |
+
"invoke_resource_cost": 50,
|
| 32 |
+
},
|
| 33 |
"gemini-2.5-flash-preview-04-17": {
|
| 34 |
"description": "Adaptive thinking, cost efficiency",
|
| 35 |
"create_expense_cost": 20,
|