mirror of
https://github.com/CopilotKit/CopilotKit.git
synced 2026-09-14 16:26:20 +08:00
Ran Shem Tov
f13f8865a8
Add a zodState() helper that attaches a lazy `~standard.jsonSchema.input` to a zod schema so LangGraph's StateSchema.getJsonSchema() emits the field into the graph's output_schema. Without it, zod v4 fields carry `~standard.validate` + `vendor` only, `isStandardJSONSchema` returns false, and the field is silently dropped from output_schema — which in turn causes the AG-UI LangGraphAgent proxy to filter the value out of STATE_SNAPSHOT events on the wire, so the frontend never sees it even though the underlying thread state has the data. Apply zodState to the middleware's own `copilotkit` state field so it surfaces in output_schema and export it for demos to use on custom state fields (todos, documents, etc.). Uses `z.toJSONSchema` when available (zod v4 subpath) and falls back to an empty object, which is sufficient to make langgraph-api include the key in output_schema.
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
"""
|
|
Fixed-schema A2UI tool: flight search results.
|
|
|
|
Schema is loaded from a JSON file. Only the data changes per invocation.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from typing import TypedDict
|
|
|
|
from copilotkit import a2ui
|
|
from langchain.tools import tool
|
|
|
|
CATALOG_ID = "copilotkit://app-dashboard-catalog"
|
|
SURFACE_ID = "flight-search-results"
|
|
FLIGHT_SCHEMA = a2ui.load_schema(
|
|
Path(__file__).parent / "a2ui" / "schemas" / "flight_schema.json"
|
|
)
|
|
|
|
|
|
class Flight(TypedDict):
|
|
id: str
|
|
airline: str
|
|
airlineLogo: str
|
|
flightNumber: str
|
|
origin: str
|
|
destination: str
|
|
date: str
|
|
departureTime: str
|
|
arrivalTime: str
|
|
duration: str
|
|
status: str
|
|
statusIcon: str
|
|
price: str
|
|
|
|
|
|
@tool
|
|
def search_flights(flights: list[Flight]) -> str:
|
|
"""Search for flights and display the results as rich cards. Return exactly 2 flights.
|
|
|
|
Each flight must have: id, airline (e.g. "United Airlines"),
|
|
airlineLogo (use Google favicon API: https://www.google.com/s2/favicons?domain={airline_domain}&sz=128
|
|
e.g. "https://www.google.com/s2/favicons?domain=united.com&sz=128" for United,
|
|
"https://www.google.com/s2/favicons?domain=delta.com&sz=128" for Delta,
|
|
"https://www.google.com/s2/favicons?domain=aa.com&sz=128" for American,
|
|
"https://www.google.com/s2/favicons?domain=alaskaair.com&sz=128" for Alaska),
|
|
flightNumber, origin, destination,
|
|
date (short readable format like "Tue, Mar 18" — use near-future dates),
|
|
departureTime, arrivalTime,
|
|
duration (e.g. "4h 25m"), status (e.g. "On Time" or "Delayed"),
|
|
statusIcon (colored dot: use "https://placehold.co/12/22c55e/22c55e.png"
|
|
for On Time, "https://placehold.co/12/eab308/eab308.png" for Delayed,
|
|
"https://placehold.co/12/ef4444/ef4444.png" for Cancelled),
|
|
and price (e.g. "$289").
|
|
"""
|
|
return a2ui.render(
|
|
operations=[
|
|
a2ui.create_surface(SURFACE_ID, catalog_id=CATALOG_ID),
|
|
a2ui.update_components(SURFACE_ID, FLIGHT_SCHEMA),
|
|
a2ui.update_data_model(SURFACE_ID, {"flights": flights}),
|
|
],
|
|
)
|