helloparthshah commited on
Commit
7ec791b
·
1 Parent(s): 967c695

Storing system prompt as well in the models file

Browse files
Files changed (1) hide show
  1. tools/agent_creater_tool.py +1 -0
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
  }