agent-framework/dotnet/tests/Microsoft.Agents.Workflows.Declarative.IntegrationTests/Framework/AgentFixture.cs

// Copyright (c) Microsoft. All rights reserved.

using System;
using System.Collections.Immutable;
using System.Threading;
using System.Threading.Tasks;
using Shared.IntegrationTests;

namespace Microsoft.Agents.Workflows.Declarative.IntegrationTests.Framework;

public sealed class AgentFixture : IDisposable
{
    private static ImmutableDictionary<string, string?>? s_agentMap;

    internal async Task<ImmutableDictionary<string, string?>> GetAgentsAsync(AzureAIConfiguration config, CancellationToken cancellationToken = default)
    {
        s_agentMap ??= await AgentFactory.CreateAsync("Agents", config, cancellationToken);

        return s_agentMap;
    }

    public void Dispose()
    {
        GC.SuppressFinalize(this);
    }
}