Add AgentWorkflowBuilder with Sequential, Concurrent, Handoff (#792)

2026-06-16 21:04:09 +08:00 · 2025-09-19 19:16:07 +00:00
parent aba094b5cf
commit 52790b9f6a
13 changed files with 1068 additions and 53 deletions
@@ -0,0 +1,23 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+  <PropertyGroup>
+    <OutputType>Exe</OutputType>
+    <TargetFramework>net9.0</TargetFramework>
+
+    <Nullable>enable</Nullable>
+    <ImplicitUsings>disable</ImplicitUsings>
+  </PropertyGroup>
+
+  <ItemGroup>
+    <PackageReference Include="Azure.AI.OpenAI" />
+    <PackageReference Include="Azure.Identity" />
+    <PackageReference Include="Microsoft.Extensions.AI.OpenAI" />
+  </ItemGroup>
+
+  <ItemGroup>
+    <ProjectReference Include="..\..\..\..\..\src\Microsoft.Agents.Workflows\Microsoft.Agents.Workflows.csproj" />
+    <ProjectReference Include="..\..\..\..\..\src\Microsoft.Extensions.AI.Agents.AzureAI\Microsoft.Extensions.AI.Agents.AzureAI.csproj" />
+    <ProjectReference Include="..\..\..\..\..\src\Microsoft.Extensions.AI.Agents\Microsoft.Extensions.AI.Agents.csproj" />
+  </ItemGroup>
+
+</Project>
@@ -0,0 +1,104 @@
+// Copyright (c) Microsoft. All rights reserved.
+
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+using Azure.AI.OpenAI;
+using Azure.Identity;
+using Microsoft.Agents.Workflows;
+using Microsoft.Extensions.AI;
+using Microsoft.Extensions.AI.Agents;
+
+namespace WorkflowAgentsInWorkflowsSample;
+
+/// <summary>
+/// This sample introduces the use of AI agents as executors within a workflow,
+/// using <see cref="AgentWorkflowBuilder"/> to compose the agents into one of
+/// several common patterns.
+/// </summary>
+/// <remarks>
+/// Pre-requisites:
+/// - An Azure OpenAI chat completion deployment must be configured.
+/// </remarks>
+public static class Program
+{
+    private static async Task Main()
+    {
+        // Set up the Azure OpenAI client.
+        var endpoint = Environment.GetEnvironmentVariable("AZURE_OPENAI_ENDPOINT") ?? throw new InvalidOperationException("AZURE_OPENAI_ENDPOINT is not set.");
+        var deploymentName = Environment.GetEnvironmentVariable("AZURE_OPENAI_DEPLOYMENT_NAME") ?? "gpt-4o-mini";
+        var client = new AzureOpenAIClient(new Uri(endpoint), new AzureCliCredential()).GetChatClient(deploymentName).AsIChatClient();
+
+        Console.Write("Choose workflow type ('sequential', 'concurrent', 'handoffs'): ");
+        switch (Console.ReadLine())
+        {
+            case "sequential":
+                await RunWorkflowAsync(
+                    AgentWorkflowBuilder.BuildSequential(from lang in (string[])["French", "Spanish", "English"] select GetTranslationAgent(lang, client)),
+                    [new(ChatRole.User, "Hello, world!")]);
+                break;
+
+            case "concurrent":
+                await RunWorkflowAsync(
+                    AgentWorkflowBuilder.BuildConcurrent(from lang in (string[])["French", "Spanish", "English"] select GetTranslationAgent(lang, client)),
+                    [new(ChatRole.User, "Hello, world!")]);
+                break;
+
+            case "handoffs":
+                ChatClientAgent historyTutor = new(client,
+                    "You provide assistance with historical queries. Explain important events and context clearly. Only respond about history.",
+                    "history_tutor",
+                    "Specialist agent for historical questions");
+                ChatClientAgent mathTutor = new(client,
+                    "You provide help with math problems. Explain your reasoning at each step and include examples. Only respond about math.",
+                    "math_tutor",
+                    "Specialist agent for math questions");
+                ChatClientAgent triageAgent = new(client,
+                    "You determine which agent to use based on the user's homework question. ALWAYS handoff to another agent.",
+                    "triage_agent",
+                    "Routes messages to the appropriate specialist agent");
+                var workflow = AgentWorkflowBuilder.StartHandoffWith(triageAgent)
+                    .WithHandoff(triageAgent, [mathTutor, historyTutor])
+                    .WithHandoff(mathTutor, triageAgent)
+                    .WithHandoff(historyTutor, triageAgent)
+                    .Build();
+
+                List<ChatMessage> messages = [];
+                while (true)
+                {
+                    Console.Write("Q: ");
+                    messages.Add(new(ChatRole.User, Console.ReadLine()!));
+                    messages.AddRange(await RunWorkflowAsync(workflow, messages));
+                }
+
+            default:
+                throw new InvalidOperationException("Invalid workflow type.");
+        }
+
+        static async Task<List<ChatMessage>> RunWorkflowAsync(Workflow<List<ChatMessage>> workflow, List<ChatMessage> messages)
+        {
+            StreamingRun run = await InProcessExecution.StreamAsync(workflow, messages);
+            await run.TrySendMessageAsync(new TurnToken(emitEvents: true));
+            await foreach (WorkflowEvent evt in run.WatchStreamAsync().ConfigureAwait(false))
+            {
+                if (evt is AgentRunUpdateEvent e)
+                {
+                    Console.WriteLine($"{e.ExecutorId}: {e.Data}");
+                }
+                else if (evt is WorkflowCompletedEvent completed)
+                {
+                    return (List<ChatMessage>)completed.Data!;
+                }
+            }
+
+            return [];
+        }
+    }
+
+    /// <summary>Creates a translation agent for the specified target language.</summary>
+    private static ChatClientAgent GetTranslationAgent(string targetLanguage, IChatClient chatClient) =>
+        new(chatClient,
+            $"You are a translation assistant who only responds in {targetLanguage}. Respond to any " +
+            $"input by outputting the name of the input language and then translating the input to {targetLanguage}.");
+}