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,43 @@
+// Copyright (c) Microsoft. All rights reserved.
+
+using System.Collections.Generic;
+using System.Text.Json;
+using System.Threading.Tasks;
+using Microsoft.Agents.AI.Workflows.Checkpointing;
+
+namespace Microsoft.Agents.AI.Workflows.UnitTests;
+
+internal sealed class InMemoryJsonStore : JsonCheckpointStore
+{
+    private readonly Dictionary<string, SessionCheckpointCache<JsonElement>> _store = [];
+
+    private SessionCheckpointCache<JsonElement> EnsureSessionStore(string sessionId)
+    {
+        if (!this._store.TryGetValue(sessionId, out SessionCheckpointCache<JsonElement>? runStore))
+        {
+            runStore = this._store[sessionId] = new();
+        }
+
+        return runStore;
+    }
+
+    public override ValueTask<CheckpointInfo> CreateCheckpointAsync(string sessionId, JsonElement value, CheckpointInfo? parent = null)
+    {
+        return new(this.EnsureSessionStore(sessionId).Add(sessionId, value));
+    }
+
+    public override ValueTask<JsonElement> RetrieveCheckpointAsync(string sessionId, CheckpointInfo key)
+    {
+        if (!this.EnsureSessionStore(sessionId).TryGet(key, out JsonElement result))
+        {
+            throw new KeyNotFoundException($"Could not retrieve checkpoint with id {key.CheckpointId} for session {sessionId}");
+        }
+
+        return new(result);
+    }
+
+    public override ValueTask<IEnumerable<CheckpointInfo>> RetrieveIndexAsync(string sessionId, CheckpointInfo? withParent = null)
+    {
+        return new(this.EnsureSessionStore(sessionId).Index);
+    }
+}