refactor

deleteAgents.py

@@ -1,8 +1,8 @@
 import json
 import ollama
-from src.utils.streamlit_interface import output_assistant_response
+from src.manager.utils.streamlit_interface import output_assistant_response
 
-with open("./models/models.json", "r", encoding="utf8") as f:
+with open("./src/models/models.json", "r", encoding="utf8") as f:
     models = f.read()
 models = json.loads(models)
 for agent in models:

main.py

@@ -1,7 +1,7 @@
 import gradio as gr
 
 import base64
-from src.manager import GeminiManager
+from src.manager.manager import GeminiManager
 
 model_manager = GeminiManager(gemini_model="gemini-2.0-flash")
 

{default_tools → src/default_tools}/agent_creater_tool.py

@@ -1,6 +1,6 @@
-from src.agent_manager import AgentManager
-from src.config.model_selector import choose_best_model
-from src.utils.runtime_selector import detect_runtime_environment
+from src.manager.agent_manager import AgentManager
+from src.manager.config.model_selector import choose_best_model
+from src.manager.utils.runtime_selector import detect_runtime_environment
 __all__ = ['AgentCreator']
 
 class AgentCreator():

{default_tools → src/default_tools}/ask_agent_tool.py

@@ -1,5 +1,5 @@
-from src.budget_manager import BudgetManager
-from src.agent_manager import AgentManager
+from src.manager.budget_manager import BudgetManager
+from src.manager.agent_manager import AgentManager
 
 __all__ = ['AskAgent']
 

{default_tools → src/default_tools}/fire_agent.py

@@ -1,4 +1,4 @@
-from src.agent_manager import AgentManager
+from src.manager.agent_manager import AgentManager
 
 __all__ = ['FireAgent']
 

{default_tools → src/default_tools}/get_agents_tool.py

@@ -1,4 +1,4 @@
-from src.agent_manager import AgentManager
+from src.manager.agent_manager import AgentManager
 
 __all__ = ['GetAgents']
 

{default_tools → src/default_tools}/get_budget.py

@@ -1,7 +1,7 @@
 
 __all__ = ['GetBudget']
 
-from src.budget_manager import BudgetManager
+from src.manager.budget_manager import BudgetManager
 
 
 class GetBudget():

{default_tools → src/default_tools}/test_cost/agent_creator_tool.py

@@ -1,7 +1,7 @@
-from src.agent_manager import AgentManager
-from src.config.model_selector import choose_best_model
-from src.utils.runtime_selector import detect_runtime_environment
-from cost_benefit import get_best_model
+from src.manager.agent_manager import AgentManager
+from src.manager.config.model_selector import choose_best_model
+from src.manager.utils.runtime_selector import detect_runtime_environment
+
 __all__ = ['AgentCreator']
 
 class AgentCreator():

src/default_tools/update_memory.py

@@ -0,0 +1,35 @@
+
+__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/{agent_manager.py → manager/agent_manager.py}

@@ -3,14 +3,14 @@ from typing import Dict, Type, Any, Optional, Tuple
 import os
 import json
 import ollama
-from src.utils.singleton import singleton
-from src.utils.streamlit_interface import output_assistant_response
+from src.manager.utils.singleton import singleton
+from src.manager.utils.streamlit_interface import output_assistant_response
 from google import genai
 from google.genai import types
 from google.genai.types import *
 import os
 from dotenv import load_dotenv
-from src.budget_manager import BudgetManager
+from src.manager.budget_manager import BudgetManager
 
 class Agent(ABC):
     

src/{budget_manager.py → manager/budget_manager.py}

@@ -1,4 +1,4 @@
-from src.utils.singleton import singleton
+from src.manager.utils.singleton import singleton
 
 @singleton
 class BudgetManager():

src/{config → manager/config}/model_selector.py

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

src/{llm_models.py → manager/llm_models.py}

@@ -5,7 +5,7 @@ from pathlib import Path
 from google import genai
 from google.genai import types
 from mistralai import Mistral
-from src.utils.streamlit_interface import output_assistant_response
+from src.manager.utils.streamlit_interface import output_assistant_response
 
 
 class AbstractModelManager(ABC):

src/{manager.py → manager/manager.py}

@@ -4,8 +4,8 @@ from google.genai.types import *
 import os
 from dotenv import load_dotenv
 import sys
-from src.tool_manager import ToolManager
-from src.utils.suppress_outputs import suppress_output
+from src.manager.tool_manager import ToolManager
+from src.manager.utils.suppress_outputs import suppress_output
 import logging
 import gradio as gr
 
@@ -17,7 +17,7 @@ logger.addHandler(handler)
 
 class GeminiManager:
     def __init__(self, toolsLoader: ToolManager = None,
-                 system_prompt_file="./models/system3.prompt",
+                 system_prompt_file="./src/models/system3.prompt",
                  gemini_model="gemini-2.5-pro-exp-03-25",
                  local_only=False, allow_tool_creation=True,
                  cloud_only=False, use_economy=True):

src/{tool_manager.py → manager/tool_manager.py}

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

src/{utils → manager/utils}/streamlit_interface.py

@@ -1,4 +1,4 @@
-from src.utils.singleton import singleton
+from src.manager.utils.singleton import singleton
 
 def get_user_message(message):
     user_input = input(message)

{tools → src/tools}/wordle_tool.py

@@ -33,7 +33,7 @@ class WordleTool():
     def __init__(self):
         self.secret_word = None
         self.word_list_url = "https://github.com/kiprobinson/wordle-solver/raw/main/app/resources/word-list.txt"
-        self.file_path = "secret_word.json"  # Path for storing the secret word
+        self.file_path = "src/data/secret_word.json"  # Path for storing the secret word
 
     def _load_word_list(self):
         try: