Spaces:
Running
Running
Harshil Patel
commited on
Commit
·
6115f71
1
Parent(s):
0e323dc
update agent creator to specify base model
Browse files
src/tools/default_tools/agent_creater_tool.py
CHANGED
|
@@ -31,8 +31,7 @@ class AgentCreator():
|
|
| 31 |
"description": "Description of the agent. This is a string that describes the agent and its capabilities. It should be a single line description.",
|
| 32 |
},
|
| 33 |
},
|
| 34 |
-
"required": ["agent_name", "system_prompt", "description"],
|
| 35 |
-
#"required": ["agent_name", "base_model", "system_prompt", "description"],
|
| 36 |
},
|
| 37 |
"creates": {
|
| 38 |
"selector": "base_model",
|
|
@@ -110,18 +109,13 @@ class AgentCreator():
|
|
| 110 |
def run(self, **kwargs):
|
| 111 |
print("Running Agent Creator")
|
| 112 |
agent_name = kwargs.get("agent_name")
|
| 113 |
-
|
| 114 |
-
|
| 115 |
-
env = detect_runtime_environment()
|
| 116 |
-
print(f"\n[DEBUG] Detected Runtime Environment: {env}")
|
| 117 |
print(f"[DEBUG] Selected Model: {base_model}")
|
| 118 |
|
| 119 |
system_prompt = kwargs.get("system_prompt")
|
| 120 |
description = kwargs.get("description")
|
| 121 |
create_cost = self.inputSchema["creates"]["types"][base_model]["create_cost"]
|
| 122 |
-
if base_model not in self.inputSchema["creates"]["types"]:
|
| 123 |
-
print(f"[WARN] Auto-selected model '{base_model}' not in schema. Falling back to gemini-2.0-flash")
|
| 124 |
-
base_model = "gemini-2.0-flash"
|
| 125 |
invoke_cost = self.inputSchema["creates"]["types"][base_model]["invoke_cost"]
|
| 126 |
|
| 127 |
agent_manager = AgentManager()
|
|
|
|
| 31 |
"description": "Description of the agent. This is a string that describes the agent and its capabilities. It should be a single line description.",
|
| 32 |
},
|
| 33 |
},
|
| 34 |
+
"required": ["agent_name", "base_model", "system_prompt", "description"],
|
|
|
|
| 35 |
},
|
| 36 |
"creates": {
|
| 37 |
"selector": "base_model",
|
|
|
|
| 109 |
def run(self, **kwargs):
|
| 110 |
print("Running Agent Creator")
|
| 111 |
agent_name = kwargs.get("agent_name")
|
| 112 |
+
base_model = kwargs.get("base_model")
|
| 113 |
+
|
|
|
|
|
|
|
| 114 |
print(f"[DEBUG] Selected Model: {base_model}")
|
| 115 |
|
| 116 |
system_prompt = kwargs.get("system_prompt")
|
| 117 |
description = kwargs.get("description")
|
| 118 |
create_cost = self.inputSchema["creates"]["types"][base_model]["create_cost"]
|
|
|
|
|
|
|
|
|
|
| 119 |
invoke_cost = self.inputSchema["creates"]["types"][base_model]["invoke_cost"]
|
| 120 |
|
| 121 |
agent_manager = AgentManager()
|