AgentService.cs

using Microsoft.Extensions.AI;
using Microsoft.Extensions.VectorData;

public class AgentService
{
    private readonly IChatClient _chatClient;
    private readonly VectorStoreCollection<string, ClimateFact> _collection;
    private readonly IEmbeddingGenerator<string, Embedding<float>> _embeddingGenerator;
    private readonly IList<AITool> _tools;

    public AgentService(
        IChatClient chatClient,
        VectorStoreCollection<string, ClimateFact> collection,
        IEmbeddingGenerator<string, Embedding<float>> embeddingGenerator)
    {
        _chatClient = chatClient;
        _collection = collection;
        _embeddingGenerator = embeddingGenerator;

        _tools =
        [
            AIFunctionFactory.Create(ClimateSearchToolAsync, "climate_search_tool"),
            AIFunctionFactory.Create(SystemInfoTool, "system_info_tool"),
        ];
    }

    private async Task<string> ClimateSearchToolAsync(string query)
    {
        var embeddings = await _embeddingGenerator.GenerateAsync([query], new EmbeddingGenerationOptions { Dimensions = 384 });
        var embedding = embeddings[0].Vector;
        var results = _collection.SearchAsync(embedding, 1, new VectorSearchOptions<ClimateFact>());
        await foreach (var result in results)
        {
            return result.Record.Text;
        }
        return "No results found.";
    }

    private string SystemInfoTool(string query)
    {
        return "I am a .NET 10 AI Agent using the Microsoft Agent Framework";
    }

    public async Task<AskResponse> AskAsync(string question)
    {
        var response = await _chatClient.GetResponseAsync(question, new ChatOptions { Tools = _tools, ToolMode = ChatToolMode.Auto });
        return new AskResponse(response.Text, "_tool_", "_thought_");
    }
}