helloparthshah commited on
Commit
5ca37ce
·
1 Parent(s): 657847e

Fixing agent creator tool

Browse files
src/tools/default_tools/agent_creater_tool.py CHANGED
@@ -46,8 +46,11 @@ class AgentCreator():
46
  description = kwargs.get("description")
47
  model_costs = AgentCostManager().get_costs()
48
  if base_model not in model_costs:
49
- print(f"[WARN] Auto-selected model '{base_model}' not in schema. Falling back to gemini-2.0-flash")
50
- base_model = "gemini-2.0-flash"
 
 
 
51
  create_resource_cost = model_costs[base_model].get("create_resource_cost", 0)
52
  invoke_resource_cost = model_costs[base_model].get("invoke_resource_cost", 0)
53
  create_expense_cost = model_costs[base_model].get("create_expense_cost", 0)
 
46
  description = kwargs.get("description")
47
  model_costs = AgentCostManager().get_costs()
48
  if base_model not in model_costs:
49
+ return {
50
+ "status": "error",
51
+ "message": f"Model {base_model} not found in the cost manager.",
52
+ "output": None
53
+ }
54
  create_resource_cost = model_costs[base_model].get("create_resource_cost", 0)
55
  invoke_resource_cost = model_costs[base_model].get("invoke_resource_cost", 0)
56
  create_expense_cost = model_costs[base_model].get("create_expense_cost", 0)