AG-UI 与 A2UI 接入

from langchain_core.messages import AIMessage, HumanMessage
from langgraph.graph import END, START, MessagesState, StateGraph
from ksadk.a2ui import build_a2ui_presentation
from ksadk.a2ui.catalog import catalog_manifest
from ksadk.interaction.langgraph import request_input

SCHEMA = {"type": "object", "properties": {
    "name": {"type": "string", "minLength": 2},
}, "required": ["name"], "additionalProperties": False}


def collect(state):
    sid = "parameters-" + state["messages"][-1].id
    def op(**body):
        return {"version": "v0.9.1", **body}
    presentation = build_a2ui_presentation([
        op(createSurface={"surfaceId": sid,
                          "catalogId": catalog_manifest()["catalog_id"]}),
        op(updateComponents={"surfaceId": sid, "components": [
            {"id": "root", "component": "Column", "children": ["name", "submit"]},
            {"id": "name", "component": "TextField", "label": "Name",
             "value": {"path": "/answer/name"}},
            {"id": "submit", "component": "Button", "child": "submit-label",
             "action": {"event": {"name": "submit",
                                  "context": {"payload": {"path": "/answer"}}}}},
            {"id": "submit-label", "component": "Text", "text": "Submit"},
        ]}),
        op(updateDataModel={"surfaceId": sid, "path": "/", "value": {"answer": {"name": ""}}}),
    ], title="Confirm parameters")
    answer = request_input("Enter a name", SCHEMA, presentation=presentation)
    return {"messages": [AIMessage(content="Received: " + answer["name"])]}


def ksadk_prepare_state(payload, session_context):
    return {"messages": payload.get("messages") or [
        HumanMessage(content=payload.get("input", ""))
    ]}


def ksadk_graph_factory(*, checkpointer):
    graph = StateGraph(MessagesState)
    graph.add_node("collect", collect)
    graph.add_edge(START, "collect")
    graph.add_edge("collect", END)
    return graph.compile(checkpointer=checkpointer)


root_agent = ksadk_graph_factory(checkpointer=None)

name: input-demo
framework: langgraph
artifact_type: Code
entry_point: agent.py
agent_variable: root_agent

KSADK_LANGGRAPH_AUTO_CHECKPOINT=1 agentengine web .

from ksadk.compat.copilotkit_a2ui import build_ksadk_a2ui_tool

# model is your configured LangChain chat model.
generate_ui = build_ksadk_a2ui_tool({
    "model": model,
    "recovery": {"maxAttempts": 3},
})