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,