WEBDIENSTE/openswarm
1
0
Fork 0
mirror of https://github.com/openswarm-ai/openswarm.git synced 2026-09-11 12:17:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

[Haik]: ckpt, agent toolkit done. Onto browser

Browse Source
This commit is contained in:
haikdc
2026-04-02 13:48:25 -07:00
parent fcaa15a0d9
commit 175806bf89
2 changed files with 54 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/agents/HaikFix/tools/BUILTIN_TOOLKIT/BUILTIN_TOOLKIT.py
+20 -16
View File
@@ -1,24 +1,28 @@
from backend.apps.agents.HaikFix.tools.shared_structs.Toolkit import Toolkit
from backend.apps.agents.HaikFix.tools.shared_structs.Tool import Tool
# BUILTIN_TOOLKIT.py
from backend.apps.agents.HaikFix.tools.BUILTIN_TOOLKIT.components.AGENTS_TOOLKIT import AGENTS_TOOLKIT
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
BUILTIN_TOOLKIT = Toolkit(
name="builtin",
description="Builtin tools",
nested_toolkits=[
AGENTS_TOOLKIT,
FILESYSTEM_TOOLKIT,
INTERACTION_TOOLKIT,
PLANNING_TOOLKIT,
SCHEDULING_TOOLKIT,
SEARCH_TOOLKIT,
SYSTEM_TOOLKIT,
]
)
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/BUILTIN_TOOLKIT/components/AGENTS_TOOLKIT.py
+34 -9
View File
@@ -1,11 +1,36 @@
from typing import Dict
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.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 (
make_create_agent_handler, CreateAgentInput,
)
from backend.apps.agents.HaikFix.tools.BUILTIN_TOOLKIT.components.custom_handlers.make_invoke_agent_handler import (
make_invoke_agent_handler, InvokeAgentInput,
)
from backend.apps.agents.HaikFix.Agent.Agent import Agent
AGENTS_TOOLKIT = Toolkit(
name="agents",
description="Tools for creating and invoking agents",
tools=[
Tool(name="CreateAgent", deferred=False, permission="allow"),
Tool(name="InvokeAgent", deferred=False, permission="allow"),
]
)
def make_agents_toolkit(parent: Agent, agent_registry: Dict[str, Agent]) -> Toolkit:
return Toolkit(
name="agents",
description="Tools for creating and invoking agents",
tools=[
SDK_MCP_Tool(
name="CreateAgent",
description="Spawn a sub-agent to handle a complex subtask",
deferred=False,
permission="allow",
server_name="openswarm-agents",
input_schema=CreateAgentInput,
handler=make_create_agent_handler(parent),
),
SDK_MCP_Tool(
name="InvokeAgent",
description="Invoke a copy of an existing agent with a new message, preserving full conversation context",
deferred=False,
permission="allow",
server_name="openswarm-agents",
input_schema=InvokeAgentInput,
handler=make_invoke_agent_handler(agent_registry),
),
],
)