Fixing tool deletion

src/manager/agent_manager.py

@@ -12,6 +12,9 @@ import os
 from dotenv import load_dotenv
 from src.manager.budget_manager import BudgetManager
 
+MODEL_PATH = "./src/models/"
+MODEL_FILE_PATH = "./src/models/models.json"
+
 class Agent(ABC):
     
     def __init__(self, agent_name: str, base_model: str, system_prompt: str, creation_cost: str, invoke_cost: str):
@@ -105,7 +108,10 @@ class AgentManager():
         
         self._load_agents()
     
-    def create_agent(self, agent_name: str, base_model: str, system_prompt: str, description: str = "", create_cost: float = 0, invoke_cost: float = 0, 
+    def create_agent(self, agent_name: str, 
+                     base_model: str, system_prompt: str, 
+                     description: str = "", create_cost: float = 0, 
+                     invoke_cost: float = 0, 
                  **additional_params) -> Tuple[Agent, int]:
         
         if agent_name in self._agents:
@@ -163,8 +169,8 @@ class AgentManager():
     def list_agents(self) -> dict:
         """Return agent information (name, description, costs)"""
         try:
-            if os.path.exists("./models/models.json"):
-                with open("./models/models.json", "r", encoding="utf8") as f:
+            if os.path.exists(MODEL_FILE_PATH):
+                with open(MODEL_FILE_PATH, "r", encoding="utf8") as f:
                     full_models = json.loads(f.read())
                     
                 # Create a simplified version with only the description and costs
@@ -191,12 +197,12 @@ class AgentManager():
         
         del self._agents[agent_name]
         try:
-            if os.path.exists("./models/models.json"):
-                with open("./models/models.json", "r", encoding="utf8") as f:
+            if os.path.exists(MODEL_FILE_PATH):
+                with open(MODEL_FILE_PATH, "r", encoding="utf8") as f:
                     models = json.loads(f.read())
                     
                 del models[agent_name]
-                with open("./models/models.json", "w", encoding="utf8") as f:
+                with open(MODEL_FILE_PATH, "w", encoding="utf8") as f:
                     f.write(json.dumps(models, indent=4))
         except Exception as e:
             output_assistant_response(f"Error deleting agent: {e}")
@@ -216,11 +222,11 @@ class AgentManager():
         """Save a single agent to the models.json file"""
         try:
             # Ensure the directory exists
-            os.makedirs("./models", exist_ok=True)
+            os.makedirs(MODEL_PATH, exist_ok=True)
             
             # Read existing models file or create empty dict if it doesn't exist
             try:
-                with open("./models/models.json", "r", encoding="utf8") as f:
+                with open(MODEL_FILE_PATH, "r", encoding="utf8") as f:
                     models = json.loads(f.read())
             except (FileNotFoundError, json.JSONDecodeError):
                 models = {}
@@ -239,7 +245,7 @@ class AgentManager():
                 models[agent_name][key] = value
             
             # Write the updated models back to the file
-            with open("./models/models.json", "w", encoding="utf8") as f:
+            with open(MODEL_FILE_PATH, "w", encoding="utf8") as f:
                 f.write(json.dumps(models, indent=4))
                 
         except Exception as e:
@@ -259,10 +265,10 @@ class AgentManager():
     def _load_agents(self) -> None:
         """Load agent configurations from disk"""
         try:
-            if not os.path.exists("./models/models.json"):
+            if not os.path.exists(MODEL_FILE_PATH):
                 return
                 
-            with open("./models/models.json", "r", encoding="utf8") as f:
+            with open(MODEL_FILE_PATH, "r", encoding="utf8") as f:
                 models = json.loads(f.read())
             
             for name, data in models.items():

src/manager/manager.py

@@ -98,14 +98,11 @@ class GeminiManager:
                 self.toolsLoader.load_tools()
             except Exception as e:
                 logger.info(f"Error loading tools: {e}. Deleting the tool.")
-                thinking += f"Error loading tools: {e}. Deleting the tool.\n"
                 yield {
                     "role": "assistant",
-                    "content": thinking,
+                    "content": f"Error loading tools: {e}. Deleting the tool.\n",
                     "metadata": {
-                        "title": title,
-                        "id": i,
-                        "status": "done",
+                        "title": "Trying to load the newly created tool",
                     }
                 }
                 # delete the created tool

src/models/models.json

@@ -1 +1,16 @@
-{}
+{
+    "RecipeAgent": {
+        "base_model": "gemini-2.5-flash-preview-04-17",
+        "description": "Agent that extracts recipe from a website",
+        "system_prompt": "You are a recipe extraction agent. Given a URL, you extract the title, ingredients, and instructions from the website. If you cannot find the recipe, you return an error message.",
+        "create_cost": 20,
+        "invoke_cost": 50
+    },
+    "TimeAgent": {
+        "base_model": "gemini-2.0-flash",
+        "description": "An agent that provides the current time.",
+        "system_prompt": "You are a helpful agent that provides the current time in the format YYYY-MM-DD HH:MM:SS. Use the current date and time to respond to the user. Do not use any external tools or APIs.",
+        "create_cost": 20,
+        "invoke_cost": 50
+    }
+}

src/models/system3.prompt

@@ -34,7 +34,7 @@ If you're over this budget, you can no longer create new tools. In case this hap
 </Info>
 
 <Info>
-Once the you're finished with a conversation, you can use the DigestConversation to clear the conversation history and create a summary of the interaction. This summary can be used to improve the system's performance and user experience in future interactions.
+If user shares something important or sensitive, you should usee the UpdateMemory tool to store this information. Include details such as the user's preferences, how you solved errors, and any other relevant context that could help in future interactions. This will help you provide better assistance in the future.
 </Info>
 
 Here's a set of rules you must follow:
@@ -123,5 +123,9 @@ When you go over the resource budget, you must carefully evaluate which agent is
 </Rule>
 
 <Rule>
-When using the DigestConversation tool, ensure that the summary is clear and concise, capturing the main points of the conversation without unnecessary details. Include any important information such as how errors were solved, user's preferences, and any other relevant context that could help in future interactions.
+Every time you encounter an error and fix it, you should use the UpdateMemory tool to store the FIX.
+</Rule>
+
+<Rule>
+Every time user corrects you or provides feedback, you should use the UpdateMemory tool to store this information.
 </Rule>

src/tools/default_tools/tool_creator.py

@@ -23,6 +23,15 @@ class ToolCreator():
             "required": ["name", "tool_code"],
         }
     }
+    
+    def validate_tool_code(self, tool_code):
+        # Basic validation to check if the code is a valid Python function
+        try:
+            compile(tool_code, '<string>', 'exec')
+            return True, None
+        except SyntaxError as e:
+            print(f"Syntax error in tool code: {e}")
+            return False, e
 
     def run(self, **kwargs):
         print("Running Tool Creator")
@@ -31,7 +40,7 @@ class ToolCreator():
         print(f"Tool Name: {name}")
         print(f"Tool Content: {content}")
         # Create the tool file
-        tool_file_path = f"tools/{name}.py"
+        tool_file_path = f"src/tools/user_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}")

src/tools/default_tools/tool_deletor.py

@@ -30,7 +30,7 @@ class ToolDeletor():
         name = kwargs.get("name")
         file_path = kwargs.get("file_path")
         # make sure the file path is in tools/
-        if not file_path.startswith("tools/"):
+        if not file_path.startswith("src/tools/user_tools/"):
             return {
                 "status": "error",
                 "message": "File path must start with tools/",

src/tools/default_tools/update_memory.py

@@ -1,7 +1,7 @@
 
 __all__ = ['UpdateMemory']
 
-from src.manager.budget_manager import BudgetManager
+import json
 
 
 class UpdateMemory():
@@ -24,9 +24,32 @@ class UpdateMemory():
 
 
     def run(self, **kwargs):
+        # save it to src/data/memory.json
+        memory = kwargs.get("memory")
+        if not memory:
+            return {
+                "status": "error",
+                "message": "Memory is required",
+                "output": None
+            }
+        # add the memory to the memory.json file which is list of strings
+        # create the file if it does not exist
+        try:
+            with open("src/data/memory.json", "r") as f:
+                memory_list = json.load(f)
+        except FileNotFoundError:
+            memory_list = []
+        except json.JSONDecodeError:
+            memory_list = []
+        
+        # add the new memory to the list
+        memory_list.append(memory)
+        # save the list to the file
+        with open("src/data/memory.json", "w") as f:
+            json.dump(memory_list, f)
+        # return the list of memories
         return {
-            "status": "fail",
-            "message": "This tool is not implemented yet",
-            "output": None
-        }
-        
+            "status": "success",
+            "message": "Memory updated successfully",
+            "output": memory_list
+        }