autogen/dotnet/samples/dev-team/DevTeam.Agents/DeveloperLead/DeveloperLead.cs

71 lines
2.2 KiB
C#

// Copyright (c) Microsoft Corporation. All rights reserved.
// DeveloperLead.cs
using DevTeam.Shared;
using Microsoft.AutoGen.Agents;
using Microsoft.AutoGen.Contracts;
using Microsoft.AutoGen.Core;
using Microsoft.SemanticKernel;
using Microsoft.SemanticKernel.Connectors.OpenAI;
using Microsoft.SemanticKernel.Memory;
namespace DevTeam.Agents;
[TopicSubscription("devteam")]
public class DeveloperLead(IAgentWorker worker, Kernel kernel, ISemanticTextMemory memory, [FromKeyedServices("EventTypes")] EventTypes typeRegistry, ILogger<DeveloperLead> logger)
: SKAiAgent<DeveloperLeadState>(worker, memory, kernel, typeRegistry), ILeadDevelopers,
IHandle<DevPlanRequested>,
IHandle<DevPlanChainClosed>
{
public async Task Handle(DevPlanRequested item)
{
var plan = await CreatePlan(item.Ask);
var evt = new DevPlanGenerated
{
Org = item.Org,
Repo = item.Repo,
IssueNumber = item.IssueNumber,
Plan = plan
};
await PublishMessageAsync(evt);
}
public async Task Handle(DevPlanChainClosed item)
{
// TODO: Get plan from state
var lastPlan = ""; // _state.State.History.Last().Message
var evt = new DevPlanCreated
{
Plan = lastPlan
};
await PublishMessageAsync(evt);
}
public async Task<string> CreatePlan(string ask)
{
try
{
var context = new KernelArguments { ["input"] = AppendChatHistory(ask) };
var instruction = "Consider the following architectural guidelines:!waf!";
var enhancedContext = await AddKnowledge(instruction, "waf", context);
var settings = new OpenAIPromptExecutionSettings
{
ResponseFormat = "json_object",
MaxTokens = 4096,
Temperature = 0.8,
TopP = 1
};
return await CallFunction(DevLeadSkills.Plan, enhancedContext, settings);
}
catch (Exception ex)
{
logger.LogError(ex, "Error creating development plan");
return "";
}
}
}
public interface ILeadDevelopers
{
public Task<string> CreatePlan(string ask);
}