helloparthshah commited on
Commit
40afc88
·
1 Parent(s): b69c3f6

Fixing tool schema

Browse files
Files changed (3) hide show
  1. CEO/CEO.py +1 -1
  2. tools/tool_creator.py +1 -1
  3. 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