chore: import upstream snapshot with attribution
This commit is contained in:
wehub-resource-sync
@@ -0,0 +1,163 @@
|
||||
// @region[backend-render-operations]
|
||||
// @region[backend-schema-json-load]
|
||||
using System.ClientModel;
|
||||
using System.ComponentModel;
|
||||
using System.Text.Json;
|
||||
using Microsoft.Agents.AI;
|
||||
using Microsoft.Extensions.AI;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using OpenAI;
|
||||
|
||||
/// <summary>
|
||||
/// Factory for the A2UI — Fixed Schema agent.
|
||||
///
|
||||
/// Mirrors the LangGraph `src/agents/a2ui_fixed.py` reference: the frontend
|
||||
/// owns a pre-authored component tree (see
|
||||
/// `src/app/demos/a2ui-fixed-schema/a2ui/definitions.ts` + flight_schema.json)
|
||||
/// and the agent only streams *data* into the data model via a dedicated
|
||||
/// `display_flight` tool that emits an <c>a2ui_operations</c> container.
|
||||
/// The A2UI middleware detects that container in the tool result and
|
||||
/// forwards rendered surfaces to the frontend.
|
||||
/// </summary>
|
||||
public class A2uiFixedSchemaAgent
|
||||
{
|
||||
private const string DefaultOpenAiEndpoint = "https://models.inference.ai.azure.com";
|
||||
private const string CatalogId = "copilotkit://flight-fixed-catalog";
|
||||
private const string SurfaceId = "flight-fixed-schema";
|
||||
|
||||
private readonly OpenAIClient _openAiClient;
|
||||
private readonly ILogger _logger;
|
||||
private readonly JsonSerializerOptions _jsonSerializerOptions;
|
||||
|
||||
public A2uiFixedSchemaAgent(IConfiguration configuration, ILoggerFactory loggerFactory, JsonSerializerOptions jsonSerializerOptions)
|
||||
{
|
||||
ArgumentNullException.ThrowIfNull(configuration);
|
||||
ArgumentNullException.ThrowIfNull(loggerFactory);
|
||||
ArgumentNullException.ThrowIfNull(jsonSerializerOptions);
|
||||
|
||||
_logger = loggerFactory.CreateLogger<A2uiFixedSchemaAgent>();
|
||||
_jsonSerializerOptions = jsonSerializerOptions;
|
||||
|
||||
var githubToken = configuration["GitHubToken"]
|
||||
?? throw new InvalidOperationException(
|
||||
"GitHubToken not found in configuration. " +
|
||||
"Please set it using: dotnet user-secrets set GitHubToken \"<your-token>\" " +
|
||||
"or get it using: gh auth token");
|
||||
|
||||
var endpointEnv = Environment.GetEnvironmentVariable("OPENAI_BASE_URL");
|
||||
var endpoint = endpointEnv ?? DefaultOpenAiEndpoint;
|
||||
|
||||
_openAiClient = new(
|
||||
new ApiKeyCredential(githubToken),
|
||||
AimockHeaderPolicy.CreateOpenAIClientOptions(endpoint));
|
||||
}
|
||||
|
||||
public AIAgent Create()
|
||||
{
|
||||
var chatClient = _openAiClient.GetChatClient("gpt-4o-mini").AsIChatClient();
|
||||
|
||||
return new ChatClientAgent(
|
||||
chatClient,
|
||||
name: "A2uiFixedSchemaAgent",
|
||||
description: @"You help users find flights. When asked about a flight, call
|
||||
`display_flight` with origin, destination, airline, and price.
|
||||
Use short airport codes (e.g. ""SFO"", ""JFK"") for origin/destination and a price
|
||||
string like ""$289"". Keep any chat reply to one short sentence.",
|
||||
tools: [
|
||||
AIFunctionFactory.Create(DisplayFlight, options: new() { Name = "display_flight", SerializerOptions = _jsonSerializerOptions })
|
||||
]);
|
||||
}
|
||||
|
||||
// The fixed-schema flight component tree. .NET doesn't ship a
|
||||
// JSON-loading helper analogous to LangGraph Python's
|
||||
// `a2ui.load_schema(...)`, so the schema is declared inline as a
|
||||
// C# array — equivalent to deserialising a `flight_schema.json`
|
||||
// file at startup. Matches the LangGraph reference at
|
||||
// `src/agents/a2ui_schemas/flight_schema.json`. Frontend renders
|
||||
// this via the registered catalog (`copilotkit://flight-fixed-catalog`).
|
||||
private static readonly object[] FlightSchema = new object[]
|
||||
{
|
||||
new { id = "root", component = "Card", child = "content" },
|
||||
new { id = "content", component = "Column", children = new[] { "title", "route", "meta", "bookButton" } },
|
||||
new { id = "title", component = "Title", text = "Flight Details" },
|
||||
new
|
||||
{
|
||||
id = "route",
|
||||
component = "Row",
|
||||
justify = "spaceBetween",
|
||||
align = "center",
|
||||
children = new[] { "from", "arrow", "to" },
|
||||
},
|
||||
new { id = "from", component = "Airport", code = new { path = "/origin" } },
|
||||
new { id = "arrow", component = "Arrow" },
|
||||
new { id = "to", component = "Airport", code = new { path = "/destination" } },
|
||||
new
|
||||
{
|
||||
id = "meta",
|
||||
component = "Row",
|
||||
justify = "spaceBetween",
|
||||
align = "center",
|
||||
children = new[] { "airline", "price" },
|
||||
},
|
||||
new { id = "airline", component = "AirlineBadge", name = new { path = "/airline" } },
|
||||
new { id = "price", component = "PriceTag", amount = new { path = "/price" } },
|
||||
new
|
||||
{
|
||||
id = "bookButton",
|
||||
component = "Button",
|
||||
variant = "primary",
|
||||
child = "bookButtonLabel",
|
||||
action = new
|
||||
{
|
||||
@event = new
|
||||
{
|
||||
name = "book_flight",
|
||||
context = new
|
||||
{
|
||||
origin = new { path = "/origin" },
|
||||
destination = new { path = "/destination" },
|
||||
airline = new { path = "/airline" },
|
||||
price = new { path = "/price" },
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
new { id = "bookButtonLabel", component = "Text", text = "Book flight" },
|
||||
};
|
||||
// @endregion[backend-schema-json-load]
|
||||
|
||||
[Description("Show a flight card for the given trip. Use short airport codes (e.g. SFO, JFK) for origin/destination and a price string like $289.")]
|
||||
private object DisplayFlight(
|
||||
[Description("Origin airport code (e.g. SFO)")] string origin,
|
||||
[Description("Destination airport code (e.g. JFK)")] string destination,
|
||||
[Description("Airline name")] string airline,
|
||||
[Description("Price string (e.g. $289)")] string price)
|
||||
{
|
||||
_logger.LogInformation("FixedSchema DisplayFlight: {Origin} -> {Destination} on {Airline} at {Price}", origin, destination, airline, price);
|
||||
|
||||
var operations = new object[]
|
||||
{
|
||||
new { version = "v0.9", createSurface = new { surfaceId = SurfaceId, catalogId = CatalogId } },
|
||||
new { version = "v0.9", updateComponents = new { surfaceId = SurfaceId, components = FlightSchema } },
|
||||
new
|
||||
{
|
||||
version = "v0.9",
|
||||
updateDataModel = new
|
||||
{
|
||||
surfaceId = SurfaceId,
|
||||
path = "/",
|
||||
value = new
|
||||
{
|
||||
origin,
|
||||
destination,
|
||||
airline,
|
||||
price,
|
||||
},
|
||||
},
|
||||
},
|
||||
};
|
||||
|
||||
return new { a2ui_operations = operations };
|
||||
}
|
||||
// @endregion[backend-render-operations]
|
||||
}
|
||||
Reference in New Issue
Block a user