diff --git a/src/AntSK.Domain/Domain/Model/Constant/KmsConstantcs.cs b/src/AntSK.Domain/Domain/Model/Constant/KmsConstantcs.cs index 7433403..4f7de00 100644 --- a/src/AntSK.Domain/Domain/Model/Constant/KmsConstantcs.cs +++ b/src/AntSK.Domain/Domain/Model/Constant/KmsConstantcs.cs @@ -10,7 +10,9 @@ namespace AntSK.Domain.Domain.Model.Constant { public const string KmsIdTag = "kmsid"; public const string FileIdTag = "fileid"; + public const string AppIdTag = "appid"; public const string KmsIndex = "kms"; + public const string FileIndex = "kms"; public const string KmsSearchNull="知识库未搜索到相关内容"; public const string KmsPrompt = @"使用标记的内容作为你的知识: diff --git a/src/AntSK.Domain/Domain/Service/ChatService.cs b/src/AntSK.Domain/Domain/Service/ChatService.cs index 981cfe3..7771eb1 100644 --- a/src/AntSK.Domain/Domain/Service/ChatService.cs +++ b/src/AntSK.Domain/Domain/Service/ChatService.cs @@ -99,17 +99,17 @@ namespace AntSK.Domain.Domain.Service if (!status) { var result = await memory.ImportDocumentAsync(new Document(fileId).AddFile(filePath) - .AddTag(KmsConstantcs.KmsIdTag, app.Id) + .AddTag(KmsConstantcs.AppIdTag, app.Id) .AddTag(KmsConstantcs.FileIdTag, fileId) - , index: KmsConstantcs.KmsIndex); + , index: KmsConstantcs.FileIndex); } var filters = new List() { - new MemoryFilter().ByTag(KmsConstantcs.KmsIdTag, app.Id), + new MemoryFilter().ByTag(KmsConstantcs.AppIdTag, app.Id), new MemoryFilter().ByTag(KmsConstantcs.FileIdTag, fileId) }; - var searchResult = await memory.SearchAsync(questions, index: KmsConstantcs.KmsIndex, filters: filters); + var searchResult = await memory.SearchAsync(questions, index: KmsConstantcs.FileIndex, filters: filters); relevantSourceList.AddRange(searchResult.Results.SelectMany(item => item.Partitions.Select(part => new RelevantSource() { SourceName = item.SourceName, diff --git a/src/AntSK/Pages/Setting/DelKms/DelKms.razor b/src/AntSK/Pages/Setting/DelKms/DelKms.razor index 6e002e7..245a76c 100644 --- a/src/AntSK/Pages/Setting/DelKms/DelKms.razor +++ b/src/AntSK/Pages/Setting/DelKms/DelKms.razor @@ -72,6 +72,7 @@ } kmsDetails_Repositories.GetDB().Ado.ExecuteCommand("DROP TABLE IF EXISTS \"km-kms\""); + kmsDetails_Repositories.GetDB().Ado.ExecuteCommand("DROP TABLE IF EXISTS \"km-file\""); var kmsDetails = await kmsDetails_Repositories.GetListAsync(); foreach (var detail in kmsDetails) {