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

using System.ComponentModel;
using Microsoft.Agents;
using Microsoft.Extensions.AI;

namespace Steps;

public sealed class Step02_UsingTools(ITestOutputHelper output) : AgentSample(output)
{
    [Theory]
    [InlineData(ChatClientProviders.OpenAI)]
    [InlineData(ChatClientProviders.AzureOpenAI)]
    public async Task RunningWithTools(ChatClientProviders provider)
    {
        // Get the chat client to use for the agent.
        using var chatClient = base.GetChatClient(provider);

        // Define the agent
        var menuTools = new MenuTools();
        ChatClientAgent agent =
            new(chatClient, new()
            {
                Name = "Host",
                Instructions = "Answer questions about the menu.",
                ChatOptions = new()
                {
                    Tools = [
                        AIFunctionFactory.Create(menuTools.GetMenu),
                        AIFunctionFactory.Create(menuTools.GetSpecials),
                        AIFunctionFactory.Create(menuTools.GetItemPrice)
                    ]
                }
            });

        // Create the chat history thread to capture the agent interaction.
        var thread = agent.GetNewThread();

        // Respond to user input, invoking functions where appropriate.
        await RunAgentAsync("Hello");
        await RunAgentAsync("What is the special soup and its price?");
        await RunAgentAsync("What is the special drink and its price?");
        await RunAgentAsync("Thank you");

        async Task RunAgentAsync(string input)
        {
            this.WriteUserMessage(input);
            var response = await agent.RunAsync(input, thread);
            this.WriteResponseOutput(response);
        }
    }

    [Theory]
    [InlineData(ChatClientProviders.OpenAI)]
    [InlineData(ChatClientProviders.AzureOpenAI)]
    public async Task StreamingRunWithTools(ChatClientProviders provider)
    {
        // Get the chat client to use for the agent.
        using var chatClient = base.GetChatClient(provider);

        // Define the agent
        var menuTools = new MenuTools();
        ChatClientAgent agent =
            new(chatClient, new()
            {
                Name = "Host",
                Instructions = "Answer questions about the menu.",
                ChatOptions = new()
                {
                    Tools = [
                        AIFunctionFactory.Create(menuTools.GetMenu),
                        AIFunctionFactory.Create(menuTools.GetSpecials),
                        AIFunctionFactory.Create(menuTools.GetItemPrice)
                    ]
                }
            });

        // Create the chat history thread to capture the agent interaction.
        var thread = agent.GetNewThread();

        // Respond to user input, invoking functions where appropriate.
        await RunAgentAsync("Hello");
        await RunAgentAsync("What is the special soup and its price?");
        await RunAgentAsync("What is the special drink and its price?");
        await RunAgentAsync("Thank you");

        async Task RunAgentAsync(string input)
        {
            this.WriteUserMessage(input);
            await foreach (var update in agent.RunStreamingAsync(input, thread))
            {
                this.WriteAgentOutput(update);
            }
        }
    }

    private sealed class MenuTools
    {
        [Description("Get the full menu items.")]
        public MenuItem[] GetMenu()
        {
            return s_menuItems;
        }

        [Description("Get the specials from the menu.")]
        public IEnumerable<MenuItem> GetSpecials()
        {
            return s_menuItems.Where(i => i.IsSpecial);
        }

        [Description("Get the price of a menu item.")]
        public float? GetItemPrice([Description("The name of the menu item.")] string menuItem)
        {
            return s_menuItems.FirstOrDefault(i => i.Name.Equals(menuItem, StringComparison.OrdinalIgnoreCase))?.Price;
        }

        private static readonly MenuItem[] s_menuItems = [
            new() { Category = "Soup", Name = "Clam Chowder", Price = 4.95f, IsSpecial = true },
            new() { Category = "Soup", Name = "Tomato Soup", Price = 4.95f, IsSpecial = false },
            new() { Category = "Salad", Name = "Cobb Salad", Price = 9.99f },
            new() { Category = "Salad", Name = "House Salad", Price = 4.95f },
            new() { Category = "Drink", Name = "Chai Tea", Price = 2.95f, IsSpecial = true },
            new() { Category = "Drink", Name = "Soda", Price = 1.95f },
        ];

        public sealed class MenuItem
        {
            public string Category { get; set; } = string.Empty;
            public string Name { get; set; } = string.Empty;
            public float Price { get; set; }
            public bool IsSpecial { get; set; }
        }
    }
}