AgentGroupChat with Vector database support #10429

dehlers1000 · 2025-02-05T23:18:28Z

dehlers1000
Feb 5, 2025

I want to use the rag pattern with the AgentGroupChat but I don't see any way to hook into my ChatCompletionAgent. I tried to find a way to use Semantic Memory with AgentGroupChat but there does not seem to be a way. Also, using Azure OpenAI File Search or the new VectorDB support would be nice. I have these working with OpenAIAssistentAgent but not with a ChatCompletionAgent. For demonstrative purposes, I want the Program Manager to search a vector database for technology standards used in the organization. I did use a plugin function to write the code the Software Engineer agent created. I am thinking a way to accomplish this is to create a plugin that calls a vector database and returns information (.i.e. Semantic Memory, etc.) Here is the code below:

using System.ComponentModel;
using Microsoft.SemanticKernel;
using Microsoft.SemanticKernel.Agents;
using Microsoft.SemanticKernel.Agents.OpenAI;
using Microsoft.SemanticKernel.ChatCompletion;
using Microsoft.SemanticKernel.Connectors.OpenAI;
using Microsoft.SemanticKernel.Agents.Chat;
using Kernel = Microsoft.SemanticKernel.Kernel;
using System.Text.Json;

namespace SemanticKernelCoderAgentConsole
{
internal class Program
{
static async Task Main(string[] args)
{
var appSettings = JsonDocument.Parse(File.ReadAllText("appsettings.json")).RootElement;
string OaiApiKey = appSettings.GetProperty("OaiApiKey").GetString();
string azureOpenAIEndpoint = "https://xyz/";

        var builder = Kernel.CreateBuilder();

        builder.AddAzureOpenAIChatCompletion(
                "gpt-4o",
                azureOpenAIEndpoint,
                apiKey: OaiApiKey,
                "gpt-4-service")
            .Build();

        // Lets given the Persona to Agents

        string ProgramManager = """
            You are a program manager which will take the requirement and create a plan for creating app. Program Manager understands the 
            user requirements and form the detail documents with requirements. 
        """;

        string SoftwareEngineer = """
           You are Software Engineer, and your goal is create web app using HTML and JavaScript by taking into consideration all
           the requirements given by Program Manager. 
        """;

        string ProjectManager = """
            You are a manager which will review software engineer code, and make sure all client requirements are completed. 
            You are the guardian of quality, ensuring the final product meets all specifications and receives the green light for release.
             Double check that the html code will render a ui correctly. If it can be improved, provide feedback to the software engineer
             and do not approve and do not use the word "approve" until the code is satisfactory.
             Show the html code that you are reviewing.
             Once all client requirements are completed, you can approve the request by just responding "perfecto"
        """;

#pragma warning disable SKEXP0110, SKEXP0001 // Rethrow to preserve stack details

        FileWriterPlugin fileWriterPlugin = new();
        builder.Plugins.AddFromObject(fileWriterPlugin);

        Kernel kernel = builder.Build();

        ChatCompletionAgent ProgramManagerAgent =
                   new()
                   {
                       Instructions = ProgramManager,
                       Name = "ProgramManagerAgent",
                       Kernel = kernel
                   };

        ChatCompletionAgent SoftwareEngineerAgent =
                   new()
                   {
                       Instructions = SoftwareEngineer,
                       Name = "SoftwareEngineerAgent",
                       Kernel = kernel,
                       Arguments = new KernelArguments(
                        new OpenAIPromptExecutionSettings()
                        {
                            FunctionChoiceBehavior = FunctionChoiceBehavior.Auto()
                        })
                   };

        ChatCompletionAgent ProjectManagerAgent =
                   new()
                   {
                       Instructions = ProjectManager,
                       Name = "ProjectManagerAgent",
                       Kernel = kernel
                   };

        AgentGroupChat chat =
        new(ProgramManagerAgent, SoftwareEngineerAgent, ProjectManagerAgent)
        {
            ExecutionSettings =
                new()
                {
                    SelectionStrategy = new SequentialSelectionStrategy() { InitialAgent = ProgramManagerAgent },
                    TerminationStrategy =
                        new ApprovalTerminationStrategy()
                        {
                            Agents = [ProjectManagerAgent],
                            MaximumIterations = 12,
                        }
                }
        };

        string input = """

        I want to develop app which will provide me a calculator. Keep it very simple.
        """;

        chat.AddChatMessage(new ChatMessageContent(AuthorRole.User, input));
        Console.WriteLine($"# {AuthorRole.User}: '{input}'");

        await foreach (var content in chat.InvokeAsync())
        {
            Console.WriteLine($"# {content.Role} - {content.AuthorName ?? "*"}: '{content.Content}'");
        }
    }
}

sealed class ApprovalTerminationStrategy : TerminationStrategy
{
    // Terminate when the final message contains the term "approve"
    protected override Task<bool> ShouldAgentTerminateAsync(Agent agent, IReadOnlyList<ChatMessageContent> history, CancellationToken cancellationToken)
        => Task.FromResult(history[history.Count - 1].Content?.Contains("perfecto", StringComparison.OrdinalIgnoreCase) ?? false);
}

}