mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
401a552735
* feat: Add DelegatingAgentSessionStore Add helper for decorator pattern for AgentSessionStore * feat: Add UserIdentityScopedSessionStore Add support for using the ASP.Net Core ambient `ClaimsIdentity` User, along with a user-specified claim type to scope the session store based on authenticated identity. * fix: Harden scope mapping * fix: Add UserIdentityScopeSessionStoreOptions to avoid future breaking changes * Split UserIdentityScopedSessionStore into a separate IsolationKeyProvider and IsolationKeyScopedSessionStore * Add GetService<>() capabilities to interrogate AgentSessionStore delegation chain * Harden default for A2A hosting by using an IsolationKeyScopedAgentSessionStore when no store is available. * Pipe isolation through Hosting helper extension methods * Add comment to samples about adding SessionIsolationKeyProvider * Fix isolation key provider nullability semantics * fix A2A defaults * fixup * remove unneeded keyProvider requirement test * Add trust-model XML docs to AgentSessionStore, InMemoryAgentSessionStore, MapAGUI, A2A entry points Agent-Logs-Url: https://github.com/microsoft/agent-framework/sessions/e466c53a-faad-40a8-8b5f-83cf0dce0b1d Co-authored-by: lokitoth <6936551+lokitoth@users.noreply.github.com> * fix: Switch ClaimsBasedIsolationKeyProvider to be Singleton * matches HttpContextAccessor and related MAF services * release: Ensure new project is in the release filter * fixup: Integraitaon tests --------- Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com> Co-authored-by: lokitoth <6936551+lokitoth@users.noreply.github.com>
68 lines
3.2 KiB
C#
68 lines
3.2 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using System.Collections.Concurrent;
|
|
using System.Text.Json;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace Microsoft.Agents.AI.Hosting;
|
|
|
|
/// <summary>
|
|
/// Provides an in-memory implementation of <see cref="AgentSessionStore"/> for development and testing scenarios.
|
|
/// </summary>
|
|
/// <remarks>
|
|
/// <para>
|
|
/// This implementation stores threads in memory using a concurrent dictionary and is suitable for:
|
|
/// <list type="bullet">
|
|
/// <item><description>Single-instance development scenarios</description></item>
|
|
/// <item><description>Testing and prototyping</description></item>
|
|
/// <item><description>Scenarios where session persistence across restarts is not required</description></item>
|
|
/// </list>
|
|
/// </para>
|
|
/// <para>
|
|
/// <strong>Warning:</strong> All stored threads will be lost when the application restarts.
|
|
/// For production use with multiple instances or persistence across restarts, use a durable storage implementation
|
|
/// such as Redis, SQL Server, or Azure Cosmos DB.
|
|
/// </para>
|
|
/// <para>
|
|
/// <strong>Multi-user warning.</strong> This store keys threads by
|
|
/// <c>(agent.Id, conversationId)</c> only — it has no principal/owner dimension. When
|
|
/// the conversation identifier originates from the wire (for example, an AG-UI
|
|
/// <c>RunAgentInput.ThreadId</c> or an A2A <c>contextId</c>), any caller who knows
|
|
/// or guesses another caller's identifier can resume that other caller's persisted
|
|
/// thread. Multi-user hosts must wrap this store in
|
|
/// <see cref="IsolationKeyScopedAgentSessionStore"/> (typically by calling
|
|
/// <c>UseClaimsBasedSessionIsolation(...)</c> from
|
|
/// <c>Microsoft.Agents.AI.Hosting.AspNetCore</c> or by registering a custom
|
|
/// <see cref="SessionIsolationKeyProvider"/>) so that the conversation namespace is
|
|
/// scoped per principal. See the trust-model remarks on
|
|
/// <see cref="AgentSessionStore"/> for the full background.
|
|
/// </para>
|
|
/// </remarks>
|
|
public sealed class InMemoryAgentSessionStore : AgentSessionStore
|
|
{
|
|
private readonly ConcurrentDictionary<string, JsonElement> _threads = new();
|
|
|
|
/// <inheritdoc/>
|
|
public override async ValueTask SaveSessionAsync(AIAgent agent, string conversationId, AgentSession session, CancellationToken cancellationToken = default)
|
|
{
|
|
var key = GetKey(conversationId, agent.Id);
|
|
this._threads[key] = await agent.SerializeSessionAsync(session, cancellationToken: cancellationToken).ConfigureAwait(false);
|
|
}
|
|
|
|
/// <inheritdoc/>
|
|
public override async ValueTask<AgentSession> GetSessionAsync(AIAgent agent, string conversationId, CancellationToken cancellationToken = default)
|
|
{
|
|
var key = GetKey(conversationId, agent.Id);
|
|
JsonElement? sessionContent = this._threads.TryGetValue(key, out var existingSession) ? existingSession : null;
|
|
|
|
return sessionContent switch
|
|
{
|
|
null => await agent.CreateSessionAsync(cancellationToken).ConfigureAwait(false),
|
|
_ => await agent.DeserializeSessionAsync(sessionContent.Value, cancellationToken: cancellationToken).ConfigureAwait(false),
|
|
};
|
|
}
|
|
|
|
private static string GetKey(string conversationId, string agentId) => $"{agentId}:{conversationId}";
|
|
}
|