mirror of
https://github.com/CopilotKit/CopilotKit.git
synced 2026-09-14 16:26:20 +08:00
Jordan Ritter
2482317ccc
320 files reformatted. One-time alignment to match the ruff format check added to CI in #4812.
36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
"""
|
|
Agent Spec helpers and server tools.
|
|
|
|
This module provides utilities to:
|
|
- Load the Agent Spec JSON used by the backend
|
|
- Define server-side tool implementations
|
|
- Build an AgentSpecAgent configured with the JSON spec and tools
|
|
|
|
It is used by the FastAPI app in main.py.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from ag_ui_agentspec.agent import AgentSpecAgent
|
|
|
|
|
|
def build_a2ui_chat_agent(runtime: str = "langgraph") -> AgentSpecAgent:
|
|
from a2ui_agentspec_agent import a2ui_demo_tool_registry, a2ui_chat_json
|
|
|
|
return AgentSpecAgent(
|
|
agent_spec_config=a2ui_chat_json,
|
|
runtime=runtime,
|
|
tool_registry=a2ui_demo_tool_registry,
|
|
)
|
|
|
|
|
|
def build_agentspec_agent(runtime: str = "langgraph") -> AgentSpecAgent:
|
|
"""Create an AgentSpecAgent configured from the JSON spec and server tools."""
|
|
from agentspec_agent import get_weather, with_agentspec_agent_json
|
|
|
|
return AgentSpecAgent(
|
|
agent_spec_config=with_agentspec_agent_json,
|
|
runtime=runtime,
|
|
tool_registry={"get_weather": get_weather},
|
|
)
|