[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:
haikdc
2026-04-02 15:54:42 -07:00
parent 71845cc0ae
commit df07f82253
36 changed files with 111 additions and 111 deletions
@@ -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")
@@ -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
@@ -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
)
@@ -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,
],
)
@@ -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
@@ -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):
@@ -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,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
@@ -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"
@@ -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):
@@ -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):
@@ -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
@@ -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
@@ -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),
],
)
@@ -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",
@@ -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",
@@ -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",
@@ -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",
@@ -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",
@@ -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",
@@ -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,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):
@@ -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,
],
)
@@ -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),
],
)