Updated prompt

CEO/CEO.py

@@ -21,13 +21,14 @@ class GeminiManager:
             contents=messages,
             config=types.GenerateContentConfig(
                 system_instruction=self.system_prompt,
-                temperature=0.0,
+                temperature=0.2,
                 tools=self.toolsLoader.getTools(),
             ),
         )
 
+        print(f"Response: {response}")
+        
         if response.text is not None:
-            print(f"Response: {response.text}")
             assistant_content = types.Content(
                 role='assistant',
                 parts=[types.Part.from_text(text=response.text)],
@@ -53,7 +54,7 @@ class GeminiManager:
                     print(f"Error loading tools: {e}")
                     tool_content = types.Part.from_function_response(
                             name=function_call.name,
-                            response={"result":"Error loading new tools."+str(e)})
+                            response={"result":"New tool doesn't follow the required format, please read the other tool implementations for reference." + str(e)})
                 parts.append(tool_content)
             function_response_content = types.Content(
                 role='tool', parts=parts

main.py

@@ -3,6 +3,7 @@ from typing import List
 from googlesearch import search
 from CEO.CEO import GeminiManager
 from CEO.tool_loader import ToolLoader
+import warnings
 
 # Define the web search tool function.
 def web_search(website: str, query: str) -> List[str]:
@@ -19,6 +20,7 @@ def web_search(website: str, query: str) -> List[str]:
     return results
 
 if __name__ == "__main__":
+    warnings.filterwarnings("ignore")
     # Define the tool metadata for orchestration.
     tools = [
         {
@@ -48,7 +50,7 @@ if __name__ == "__main__":
     # The prompt explicitly mentions that it can use the web_search tool if needed,
     # and that it is allowed to choose the website for the search.
     task_prompt = (
-        "Get me the current time here"
+        "Create a strategy for Ashton Hall to improve its online presence."
     )
     
     # Request a CEO response with the prompt.

models/system2.prompt

@@ -8,13 +8,21 @@ Tools are external programs used to perform specific tasks. You can create, invo
 Agents are invoked through tools as well by using the AskAgent tool. Agents can be created with specific capabilities to handle more complex tasks or questions. Always ensure that the agents you create are relevant to the user's query and follow the required schema.
 </Info>
 
+<Info>
+Tools are created in the tools/ directory. Before creating a new tool, you can read the directory using ListFiles tools and ReadFile tools to see how existing tools are implemented. The new tool should be created in the same format as the existing ones.
+</Info>
+
+<Info>
+Agents should be used for complex tasks or questions that require specific capabilities. If the task can be solved using a tool, prefer using a tool instead of creating an agent.
+</Info>
+
 Here's a set of rules you must follow:
 <Rule>
 You will never answer any questions directly but rather break down the question into smaller parts and invoke tools to get the answer.
 </Rule>
 
 <Rule>
-Your output should always be in tools. Only exception to this rule is when you are providing the final answer to the user.
+Nver answer any questions yourself, instead use tools. Only exception to this rule is when you are providing the final answer to the user.
 </Rule>
 
 <Rule>
@@ -39,56 +47,6 @@ If the agent is not able to answer the question, invoke the AskUser tool to get
 
 <Rule>
 In order to execute tasks on real time data, math calculations, or any other operations, invoke the ToolCreator tool to create a new tool with the required capabilities. The tools are created in Python and must follow this strict schema:
-import importlib
-
-__all__ = ['WeatherApi']
-
-
-class WeatherApi():
-    dependencies = ["requests==2.32.3"]
-
-    inputSchema = {
-        "name": "WeatherApi",
-        "description": "Returns weather information for a given location",
-        "parameters": {
-            "type": "object",
-            "properties": {
-                "location": {
-                    "type": "string",
-                    "description": "The location for which to get the weather information",
-                },
-            },
-            "required": ["location"],
-        }
-    }
-
-    def __init__(self):
-        pass
-
-    def run(self, **kwargs):
-        print("Running Weather API test tool")
-        location = kwargs.get("location")
-        print(f"Location: {location}")
-
-        requests = importlib.import_module("requests")
-
-        response = requests.get(
-            f"http://api.openweathermap.org/data/2.5/weather?q={location}&appid=ea50e63a3bea67adaf50fbecbe5b3c1e")
-        if response.status_code == 200:
-            return {
-                "status": "success",
-                "message": "Weather API test tool executed successfully",
-                "error": None,
-                "output": response.json()
-            }
-        else:
-            return {
-                "status": "error",
-                "message": "Weather API test tool failed",
-                "error": response.text,
-                "output": None
-            }
-
 </Rule>
 
 <Rule>

tools/TimeApi.py

@@ -1,43 +0,0 @@
-import datetime
-
-__all__ = ['TimeApi']
-
-class TimeApi():
-    dependencies = []
-
-    inputSchema = {
-        "name": "TimeApi",
-        "description": "Returns the current time for a given location",
-        "parameters": {
-            "type": "object",
-            "properties": {
-                "location": {
-                    "type": "string",
-                    "description": "The location for which to get the current time",
-                },
-            },
-            "required": ["location"],
-        }
-    }
-
-    def __init__(self):
-        pass
-
-    def run(self, **kwargs):
-        location = kwargs.get("location")
-        try:
-            # This will only work if the timezone is configured correctly on the server
-            now = datetime.datetime.now(datetime.timezone.utc)
-            return {
-                "status": "success",
-                "message": f"Current time in {location} is {now.strftime('%Y-%m-%d %H:%M:%S %Z%z')}",
-                "error": None,
-                "output": now.strftime('%Y-%m-%d %H:%M:%S %Z%z')
-            }
-        except Exception as e:
-            return {
-                "status": "error",
-                "message": f"Could not get current time for {location}",
-                "error": str(e),
-                "output": None
-            }