Some more refactor

main.py

@@ -3,8 +3,6 @@ import gradio as gr
 import base64
 from src.manager.manager import GeminiManager
 
-model_manager = GeminiManager(gemini_model="gemini-2.0-flash")
-
 _logo_bytes = open("HASHIRU_LOGO.png", "rb").read()
 _logo_b64 = base64.b64encode(_logo_bytes).decode()
 _header_html = f"""
@@ -45,6 +43,8 @@ def update_model(model_name):
 
 
 with gr.Blocks(css=css, fill_width=True, fill_height=True) as demo:
+    model_manager = GeminiManager(gemini_model="gemini-2.0-flash")
+    
     with gr.Column(scale=1):
         with gr.Row(scale=0):
             gr.Markdown(_header_html)

src/default_tools/update_memory.py

@@ -1,35 +0,0 @@
-
-__all__ = ['GetBudget']
-
-from src.manager.budget_manager import BudgetManager
-
-
-class GetBudget():
-    dependencies = []
-
-    inputSchema = {
-        "name": "GetBudget",
-        "description": "Retrieves the current budget available.",
-        "parameters": {
-            "type": "object",
-            "properties":{},
-            "required": [],
-        },
-    }
-
-
-    def run(self, **kwargs):
-        budget_manager = BudgetManager()
-        total_budget = budget_manager.get_total_budget()
-        current_expense = budget_manager.get_current_expense()
-        current_remaining_budget = budget_manager.get_current_remaining_budget()
-        return {
-            "status": "success",
-            "message": "Budget retrieved successfully",
-            "output": {
-                "total_budget": total_budget,
-                "current_expense": current_expense,
-                "current_remaining_budget": current_remaining_budget
-            }
-        }
-        

src/manager/config/model_selector.py

@@ -1,5 +1,5 @@
 from src.manager.utils.runtime_selector import detect_runtime_environment
-from cost_benefit import get_best_model
+from src.cost_benefit import get_best_model
 import os
 from dotenv import load_dotenv
 load_dotenv()

src/manager/tool_manager.py

@@ -5,18 +5,16 @@ import types
 from typing import List
 import pip
 from google.genai import types
-import sys
 
 from src.manager.budget_manager import BudgetManager
 from src.manager.utils.singleton import singleton
 from src.manager.utils.suppress_outputs import suppress_output
-from src.default_tools.get_agents_tool import GetAgents
-from src.default_tools.tool_deletor import ToolDeletor
+from src.tools.default_tools.tool_deletor import ToolDeletor
 from src.manager.utils.streamlit_interface import output_assistant_response
 
 toolsImported = []
 
-TOOLS_DIRECTORIES = [os.path.abspath("./src/default_tools"), os.path.abspath("./src/tools")]
+TOOLS_DIRECTORIES = [os.path.abspath("./src/tools/default_tools"), os.path.abspath("./src/tools/user_tools")]
 
 class Tool:
     def __init__(self, toolClass):

src/models/system3.prompt

@@ -79,7 +79,7 @@ Before firing an agent, evaluate its potential future usefulness in the conversa
 </Rule>
 
 <Rule>
-Tools are created in the tools/ directory. Before creating a new tool, you MUST read the directory using ListFiles tools and ReadFile tools to see how existing tools are implemented.
+Tools are created in the src/tools/default_tools and src/tools/user_tools directory. Before creating a new tool, you MUST 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.
 </Rule>
 

src/tools/default_tools/update_memory.py

@@ -0,0 +1,32 @@
+
+__all__ = ['UpdateMemory']
+
+from src.manager.budget_manager import BudgetManager
+
+
+class UpdateMemory():
+    dependencies = []
+
+    inputSchema = {
+        "name": "UpdateMemory",
+        "description": "Updates the memory of the AI agent. This tool is used to update the memory of the AI agent with new information.",
+        "parameters": {
+            "type": "object",
+            "properties":{
+                "memory": {
+                    "type": "string",
+                    "description": "The new memory to be added to the AI agent's memory.",
+                }
+            },
+            "required": ["memory"],
+        },
+    }
+
+
+    def run(self, **kwargs):
+        return {
+            "status": "fail",
+            "message": "This tool is not implemented yet",
+            "output": None
+        }
+