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