mirror of
https://github.com/openswarm-ai/openswarm.git
synced 2026-09-11 12:17:45 +02:00
[Haik]: ckpt, swapped the HaikFix folder to be parallel to the agents folder bc ima start the full replacement process i think
This commit is contained in:
@@ -11,10 +11,10 @@ from typing import List, Literal, Optional
|
||||
from typeguard import typechecked
|
||||
|
||||
from backend.apps.agents.manager.ws_manager import ws_manager
|
||||
from backend.apps.agents.HaikFix.Agent.run_agent_loop.run_agent_loop import run_agent_loop
|
||||
from backend.apps.agents.HaikFix.Agent.shared_structs.Message.Message import Message
|
||||
from backend.apps.agents.HaikFix.Agent.shared_structs.ApprovalRequest import ApprovalRequest
|
||||
from backend.apps.agents.HaikFix.Agent.shared_structs.MessageLog import MessageLog
|
||||
from backend.apps.HaikFix.Agent.run_agent_loop.run_agent_loop import run_agent_loop
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.Message import Message
|
||||
from backend.apps.HaikFix.Agent.shared_structs.ApprovalRequest import ApprovalRequest
|
||||
from backend.apps.HaikFix.Agent.shared_structs.MessageLog import MessageLog
|
||||
|
||||
os.environ.setdefault("CLAUDE_CODE_STREAM_CLOSE_TIMEOUT", "3600000")
|
||||
|
||||
+2
-2
@@ -1,5 +1,5 @@
|
||||
from backend.apps.agents.HaikFix.Agent.shared_structs.Message.Message import Message
|
||||
from backend.apps.agents.HaikFix.Agent.shared_structs.Message.agent_outputs import ToolCallContent
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.Message import Message
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.agent_outputs import ToolCallContent
|
||||
from claude_agent_sdk.types import TextBlock, ToolUseBlock, AssistantMessage
|
||||
from backend.apps.agents.manager.ws_manager import ws_manager
|
||||
from typeguard import typechecked
|
||||
+2
-2
@@ -6,10 +6,10 @@ from datetime import datetime
|
||||
from uuid import uuid4
|
||||
from typeguard import typechecked
|
||||
|
||||
from backend.apps.agents.HaikFix.Agent.shared_structs.Message.agent_inputs import (
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.agent_inputs import (
|
||||
PromptBlock, TextPromptBlock, ImagePromptBlock, ImageSource, ContextPath, SkillMeta
|
||||
)
|
||||
from backend.apps.agents.HaikFix.Agent.shared_structs.Message.agent_outputs import (
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.agent_outputs import (
|
||||
ToolCallContent, ToolResultContent
|
||||
)
|
||||
|
||||
+1
-1
@@ -2,7 +2,7 @@ from typeguard import typechecked
|
||||
from pydantic import BaseModel, Field
|
||||
from typing import List
|
||||
|
||||
from backend.apps.agents.HaikFix.Agent.shared_structs.Message.Message import AnyMessage
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.Message import AnyMessage
|
||||
|
||||
|
||||
class MessageLog(BaseModel):
|
||||
@@ -0,0 +1,35 @@
|
||||
# BUILTIN_TOOLKIT.py
|
||||
|
||||
from typing import Dict
|
||||
from typeguard import typechecked
|
||||
from backend.apps.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.agents_toolkit.make_agents_toolkit import make_agents_toolkit
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_toolkit import make_browser_toolkit
|
||||
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.pre_existing_toolkits.FILESYSTEM_TOOLKIT import FILESYSTEM_TOOLKIT
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.pre_existing_toolkits.INTERACTION_TOOLKIT import INTERACTION_TOOLKIT
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.pre_existing_toolkits.PLANNING_TOOLKIT import PLANNING_TOOLKIT
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.pre_existing_toolkits.SCHEDULING_TOOLKIT import SCHEDULING_TOOLKIT
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.pre_existing_toolkits.SEARCH_TOOLKIT import SEARCH_TOOLKIT
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.pre_existing_toolkits.SYSTEM_TOOLKIT import SYSTEM_TOOLKIT
|
||||
from backend.apps.HaikFix.Agent.Agent import Agent
|
||||
|
||||
@typechecked
|
||||
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),
|
||||
make_browser_toolkit(parent),
|
||||
FILESYSTEM_TOOLKIT,
|
||||
INTERACTION_TOOLKIT,
|
||||
PLANNING_TOOLKIT,
|
||||
SCHEDULING_TOOLKIT,
|
||||
SEARCH_TOOLKIT,
|
||||
SYSTEM_TOOLKIT,
|
||||
],
|
||||
)
|
||||
+3
-3
@@ -1,11 +1,11 @@
|
||||
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 (
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.agent_outputs import ToolResponse
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.Message import (
|
||||
AnyMessage, UserMessage, AssistantMessage,
|
||||
)
|
||||
from backend.apps.agents.HaikFix.Agent.Agent import Agent
|
||||
from backend.apps.HaikFix.Agent.Agent import Agent
|
||||
|
||||
class CreateAgentInput(TypedDict):
|
||||
task: str
|
||||
+3
-3
@@ -1,11 +1,11 @@
|
||||
from typing import Dict, 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 (
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.agent_outputs import ToolResponse
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.Message import (
|
||||
AnyMessage, UserMessage, AssistantMessage,
|
||||
)
|
||||
from backend.apps.agents.HaikFix.Agent.Agent import Agent
|
||||
from backend.apps.HaikFix.Agent.Agent import Agent
|
||||
|
||||
|
||||
class InvokeAgentInput(TypedDict):
|
||||
+5
-5
@@ -1,13 +1,13 @@
|
||||
from typing import Dict
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.MCP_Tool import SDK_MCP_Tool
|
||||
from backend.apps.agents.HaikFix.tools.BUILTIN_TOOLKIT.components.custom_handlers.make_create_agent_handler import (
|
||||
from backend.apps.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.HaikFix.tools.shared_structs.MCP_Tool import SDK_MCP_Tool
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.agents_toolkit.handlers.make_create_agent_handler import (
|
||||
make_create_agent_handler, CreateAgentInput,
|
||||
)
|
||||
from backend.apps.agents.HaikFix.tools.BUILTIN_TOOLKIT.components.custom_handlers.make_invoke_agent_handler import (
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.agents_toolkit.handlers.make_invoke_agent_handler import (
|
||||
make_invoke_agent_handler, InvokeAgentInput,
|
||||
)
|
||||
from backend.apps.agents.HaikFix.Agent.Agent import Agent
|
||||
from backend.apps.HaikFix.Agent.Agent import Agent
|
||||
|
||||
def make_agents_toolkit(parent: Agent, agent_registry: Dict[str, Agent]) -> Toolkit:
|
||||
return Toolkit(
|
||||
+1
-1
@@ -1,6 +1,6 @@
|
||||
from typing import Dict, Any
|
||||
from typeguard import typechecked
|
||||
from backend.apps.agents.HaikFix.Agent.shared_structs.Message.agent_outputs import ToolResponse
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.agent_outputs import ToolResponse
|
||||
from backend.apps.agents.browser.executor import execute_browser_tool, _format_tool_result
|
||||
|
||||
@typechecked
|
||||
+4
-4
@@ -1,7 +1,7 @@
|
||||
from typeguard import typechecked
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.MCP_Tool import SDK_MCP_Tool
|
||||
from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_actions_toolkit.handlers.browser_action_input_schemas import (
|
||||
from backend.apps.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.HaikFix.tools.shared_structs.MCP_Tool import SDK_MCP_Tool
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_actions_toolkit.handlers.browser_action_input_schemas import (
|
||||
BrowserScreenshotInput,
|
||||
BrowserGetTextInput,
|
||||
BrowserNavigateInput,
|
||||
@@ -12,7 +12,7 @@ from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.
|
||||
BrowserScrollInput,
|
||||
BrowserWaitInput,
|
||||
)
|
||||
from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_actions_toolkit.handlers.make_browser_action_handler import make_browser_action_handler
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_actions_toolkit.handlers.make_browser_action_handler import make_browser_action_handler
|
||||
|
||||
SERVER_NAME = "openswarm-browser-actions"
|
||||
|
||||
+4
-4
@@ -2,10 +2,10 @@ import asyncio
|
||||
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.Agent import Agent
|
||||
from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_delegation_toolkit.handlers.utils.run_browser_agent import run_browser_agent
|
||||
from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_delegation_toolkit.handlers.utils.create_browser_card import create_browser_card
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.agent_outputs import ToolResponse
|
||||
from backend.apps.HaikFix.Agent.Agent import Agent
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_delegation_toolkit.handlers.utils.run_browser_agent import run_browser_agent
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_delegation_toolkit.handlers.utils.create_browser_card import create_browser_card
|
||||
|
||||
|
||||
class CreateBrowserAgentInput(TypedDict):
|
||||
+3
-3
@@ -1,9 +1,9 @@
|
||||
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.Agent import Agent
|
||||
from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_delegation_toolkit.handlers.utils.run_browser_agent import run_browser_agent
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.agent_outputs import ToolResponse
|
||||
from backend.apps.HaikFix.Agent.Agent import Agent
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_delegation_toolkit.handlers.utils.run_browser_agent import run_browser_agent
|
||||
|
||||
|
||||
class InvokeBrowserAgentInput(TypedDict):
|
||||
+5
-5
@@ -4,13 +4,13 @@ from typeguard import typechecked
|
||||
from claude_agent_sdk import ClaudeAgentOptions
|
||||
from claude_agent_sdk.types import McpServerConfig
|
||||
|
||||
from backend.apps.agents.HaikFix.Agent.Agent import Agent
|
||||
from backend.apps.agents.HaikFix.Agent.shared_structs.Message.Message import (
|
||||
from backend.apps.HaikFix.Agent.Agent import Agent
|
||||
from backend.apps.HaikFix.Agent.shared_structs.Message.Message import (
|
||||
UserMessage, AssistantMessage, AnyMessage,
|
||||
)
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.MCP_Tool import SDK_MCP_Tool
|
||||
from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_actions_toolkit.make_browser_actions_toolkit import make_browser_actions_toolkit
|
||||
from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_delegation_toolkit.handlers.utils.constants import BROWSER_AGENT_SYSTEM_PROMPT
|
||||
from backend.apps.HaikFix.tools.shared_structs.MCP_Tool import SDK_MCP_Tool
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_actions_toolkit.make_browser_actions_toolkit import make_browser_actions_toolkit
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_delegation_toolkit.handlers.utils.constants import BROWSER_AGENT_SYSTEM_PROMPT
|
||||
|
||||
|
||||
@typechecked
|
||||
+5
-5
@@ -1,13 +1,13 @@
|
||||
from typeguard import typechecked
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.MCP_Tool import SDK_MCP_Tool
|
||||
from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_delegation_toolkit.handlers.make_create_browser_agent_handler import (
|
||||
from backend.apps.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.HaikFix.tools.shared_structs.MCP_Tool import SDK_MCP_Tool
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_delegation_toolkit.handlers.make_create_browser_agent_handler import (
|
||||
make_create_browser_agent_handler, CreateBrowserAgentInput,
|
||||
)
|
||||
from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_delegation_toolkit.handlers.make_invoke_browser_agent_handler import (
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_delegation_toolkit.handlers.make_invoke_browser_agent_handler import (
|
||||
make_invoke_browser_agent_handler, InvokeBrowserAgentInput,
|
||||
)
|
||||
from backend.apps.agents.HaikFix.Agent.Agent import Agent
|
||||
from backend.apps.HaikFix.Agent.Agent import Agent
|
||||
|
||||
|
||||
@typechecked
|
||||
+18
@@ -0,0 +1,18 @@
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_actions_toolkit.make_browser_actions_toolkit import make_browser_actions_toolkit
|
||||
from backend.apps.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_delegation_toolkit.make_browser_delegation_toolkit import make_browser_delegation_toolkit
|
||||
from backend.apps.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.HaikFix.Agent.Agent import Agent
|
||||
|
||||
# # NOTE: The minor issue here is that with the current setup, any agent will prly have access to the browser actions toolkit, idk if this bad or good but for now its fine ig
|
||||
|
||||
def make_browser_toolkit(
|
||||
parent: Agent,
|
||||
) -> Toolkit:
|
||||
return Toolkit(
|
||||
name="browser",
|
||||
description="Tools for browser automation",
|
||||
nested_toolkits=[
|
||||
make_browser_delegation_toolkit(parent),
|
||||
make_browser_actions_toolkit(parent),
|
||||
],
|
||||
)
|
||||
+2
-2
@@ -1,5 +1,5 @@
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Tool import Tool
|
||||
from backend.apps.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.HaikFix.tools.shared_structs.Tool import Tool
|
||||
|
||||
FILESYSTEM_TOOLKIT = Toolkit(
|
||||
name="filesystem",
|
||||
+2
-2
@@ -1,5 +1,5 @@
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Tool import Tool
|
||||
from backend.apps.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.HaikFix.tools.shared_structs.Tool import Tool
|
||||
|
||||
INTERACTION_TOOLKIT = Toolkit(
|
||||
name="interaction",
|
||||
+2
-2
@@ -1,5 +1,5 @@
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Tool import Tool
|
||||
from backend.apps.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.HaikFix.tools.shared_structs.Tool import Tool
|
||||
|
||||
PLANNING_TOOLKIT = Toolkit(
|
||||
name="planning",
|
||||
+2
-2
@@ -1,5 +1,5 @@
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Tool import Tool
|
||||
from backend.apps.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.HaikFix.tools.shared_structs.Tool import Tool
|
||||
|
||||
SCHEDULING_TOOLKIT = Toolkit(
|
||||
name="scheduling",
|
||||
+2
-2
@@ -1,5 +1,5 @@
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Tool import Tool
|
||||
from backend.apps.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.HaikFix.tools.shared_structs.Tool import Tool
|
||||
|
||||
SEARCH_TOOLKIT = Toolkit(
|
||||
name="search",
|
||||
+2
-2
@@ -1,5 +1,5 @@
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Tool import Tool
|
||||
from backend.apps.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.HaikFix.tools.shared_structs.Tool import Tool
|
||||
|
||||
SYSTEM_TOOLKIT = Toolkit(
|
||||
name="system",
|
||||
+1
-1
@@ -12,7 +12,7 @@ from claude_agent_sdk.types import (
|
||||
McpServerConfig
|
||||
)
|
||||
from typeguard import typechecked
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Tool import Tool
|
||||
from backend.apps.HaikFix.tools.shared_structs.Tool import Tool
|
||||
|
||||
class MCP_Tool(Tool):
|
||||
server_name: str
|
||||
+1
-1
@@ -1,6 +1,6 @@
|
||||
from typing import Optional
|
||||
from pydantic import BaseModel
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.TOOL_PERMISSIONS import TOOL_PERMISSIONS
|
||||
from backend.apps.HaikFix.tools.shared_structs.TOOL_PERMISSIONS import TOOL_PERMISSIONS
|
||||
from typeguard import typechecked
|
||||
|
||||
class Tool(BaseModel):
|
||||
+2
-2
@@ -1,7 +1,7 @@
|
||||
from typing import Optional, List
|
||||
from pydantic import BaseModel
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Tool import Tool
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.TOOL_PERMISSIONS import TOOL_PERMISSIONS
|
||||
from backend.apps.HaikFix.tools.shared_structs.Tool import Tool
|
||||
from backend.apps.HaikFix.tools.shared_structs.TOOL_PERMISSIONS import TOOL_PERMISSIONS
|
||||
from typeguard import typechecked
|
||||
|
||||
class Toolkit(BaseModel):
|
||||
@@ -1,35 +0,0 @@
|
||||
# BUILTIN_TOOLKIT.py
|
||||
|
||||
from typing import Dict
|
||||
from typeguard import typechecked
|
||||
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
|
||||
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
|
||||
|
||||
@typechecked
|
||||
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),
|
||||
make_browser_toolkit(parent),
|
||||
FILESYSTEM_TOOLKIT,
|
||||
INTERACTION_TOOLKIT,
|
||||
PLANNING_TOOLKIT,
|
||||
SCHEDULING_TOOLKIT,
|
||||
SEARCH_TOOLKIT,
|
||||
SYSTEM_TOOLKIT,
|
||||
],
|
||||
)
|
||||
-18
@@ -1,18 +0,0 @@
|
||||
from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_actions_toolkit.make_browser_actions_toolkit import make_browser_actions_toolkit
|
||||
from backend.apps.agents.HaikFix.tools.make_builtin_toolkit.open_swarm_toolkits.browser_toolkit.make_browser_delegation_toolkit.make_browser_delegation_toolkit import make_browser_delegation_toolkit
|
||||
from backend.apps.agents.HaikFix.tools.shared_structs.Toolkit import Toolkit
|
||||
from backend.apps.agents.HaikFix.Agent.Agent import Agent
|
||||
|
||||
# # NOTE: The minor issue here is that with the current setup, any agent will prly have access to the browser actions toolkit, idk if this bad or good but for now its fine ig
|
||||
|
||||
def make_browser_toolkit(
|
||||
parent: Agent,
|
||||
) -> Toolkit:
|
||||
return Toolkit(
|
||||
name="browser",
|
||||
description="Tools for browser automation",
|
||||
nested_toolkits=[
|
||||
make_browser_delegation_toolkit(parent),
|
||||
make_browser_actions_toolkit(parent),
|
||||
],
|
||||
)
|
||||
Reference in New Issue
Block a user