70 lines
2.6 KiB
C#
70 lines
2.6 KiB
C#
|
using BotSharp.Abstraction.Functions.Models;
|
||
|
|
using BotSharp.Abstraction.Repositories;
|
||
|
|
using BotSharp.Abstraction.Routing;
|
||
|
|
using BotSharp.Abstraction.Routing.Settings;
|
||
|
|
|
||
|
|
namespace BotSharp.Core.Routing.Handlers;
|
||
|
|
|
||
|
|
public class RetrieveDataFromAgentRoutingHandler : RoutingHandlerBase, IRoutingHandler
|
||
|
|
{
|
||
|
|
public string Name => "retrieve_data_from_agent";
|
||
|
|
|
||
|
|
public string Description => "Retrieve data from appropriate agent.";
|
||
|
|
|
||
|
|
public List<string> Parameters => new List<string>
|
||
|
|
{
|
||
|
|
"1. agent_name: the name of the agent",
|
||
|
|
"2. question: the question you will ask the agent to get the necessary data",
|
||
|
|
"3. reason: why retrieve data",
|
||
|
|
"4. args: required parameters extracted from question and hand over to the next agent. The args should be in JSON format"
|
||
|
|
};
|
||
|
|
|
||
|
|
public bool IsReasoning => true;
|
||
|
|
|
||
|
|
public RetrieveDataFromAgentRoutingHandler(IServiceProvider services, ILogger<RetrieveDataFromAgentRoutingHandler> logger, RoutingSettings settings)
|
||
|
|
: base(services, logger, settings)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
public async Task<RoleDialogModel> Handle(FunctionCallFromLlm inst)
|
||
|
|
{
|
||
|
|
if (string.IsNullOrEmpty(inst.Route.AgentName))
|
||
|
|
{
|
||
|
|
inst = await GetNextInstructionFromReasoner($"What's the next step? your response must have agent name.");
|
||
|
|
}
|
||
|
|
|
||
|
|
// Retrieve information from specific agent
|
||
|
|
var db = _services.GetRequiredService<IBotSharpRepository>();
|
||
|
|
var record = db.Agents.First(x => x.Name.ToLower() == inst.Route.AgentName.ToLower());
|
||
|
|
var response = await InvokeAgent(record.Id, new List<RoleDialogModel>
|
||
|
|
{
|
||
|
|
new RoleDialogModel(AgentRole.User, inst.Question)
|
||
|
|
});
|
||
|
|
|
||
|
|
inst.Answer = response.Content;
|
||
|
|
|
||
|
|
/*_dialogs.Add(new RoleDialogModel(AgentRole.Assistant, inst.Parameters.Question)
|
||
|
|
{
|
||
|
|
CurrentAgentId = record.Id
|
||
|
|
});*/
|
||
|
|
|
||
|
|
_router.Instruction += $"\r\n{AgentRole.Assistant}: {inst.Question}";
|
||
|
|
|
||
|
|
/*_dialogs.Add(new RoleDialogModel(AgentRole.Function, inst.Parameters.Answer)
|
||
|
|
{
|
||
|
|
FunctionName = inst.Function,
|
||
|
|
FunctionArgs = JsonSerializer.Serialize(inst.Parameters.Arguments),
|
||
|
|
ExecutionResult = inst.Parameters.Answer,
|
||
|
|
ExecutionData = response.ExecutionData,
|
||
|
|
CurrentAgentId = record.Id
|
||
|
|
});*/
|
||
|
|
|
||
|
|
_router.Instruction += $"\r\n{AgentRole.Function}: {response.Content}";
|
||
|
|
|
||
|
|
// Got the response from agent, then send to reasoner again to make the decision
|
||
|
|
inst = await GetNextInstructionFromReasoner($"What's the next step based on user's original goal and function result?");
|
||
|
|
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
}
|