SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

not retrieve knowledge from default collection.

Browse source
This commit is contained in:
Haiping Chen 2024-10-03 10:22:31 -05:00
parent cdfb297ef0
commit 2eaffc7c3a
1 changed files with 10 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
src/Plugins/BotSharp.Plugin.KnowledgeBase/Functions/KnowledgeRetrievalFn.cs
View file

@ -1,5 +1,3 @@
using BotSharp.Abstraction.VectorStorage.Extensions;
namespace BotSharp.Plugin.KnowledgeBase.Functions; namespace BotSharp.Plugin.KnowledgeBase.Functions;
public class KnowledgeRetrievalFn : IFunctionCallback public class KnowledgeRetrievalFn : IFunctionCallback
@ -21,23 +19,23 @@ public class KnowledgeRetrievalFn : IFunctionCallback
{ {
var args = JsonSerializer.Deserialize<ExtractedKnowledge>(message.FunctionArgs ?? "{}"); var args = JsonSerializer.Deserialize<ExtractedKnowledge>(message.FunctionArgs ?? "{}");
var collectionName = _settings.Default.CollectionName ?? KnowledgeCollectionName.BotSharp; // Get knowledge from vectordb
var knowledgeService = _services.GetRequiredService<IKnowledgeService>(); var hooks = _services.GetServices<IKnowledgeHook>();
var knowledges = await knowledgeService.SearchVectorKnowledge(args.Question, collectionName, new VectorSearchOptions var knowledges = new List<string>();
foreach (var hook in hooks)
{ {
Fields = new List<string> { KnowledgePayloadName.Text, KnowledgePayloadName.Answer }, var k = await hook.GetRelevantKnowledges(message, args.Question);
Confidence = 0.2f knowledges.AddRange(k);
}); }
knowledges = knowledges.Distinct().ToList();
if (!knowledges.IsNullOrEmpty()) if (!knowledges.IsNullOrEmpty())
{ {
var answers = knowledges.Select(x => x.ToQuestionAnswer()).ToList(); message.Content = string.Join("\r\n\r\n=====\r\n", knowledges);
message.Content = string.Join("\r\n\r\n=====\r\n", answers);
} }
else else
{ {
message.Content = $"I didn't find any useful knowledge related to [{args.Question}]. \r\nCan you tell me the instruction and I'll memorize it."; message.Content = $"I didn't find any useful knowledge related to [{args.Question}].";
message.StopCompletion = true;
} }
return true; return true;