Spaces:
Running
Running
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 |
-
|
50 |
-
|
|
|
|
|
|
|
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)
|