Commit
·
7ec791b
1
Parent(s):
967c695
Storing system prompt as well in the models file
Browse files
tools/agent_creater_tool.py
CHANGED
@@ -87,6 +87,7 @@ class AgentCreator():
|
|
87 |
models[agent_name] = {
|
88 |
"base_model": base_model,
|
89 |
"description": kwargs.get("description"),
|
|
|
90 |
"create_cost": self.inputSchema["creates"]["types"][base_model]["create_cost"],
|
91 |
"invoke_cost": self.inputSchema["creates"]["types"][base_model]["invoke_cost"],
|
92 |
}
|
|
|
87 |
models[agent_name] = {
|
88 |
"base_model": base_model,
|
89 |
"description": kwargs.get("description"),
|
90 |
+
"system_prompt": system_prompt,
|
91 |
"create_cost": self.inputSchema["creates"]["types"][base_model]["create_cost"],
|
92 |
"invoke_cost": self.inputSchema["creates"]["types"][base_model]["invoke_cost"],
|
93 |
}
|