48 lines
1.4 KiB
C#
48 lines
1.4 KiB
C#
|
using BotSharp.Abstraction.Functions;
|
||
|
|
using BotSharp.Core.Infrastructures;
|
||
|
|
|
||
|
|
namespace BotSharp.Plugin.KnowledgeBase.Functions;
|
||
|
|
|
||
|
|
public class MemorizeKnowledgeFn : IFunctionCallback
|
||
|
|
{
|
||
|
|
public string Name => "memorize_knowledge";
|
||
|
|
|
||
|
|
private readonly IServiceProvider _services;
|
||
|
|
private readonly KnowledgeBaseSettings _settings;
|
||
|
|
|
||
|
|
public MemorizeKnowledgeFn(IServiceProvider services, KnowledgeBaseSettings settings)
|
||
|
|
{
|
||
|
|
_services = services;
|
||
|
|
_settings = settings;
|
||
|
|
}
|
||
|
|
|
||
|
|
public async Task<bool> Execute(RoleDialogModel message)
|
||
|
|
{
|
||
|
|
var args = JsonSerializer.Deserialize<ExtractedKnowledge>(message.FunctionArgs ?? "{}");
|
||
|
|
|
||
|
|
var embedding = _services.GetServices<ITextEmbedding>()
|
||
|
|
.First(x => x.GetType().FullName.EndsWith(_settings.TextEmbedding));
|
||
|
|
|
||
|
|
var vector = await embedding.GetVectorsAsync(new List<string>
|
||
|
|
{
|
||
|
|
args.Question
|
||
|
|
});
|
||
|
|
|
||
|
|
var vectorDb = _services.GetRequiredService<IVectorDb>();
|
||
|
|
|
||
|
|
await vectorDb.CreateCollection("lessen", vector[0].Length);
|
||
|
|
|
||
|
|
var id = Utilities.HashTextMd5(args.Question);
|
||
|
|
var result = await vectorDb.Upsert("lessen", id, vector[0],
|
||
|
|
args.Question,
|
||
|
|
new Dictionary<string, string>
|
||
|
|
{
|
||
|
|
{ "answer", args.Answer }
|
||
|
|
});
|
||
|
|
|
||
|
|
message.Content = $"Save result: {(result ? "success" : "failed")}";
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
}
|