agent-framework/dotnet/src/Microsoft.Extensions.AI.Agents/ChatCompletion/ChatClientExtensions.cs

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

using System;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;

namespace Microsoft.Extensions.AI.Agents;

internal static class ChatClientExtensions
{
    internal static IChatClient AsAgentInvokingChatClient(this IChatClient chatClient)
    {
        var chatBuilder = chatClient.AsBuilder();

        if (chatClient is not AgentInvokingChatClient agentInvokingChatClient)
        {
            chatBuilder.UseAgentInvocation();
        }

        if (chatClient.GetService<NewFunctionInvokingChatClient>() is null)
        {
            chatBuilder.Use((IChatClient innerClient, IServiceProvider services) =>
            {
                var loggerFactory = services.GetService<ILoggerFactory>();

                return new NewFunctionInvokingChatClient(innerClient, loggerFactory, services);
            });
        }

        return chatBuilder.Build();
    }
}