chore: import upstream snapshot with attribution

2026-07-13 13:39:25 +08:00
commit db620d33df
5151 changed files with 925932 additions and 0 deletions
@@ -0,0 +1,112 @@
+// Copyright (c) Microsoft. All rights reserved.
+
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Runtime.CompilerServices;
+using System.Text.Json;
+using System.Text.Json.Serialization;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.Extensions.AI;
+
+namespace Microsoft.Agents.AI.Workflows.UnitTests;
+
+internal class TestEchoAgent(string? id = null, string? name = null, string? prefix = null) : AIAgent
+{
+    protected override string? IdCore => id;
+    public override string? Name => name ?? base.Name;
+
+    public InMemoryChatHistoryProvider ChatHistoryProvider { get; } = new();
+
+    protected override async ValueTask<AgentSession> DeserializeSessionCoreAsync(JsonElement serializedState, JsonSerializerOptions? jsonSerializerOptions = null, CancellationToken cancellationToken = default)
+    {
+        return serializedState.Deserialize<EchoAgentSession>(jsonSerializerOptions) ?? await this.CreateSessionAsync(cancellationToken);
+    }
+
+    protected override ValueTask<JsonElement> SerializeSessionCoreAsync(AgentSession session, JsonSerializerOptions? jsonSerializerOptions = null, CancellationToken cancellationToken = default)
+    {
+        if (session is not EchoAgentSession typedSession)
+        {
+            throw new InvalidOperationException($"The provided session type '{session.GetType().Name}' is not compatible with this agent. Only sessions of type '{nameof(EchoAgentSession)}' can be serialized by this agent.");
+        }
+
+        return new(JsonSerializer.SerializeToElement(typedSession, jsonSerializerOptions));
+    }
+
+    protected override ValueTask<AgentSession> CreateSessionCoreAsync(CancellationToken cancellationToken = default) =>
+        new(new EchoAgentSession());
+
+    private ChatMessage UpdateSession(ChatMessage message, AgentSession? session = null)
+    {
+        this.ChatHistoryProvider.GetMessages(session).Add(message);
+
+        return message;
+    }
+
+    private IEnumerable<ChatMessage> EchoMessages(IEnumerable<ChatMessage> messages, AgentSession? session = null, AgentRunOptions? options = null)
+    {
+        foreach (ChatMessage message in messages)
+        {
+            this.UpdateSession(message, session);
+        }
+
+        IEnumerable<ChatMessage> echoMessages
+            = from message in messages
+              where message.Role == ChatRole.User &&
+                    !string.IsNullOrEmpty(message.Text)
+              select
+                    this.UpdateSession(new ChatMessage(ChatRole.Assistant, $"{prefix}{message.Text}")
+                    {
+                        AuthorName = this.Name ?? this.Id,
+                        CreatedAt = DateTimeOffset.Now,
+                        MessageId = Guid.NewGuid().ToString("N")
+                    }, session);
+
+        return echoMessages.Concat(this.GetEpilogueMessages(options).Select(m => this.UpdateSession(m, session)));
+    }
+
+    protected virtual IEnumerable<ChatMessage> GetEpilogueMessages(AgentRunOptions? options = null)
+    {
+        return [];
+    }
+
+    protected override Task<AgentResponse> RunCoreAsync(IEnumerable<ChatMessage> messages, AgentSession? session = null, AgentRunOptions? options = null, CancellationToken cancellationToken = default)
+    {
+        AgentResponse result =
+            new(this.EchoMessages(messages, session, options).ToList())
+            {
+                AgentId = this.Id,
+                CreatedAt = DateTimeOffset.Now,
+                ResponseId = Guid.NewGuid().ToString("N"),
+            };
+
+        return Task.FromResult(result);
+    }
+
+    protected override async IAsyncEnumerable<AgentResponseUpdate> RunCoreStreamingAsync(IEnumerable<ChatMessage> messages, AgentSession? session = null, AgentRunOptions? options = null, [EnumeratorCancellation] CancellationToken cancellationToken = default)
+    {
+        string responseId = Guid.NewGuid().ToString("N");
+
+        foreach (ChatMessage message in this.EchoMessages(messages, session, options).ToList())
+        {
+            yield return
+                new(message.Role, message.Contents)
+                {
+                    AgentId = this.Id,
+                    AuthorName = message.AuthorName,
+                    ResponseId = responseId,
+                    MessageId = message.MessageId,
+                    CreatedAt = message.CreatedAt
+                };
+        }
+    }
+
+    private sealed class EchoAgentSession : AgentSession
+    {
+        internal EchoAgentSession() { }
+
+        [JsonConstructor]
+        internal EchoAgentSession(AgentSessionStateBag stateBag) : base(stateBag) { }
+    }
+}