38 lines
1.1 KiB
C#
38 lines
1.1 KiB
C#
using BotSharp.Abstraction.Functions;
|
|
using BotSharp.Abstraction.Routing.Models;
|
|
using BotSharp.Plugin.KnowledgeBase.LlmContexts;
|
|
|
|
namespace BotSharp.Plugin.KnowledgeBase.Functions;
|
|
|
|
public class SearchKnowledgesFn : IFunctionCallback
|
|
{
|
|
public string Name => "search_knowledges";
|
|
private readonly IServiceProvider _services;
|
|
|
|
public SearchKnowledgesFn(IServiceProvider services)
|
|
{
|
|
_services = services;
|
|
}
|
|
|
|
public async Task<bool> Execute(RoleDialogModel message)
|
|
{
|
|
var args = JsonSerializer.Deserialize<KnowledgeContextIn>(message.FunctionArgs);
|
|
|
|
var knowledgeService = _services.GetRequiredService<IKnowledgeService>();
|
|
var knowledge = await knowledgeService.GetKnowledges(new KnowledgeRetrievalModel
|
|
{
|
|
AgentId = message.CurrentAgentId,
|
|
Question = args.Question
|
|
});
|
|
|
|
if (string.IsNullOrEmpty(knowledge))
|
|
{
|
|
message.Content = "Can't find any relevant data in local knowledge base.";
|
|
var routingCtx = _services.GetRequiredService<RoutingContext>();
|
|
routingCtx.Pop();
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}
|