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

Merge pull request #594 from iceljc/master

Browse source 
add default collection
This commit is contained in:
iceljc 2024-08-14 16:58:27 -05:00 committed by GitHub
parent e6b54c6ed3 548cd558c0
commit f6eb4bd03d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/Infrastructure/BotSharp.Abstraction/Knowledges/Settings/KnowledgeBaseSettings.cs
View file

@ -1,8 +1,11 @@
using BotSharp.Abstraction.Knowledges.Enums;
namespace BotSharp.Abstraction.Knowledges.Settings; namespace BotSharp.Abstraction.Knowledges.Settings;
public class KnowledgeBaseSettings public class KnowledgeBaseSettings
{ {
public string VectorDb { get; set; } public string VectorDb { get; set; }
public string DefaultCollection { get; set; } = KnowledgeCollectionName.BotSharp;
public KnowledgeModelSetting TextEmbedding { get; set; } public KnowledgeModelSetting TextEmbedding { get; set; }
} }

2
src/Infrastructure/BotSharp.Core/Repository/FileRepository/FileRepository.Agent.cs
View file

@ -298,7 +298,7 @@ namespace BotSharp.Core.Repository
var json = JsonSerializer.Serialize(agent, _options); var json = JsonSerializer.Serialize(agent, _options);
File.WriteAllText(agentFile, json); File.WriteAllText(agentFile, json);
UpdateAgentInstructions(inputAgent.Id, inputAgent.Instruction, agent.ChannelInstructions); UpdateAgentInstructions(inputAgent.Id, inputAgent.Instruction, inputAgent.ChannelInstructions);
UpdateAgentResponses(inputAgent.Id, inputAgent.Responses); UpdateAgentResponses(inputAgent.Id, inputAgent.Responses);
UpdateAgentTemplates(inputAgent.Id, inputAgent.Templates); UpdateAgentTemplates(inputAgent.Id, inputAgent.Templates);
UpdateAgentFunctions(inputAgent.Id, inputAgent.Functions); UpdateAgentFunctions(inputAgent.Id, inputAgent.Functions);

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

@ -22,7 +22,8 @@ public class KnowledgeRetrievalFn : IFunctionCallback
var vector = await embedding.GetVectorAsync(args.Question); var vector = await embedding.GetVectorAsync(args.Question);
var vectorDb = _services.GetServices<IVectorDb>().FirstOrDefault(x => x.Name == _settings.VectorDb); var vectorDb = _services.GetServices<IVectorDb>().FirstOrDefault(x => x.Name == _settings.VectorDb);
var knowledges = await vectorDb.Search(KnowledgeCollectionName.BotSharp, vector, new List<string> { KnowledgePayloadName.Answer }); var collectionName = !string.IsNullOrWhiteSpace(_settings.DefaultCollection) ? _settings.DefaultCollection : KnowledgeCollectionName.BotSharp;
var knowledges = await vectorDb.Search(collectionName, vector, new List<string> { KnowledgePayloadName.Answer });
if (!knowledges.IsNullOrEmpty()) if (!knowledges.IsNullOrEmpty())
{ {

5
src/Plugins/BotSharp.Plugin.KnowledgeBase/Functions/MemorizeKnowledgeFn.cs
View file

@ -26,10 +26,11 @@ public class MemorizeKnowledgeFn : IFunctionCallback
}); });
var vectorDb = _services.GetServices<IVectorDb>().FirstOrDefault(x => x.Name == _settings.VectorDb); var vectorDb = _services.GetServices<IVectorDb>().FirstOrDefault(x => x.Name == _settings.VectorDb);
await vectorDb.CreateCollection(KnowledgeCollectionName.BotSharp, vector[0].Length); var collectionName = !string.IsNullOrWhiteSpace(_settings.DefaultCollection) ? _settings.DefaultCollection : KnowledgeCollectionName.BotSharp;
await vectorDb.CreateCollection(collectionName, vector[0].Length);
var id = Guid.NewGuid().ToString(); var id = Guid.NewGuid().ToString();
var result = await vectorDb.Upsert(KnowledgeCollectionName.BotSharp, id, vector[0], var result = await vectorDb.Upsert(collectionName, id, vector[0],
args.Question, args.Question,
new Dictionary<string, string> new Dictionary<string, string>
{ {

1
src/WebStarter/appsettings.json
View file

@ -259,6 +259,7 @@
"KnowledgeBase": { "KnowledgeBase": {
"VectorDb": "Qdrant", "VectorDb": "Qdrant",
"DefaultCollection": "BotSharp",
"TextEmbedding": { "TextEmbedding": {
"Provider": "openai", "Provider": "openai",
"Model": "text-embedding-3-small" "Model": "text-embedding-3-small"