Adding tool loading logic to CEO

CEO/CEO.py

@@ -4,6 +4,7 @@ from typing import List, Dict, Optional
 from pathlib import Path
 import ollama
 from googlesearch import search
+from toolLoader import ToolLoader
 
 # Enum for Model Types
 class ModelType(Enum):
@@ -130,6 +131,11 @@ if __name__ == "__main__":
             },
         }
     ]
+
+    # Load the tools using the ToolLoader class.
+    tool_loader = ToolLoader()
+    tool_loader.load_tools()
+    tools.extend(tool_loader.getTools())
     
     # Create the Ollama model manager and ensure the model is set up.
     model_manager = OllamaModelManager()

CEO/toolLoader.py

@@ -26,35 +26,41 @@ class Tool:
     def run(self, query):
         return self.tool.run(**query)
 
-def load_tools():
-    for filename in os.listdir(TOOLS_DIRECTORY):
-        if filename.endswith(".py") and filename != "__init__.py":
-            module_name = filename[:-3]
-            spec = importlib.util.spec_from_file_location(module_name, f"{TOOLS_DIRECTORY}/{filename}")
-            foo = importlib.util.module_from_spec(spec)
-            spec.loader.exec_module(foo)
-            class_name = foo.__all__[0]
-            toolClass = getattr(foo, class_name)
-            toolObj = Tool(toolClass)
-            toolsImported.append(toolObj)
-
-def runTool(toolName, query):
-    for tool in toolsImported:
-        if tool.name == toolName:
-            return tool.run(query)
-    return None
-
-def getTools():
-    toolsList = []
-    for tool in toolsImported:
-        toolsList.append({
-            "type": "function",
-            "function": tool.inputSchema
-        })
-    return toolsList
-
-load_tools()
+class ToolLoader:
+    def __init__(self):
+        self.toolsImported = []
+        self.load_tools()
+        pass
+
+    def load_tools(self):
+        for filename in os.listdir(TOOLS_DIRECTORY):
+            if filename.endswith(".py") and filename != "__init__.py":
+                module_name = filename[:-3]
+                spec = importlib.util.spec_from_file_location(module_name, f"{TOOLS_DIRECTORY}/{filename}")
+                foo = importlib.util.module_from_spec(spec)
+                spec.loader.exec_module(foo)
+                class_name = foo.__all__[0]
+                toolClass = getattr(foo, class_name)
+                toolObj = Tool(toolClass)
+                self.toolsImported.append(toolObj)
+
+    def runTool(self, toolName, query):
+        for tool in self.toolsImported:
+            if tool.name == toolName:
+                return tool.run(query)
+        return None
+
+    def getTools(self):
+        toolsList = []
+        for tool in self.toolsImported:
+            toolsList.append({
+                "type": "function",
+                "function": tool.inputSchema
+            })
+        return toolsList
+
+toolLoader = ToolLoader()
 
 # Example usage
-print(getTools())
-print(runTool("WeatherApi", {"location": "Davis"}))
+# print(toolLoader.getTools())
+# print(toolLoader.runTool("WeatherApi", {"location": "Davis"}))

tools/test.py

@@ -0,0 +1,51 @@
+
+import importlib
+
+__all__ = ['ToolCreator']
+
+
+class ToolCreator():
+    dependencies = ["os"]
+
+    inputSchema = {
+        "name": "ToolCreator",
+        "description": "Creates a tool for the given function",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "name": {
+                    "type": "string",
+                    "description": "The name of the tool to create",
+                },
+                "content": {
+                    "type": "string",
+                    "description": "The content of the tool to create",
+                },
+                "required": ["name", "content"],
+            },
+        }
+    }
+
+    def __init__(self):
+        pass
+
+    def run(self, **kwargs):
+        print("Running Tool Creator")
+        name = kwargs.get("name")
+        content = kwargs.get("content")
+        print(f"Tool Name: {name}")
+        print(f"Tool Content: {content}")
+        # Create the tool file
+        tool_file_path = f"tools/{name}.py"
+        with open(tool_file_path, "w") as tool_file:
+            tool_file.write(content)
+        print(f"Tool file created at {tool_file_path}")
+        return {
+            "status": "success",
+            "message": "Tool created successfully",
+            "error": None,
+            "output": {
+                "tool_file_path": tool_file_path,
+                "tool_name": name,
+            }
+        }

tools/tool_creator.py

@@ -0,0 +1,50 @@
+import importlib
+
+__all__ = ['ToolCreator']
+
+
+class ToolCreator():
+    dependencies = ["os"]
+
+    inputSchema = {
+        "name": "ToolCreator",
+        "description": "Creates a tool for the given function",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "name": {
+                    "type": "string",
+                    "description": "The name of the tool to create",
+                },
+                "content": {
+                    "type": "string",
+                    "description": "The content of the tool to create",
+                },
+                "required": ["name", "content"],
+            },
+        }
+    }
+
+    def __init__(self):
+        pass
+
+    def run(self, **kwargs):
+        print("Running Tool Creator")
+        name = kwargs.get("name")
+        content = kwargs.get("content")
+        print(f"Tool Name: {name}")
+        print(f"Tool Content: {content}")
+        # Create the tool file
+        tool_file_path = f"tools/{name}.py"
+        with open(tool_file_path, "w") as tool_file:
+            tool_file.write(content)
+        print(f"Tool file created at {tool_file_path}")
+        return {
+            "status": "success",
+            "message": "Tool created successfully",
+            "error": None,
+            "output": {
+                "tool_file_path": tool_file_path,
+                "tool_name": name,
+            }
+        }