Spaces:
Running
Running
Commit
·
40afc88
1
Parent(s):
b69c3f6
Fixing tool schema
Browse files- CEO/CEO.py +1 -1
- tools/tool_creator.py +1 -1
- tools/weather_tool.py +1 -1
CEO/CEO.py
CHANGED
@@ -138,7 +138,7 @@ if __name__ == "__main__":
|
|
138 |
tools.extend(tool_loader.getTools())
|
139 |
|
140 |
# Create the Ollama model manager and ensure the model is set up.
|
141 |
-
model_manager = OllamaModelManager()
|
142 |
model_manager.create_model("mistral")
|
143 |
|
144 |
# Example prompt instructing the CEO model to create a strategy for Ashton Hall.
|
|
|
138 |
tools.extend(tool_loader.getTools())
|
139 |
|
140 |
# Create the Ollama model manager and ensure the model is set up.
|
141 |
+
model_manager = OllamaModelManager(tools=tools)
|
142 |
model_manager.create_model("mistral")
|
143 |
|
144 |
# Example prompt instructing the CEO model to create a strategy for Ashton Hall.
|
tools/tool_creator.py
CHANGED
@@ -20,8 +20,8 @@ class ToolCreator():
|
|
20 |
"type": "string",
|
21 |
"description": "The content of the tool to create",
|
22 |
},
|
23 |
-
"required": ["name", "content"],
|
24 |
},
|
|
|
25 |
}
|
26 |
}
|
27 |
|
|
|
20 |
"type": "string",
|
21 |
"description": "The content of the tool to create",
|
22 |
},
|
|
|
23 |
},
|
24 |
+
"required": ["name", "content"],
|
25 |
}
|
26 |
}
|
27 |
|
tools/weather_tool.py
CHANGED
@@ -16,8 +16,8 @@ class WeatherApi():
|
|
16 |
"type": "string",
|
17 |
"description": "The location for which to get the weather information",
|
18 |
},
|
19 |
-
"required": ["location"],
|
20 |
},
|
|
|
21 |
}
|
22 |
}
|
23 |
|
|
|
16 |
"type": "string",
|
17 |
"description": "The location for which to get the weather information",
|
18 |
},
|
|
|
19 |
},
|
20 |
+
"required": ["location"],
|
21 |
}
|
22 |
}
|
23 |
|