feat(showcase/ms-agent-dotnet): start D5 parity sweep

Author: AlemTuzlak

showcase/integrations/ms-agent-dotnet/agent/A2uiFixedSchemaAgent.cs

@@ -15,7 +15,7 @@
 /// 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
-/// `search_flights` tool that emits an <c>a2ui_operations</c> container.
+/// `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>
@@ -60,11 +60,11 @@ public AIAgent Create()
             chatClient,
             name: "A2uiFixedSchemaAgent",
             description: @"You help users find flights. When asked about a flight, call
-`search_flights` with origin, destination, airline, and price.
+`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(SearchFlights, options: new() { Name = "search_flights", SerializerOptions = _jsonSerializerOptions })
+                AIFunctionFactory.Create(DisplayFlight, options: new() { Name = "display_flight", SerializerOptions = _jsonSerializerOptions })
             ]);
     }
 
@@ -127,33 +127,37 @@ public AIAgent Create()
     // @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 string SearchFlights(
+    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 SearchFlights: {Origin} -> {Destination} on {Airline} at {Price}", origin, destination, airline, price);
+        _logger.LogInformation("FixedSchema DisplayFlight: {Origin} -> {Destination} on {Airline} at {Price}", origin, destination, airline, price);
 
         var operations = new object[]
         {
-            new { type = "create_surface", surfaceId = SurfaceId, catalogId = CatalogId },
-            new { type = "update_components", surfaceId = SurfaceId, components = FlightSchema },
+            new { version = "v0.9", createSurface = new { surfaceId = SurfaceId, catalogId = CatalogId } },
+            new { version = "v0.9", updateComponents = new { surfaceId = SurfaceId, components = FlightSchema } },
             new
             {
-                type = "update_data_model",
-                surfaceId = SurfaceId,
-                data = new
+                version = "v0.9",
+                updateDataModel = new
                 {
-                    origin,
-                    destination,
-                    airline,
-                    price,
+                    surfaceId = SurfaceId,
+                    path = "/",
+                    value = new
+                    {
+                        origin,
+                        destination,
+                        airline,
+                        price,
+                    },
                 },
             },
         };
 
-        return JsonSerializer.Serialize(new { a2ui_operations = operations });
+        return new { a2ui_operations = operations };
     }
     // @endregion[backend-render-operations]
 }

showcase/integrations/ms-agent-dotnet/agent/A2uiSecondaryToolCaller.cs

@@ -0,0 +1,121 @@
+using System.Net.Http.Headers;
+using System.Text;
+using System.Text.Json;
+using Microsoft.Extensions.Configuration;
+
+internal static class A2uiSecondaryToolCaller
+{
+    private const string DefaultOpenAiEndpoint = "https://models.inference.ai.azure.com";
+    private const string DesignToolName = "_design_a2ui_surface";
+
+    internal static async Task<string?> GetDesignToolArgumentsAsync(
+        IConfiguration configuration,
+        string systemPrompt,
+        string userContent,
+        CancellationToken cancellationToken)
+    {
+        ArgumentNullException.ThrowIfNull(configuration);
+        ArgumentNullException.ThrowIfNull(systemPrompt);
+        ArgumentNullException.ThrowIfNull(userContent);
+
+        var endpoint = (Environment.GetEnvironmentVariable("OPENAI_BASE_URL") ?? DefaultOpenAiEndpoint).TrimEnd('/');
+        var apiKey = Environment.GetEnvironmentVariable("OPENAI_API_KEY")
+            ?? configuration["OPENAI_API_KEY"]
+            ?? configuration["GitHubToken"]
+            ?? "sk-mock-local";
+
+        using var httpClient = new HttpClient
+        {
+            BaseAddress = new Uri(endpoint + "/"),
+        };
+
+        using var request = new HttpRequestMessage(HttpMethod.Post, "chat/completions");
+        request.Headers.Authorization = new AuthenticationHeaderValue("Bearer", apiKey);
+        foreach (var header in AimockHeaderContext.Get())
+        {
+            request.Headers.TryAddWithoutValidation(header.Key, header.Value);
+        }
+
+        var payload = new
+        {
+            model = "gpt-4.1",
+            messages = new object[]
+            {
+                new { role = "system", content = systemPrompt },
+                new { role = "user", content = userContent },
+            },
+            tools = new object[]
+            {
+                new
+                {
+                    type = "function",
+                    function = new
+                    {
+                        name = DesignToolName,
+                        description = "Render a dynamic A2UI v0.9 surface.",
+                        parameters = new
+                        {
+                            type = "object",
+                            properties = new Dictionary<string, object>
+                            {
+                                ["surfaceId"] = new { type = "string" },
+                                ["catalogId"] = new { type = "string" },
+                                ["components"] = new { type = "array", items = new { type = "object" } },
+                                ["data"] = new { type = "object" },
+                            },
+                            required = new[] { "surfaceId", "catalogId", "components" },
+                        },
+                    },
+                },
+            },
+            tool_choice = new
+            {
+                type = "function",
+                function = new { name = DesignToolName },
+            },
+        };
+
+        request.Content = new StringContent(
+            JsonSerializer.Serialize(payload),
+            Encoding.UTF8,
+            "application/json");
+
+        using var response = await httpClient.SendAsync(request, cancellationToken).ConfigureAwait(false);
+        var body = await response.Content.ReadAsStringAsync(cancellationToken).ConfigureAwait(false);
+        response.EnsureSuccessStatusCode();
+
+        using var document = JsonDocument.Parse(body);
+        if (!document.RootElement.TryGetProperty("choices", out var choices) ||
+            choices.ValueKind != JsonValueKind.Array ||
+            choices.GetArrayLength() == 0)
+        {
+            return null;
+        }
+
+        var message = choices[0].GetProperty("message");
+        if (!message.TryGetProperty("tool_calls", out var toolCalls) ||
+            toolCalls.ValueKind != JsonValueKind.Array ||
+            toolCalls.GetArrayLength() == 0)
+        {
+            return null;
+        }
+
+        var function = toolCalls[0].GetProperty("function");
+        var toolName = function.TryGetProperty("name", out var nameElement)
+            ? nameElement.GetString()
+            : null;
+        if (!string.Equals(toolName, DesignToolName, StringComparison.Ordinal))
+        {
+            return null;
+        }
+
+        if (!function.TryGetProperty("arguments", out var argumentsElement))
+        {
+            return null;
+        }
+
+        return argumentsElement.ValueKind == JsonValueKind.String
+            ? argumentsElement.GetString()
+            : argumentsElement.GetRawText();
+    }
+}