[Haik]: refactored the toolkits folder

backend/apps/agents/HaikFix/tools/BUILTIN_TOOLKIT/BUILTIN_TOOLKIT.py

@@ -1,28 +0,0 @@
-# BUILTIN_TOOLKIT.py
-
-from typing import Dict
-from backend.apps.agents.HaikFix.tools.shared_structs.Toolkit import Toolkit
-from backend.apps.agents.HaikFix.tools.BUILTIN_TOOLKIT.components.AGENTS_TOOLKIT import make_agents_toolkit
-from backend.apps.agents.HaikFix.tools.BUILTIN_TOOLKIT.components.FILESYSTEM_TOOLKIT import FILESYSTEM_TOOLKIT
-from backend.apps.agents.HaikFix.tools.BUILTIN_TOOLKIT.components.INTERACTION_TOOLKIT import INTERACTION_TOOLKIT
-from backend.apps.agents.HaikFix.tools.BUILTIN_TOOLKIT.components.PLANNING_TOOLKIT import PLANNING_TOOLKIT
-from backend.apps.agents.HaikFix.tools.BUILTIN_TOOLKIT.components.SCHEDULING_TOOLKIT import SCHEDULING_TOOLKIT
-from backend.apps.agents.HaikFix.tools.BUILTIN_TOOLKIT.components.SEARCH_TOOLKIT import SEARCH_TOOLKIT
-from backend.apps.agents.HaikFix.tools.BUILTIN_TOOLKIT.components.SYSTEM_TOOLKIT import SYSTEM_TOOLKIT
-from backend.apps.agents.HaikFix.Agent.Agent import Agent
-
-
-def make_builtin_toolkit(parent: Agent, agent_registry: Dict[str, Agent]) -> Toolkit:
-    return Toolkit(
-        name="builtin",
-        description="Builtin tools",
-        nested_toolkits=[
-            make_agents_toolkit(parent, agent_registry),
-            FILESYSTEM_TOOLKIT,
-            INTERACTION_TOOLKIT,
-            PLANNING_TOOLKIT,
-            SCHEDULING_TOOLKIT,
-            SEARCH_TOOLKIT,
-            SYSTEM_TOOLKIT,
-        ],
-    )

backend/apps/agents/HaikFix/tools/make_builtin_toolkit/make_builtin_toolkit.py

@@ -0,0 +1,30 @@
+# BUILTIN_TOOLKIT.py
+
+from typing import Dict
+from backend.apps.agents.HaikFix.tools.shared_structs.Toolkit import Toolkit
+from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.agents_toolkit.make_agents_toolkit import make_agents_toolkit
+# TODO: add browser toolkit
+# from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_toolkit import make_browser_toolkit
+
+from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.pre_existing_toolkits.FILESYSTEM_TOOLKIT import FILESYSTEM_TOOLKIT
+from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.pre_existing_toolkits.INTERACTION_TOOLKIT import INTERACTION_TOOLKIT
+from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.pre_existing_toolkits.PLANNING_TOOLKIT import PLANNING_TOOLKIT
+from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.pre_existing_toolkits.SCHEDULING_TOOLKIT import SCHEDULING_TOOLKIT
+from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.pre_existing_toolkits.SEARCH_TOOLKIT import SEARCH_TOOLKIT
+from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.pre_existing_toolkits.SYSTEM_TOOLKIT import SYSTEM_TOOLKIT
+from backend.apps.agents.HaikFix.Agent.Agent import Agent
+
+def make_builtin_toolkit(parent: Agent, agent_registry: Dict[str, Agent]) -> Toolkit:
+    return Toolkit(
+        name="builtin",
+        description="Builtin tools",
+        nested_toolkits=[
+            make_agents_toolkit(parent, agent_registry),
+            FILESYSTEM_TOOLKIT,
+            INTERACTION_TOOLKIT,
+            PLANNING_TOOLKIT,
+            SCHEDULING_TOOLKIT,
+            SEARCH_TOOLKIT,
+            SYSTEM_TOOLKIT,
+        ],
+    )

backend/apps/agents/HaikFix/tools/make_builtin_toolkit/open_swarm_toolkits/agents_toolkit/handlers/make_create_agent_handler.py

@@ -1,7 +1,6 @@
-from typing import Dict, Any, TypedDict
+from typing import TypedDict
 from typeguard import typechecked
 
-
 from backend.apps.agents.HaikFix.Agent.shared_structs.Message.agent_outputs import ToolResponse
 from backend.apps.agents.HaikFix.Agent.shared_structs.Message.Message import (
     AnyMessage, UserMessage, AssistantMessage,

backend/apps/agents/HaikFix/tools/make_builtin_toolkit/open_swarm_toolkits/agents_toolkit/handlers/make_invoke_agent_handler.py

@@ -1,9 +1,5 @@
-# make_invoke_agent_handler.py
-
 from typing import Dict, TypedDict
 from typeguard import typechecked
-from uuid import uuid4
-import asyncio
 
 from backend.apps.agents.HaikFix.Agent.shared_structs.Message.agent_outputs import ToolResponse
 from backend.apps.agents.HaikFix.Agent.shared_structs.Message.Message import (
@@ -19,6 +15,7 @@ class InvokeAgentInput(TypedDict):
 
 @typechecked
 def make_invoke_agent_handler(agent_registry: Dict[str, Agent]):
+    
     async def handler(args: InvokeAgentInput) -> ToolResponse:
         session_id = args["session_id"]
         message = args["message"]
@@ -52,5 +49,5 @@ def make_invoke_agent_handler(agent_registry: Dict[str, Agent]):
                 }
             ]
         }
-        
+
     return handler

backend/apps/agents/HaikFix/tools/make_builtin_toolkit/open_swarm_toolkits/browser_toolkit/handlers/make_create_browser_agent_handler.py

@@ -5,8 +5,8 @@ from typing_extensions import NotRequired
 from typeguard import typechecked
 
 from backend.apps.agents.HaikFix.Agent.shared_structs.Message.agent_outputs import ToolResponse
-from backend.apps.agents.HaikFix.tools.BUILTIN_TOOLKIT.components.custom_handlers.browser_utils.run_browser_loop import run_browser_loop
-from backend.apps.agents.HaikFix.tools.BUILTIN_TOOLKIT.components.custom_handlers.browser_utils.format_browser_result import format_browser_result
+from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.handlers.utils.run_browser_loop import run_browser_loop
+from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.handlers.utils.format_browser_result import format_browser_result
 from backend.apps.agents.HaikFix.Agent.Agent import Agent
 
 class CreateBrowserAgentInput(TypedDict):

backend/apps/agents/HaikFix/tools/make_builtin_toolkit/open_swarm_toolkits/browser_toolkit/handlers/utils/format_browser_result.py

@@ -16,7 +16,7 @@ def format_browser_result(result: Dict[str, Any], browser_id: str) -> ToolRespon
     if result["action_log"]:
         lines.append("")
         lines.append("**Actions taken:**")
-        for i, entry in enumerate(result["action_log"], 1):
+        for i, entry in enumerate[Any](result["action_log"], 1):
             tool = entry["tool"]
             inp = json.dumps(entry.get("input", {}))[:120]
             ms = entry.get("elapsed_ms", 0)