From 6265f94ef2ebfc1590f1ba551be11f49a7196e00 Mon Sep 17 00:00:00 2001
From: zyxucp <286513187@qq.com>
Date: Tue, 23 Apr 2024 13:32:57 +0800
Subject: [PATCH] =?UTF-8?q?fix=20=E4=BF=AE=E6=94=B9=E6=96=87=E6=A1=A3?=
 =?UTF-8?q?=E5=AF=BC=E5=85=A5=E9=97=AE=E7=AD=94=E5=8D=95=E7=8B=AC=E7=9A=84?=
 =?UTF-8?q?index?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/AntSK.Domain/Domain/Model/Constant/KmsConstantcs.cs | 2 ++
 src/AntSK.Domain/Domain/Service/ChatService.cs          | 8 ++++----
 src/AntSK/Pages/Setting/DelKms/DelKms.razor             | 1 +
 3 files changed, 7 insertions(+), 4 deletions(-)

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 = @"使用<data></data>标记的内容作为你的知识：
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<MemoryFilter>() {
-                        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)
         {