diff --git a/virtual-patient-web/src/main/java/com/supervision/controller/TestController.java b/virtual-patient-web/src/main/java/com/supervision/controller/TestController.java
index c23a5f96..ee4fb5b8 100644
--- a/virtual-patient-web/src/main/java/com/supervision/controller/TestController.java
+++ b/virtual-patient-web/src/main/java/com/supervision/controller/TestController.java
@@ -8,6 +8,7 @@ import cn.hutool.json.JSONObject;
 import cn.hutool.json.JSONUtil;
 import com.supervision.exception.BusinessException;
 import com.supervision.model.AskTemplateQuestionLibrary;
+import com.supervision.model.AskTemplateQuestionSimilarity;
 import com.supervision.model.ConfigPhysicalTool;
 import com.supervision.model.DiagnosisAiRecord;
 import com.supervision.service.AskTemplateQuestionLibraryService;
@@ -88,7 +89,6 @@ public class TestController {
         if (CollUtil.isEmpty(list)){
             return;
         }
-        Map<String, List<String>> similarityQuestionMap = askTemplateQuestionSimilarityService.querySimilarityQuestionMapByLibraryLists(list);
         for (AskTemplateQuestionLibrary askTemplateQuestionLibrary : list) {
             String description = askTemplateQuestionLibrary.getDescription();
             redisVectorStore.add(List.of(new Document(description,
@@ -96,12 +96,12 @@ public class TestController {
                             "libraryQuestionId", askTemplateQuestionLibrary.getId(),
                             "matchQuestionId", askTemplateQuestionLibrary.getId(),
                             "dictId", String.valueOf(askTemplateQuestionLibrary.getDictId())))));
-            List<String> question = similarityQuestionMap.get(askTemplateQuestionLibrary.getId());
-            for (String s : question) {
-                redisVectorStore.add(List.of(new Document(s,
+            List<AskTemplateQuestionSimilarity> similarityList = askTemplateQuestionSimilarityService.lambdaQuery().eq(AskTemplateQuestionSimilarity::getLibraryId, askTemplateQuestionLibrary.getId()).list();
+            for (AskTemplateQuestionSimilarity askTemplateQuestionSimilarity : similarityList) {
+                redisVectorStore.add(List.of(new Document(askTemplateQuestionSimilarity.getSimilarityQuestion(),
                         Map.of("type", "2",
-                                "libraryQuestionId", askTemplateQuestionLibrary.getId(),
-                                "matchQuestionId", askTemplateQuestionLibrary.getId(),
+                                "libraryQuestionId", askTemplateQuestionSimilarity.getLibraryId(),
+                                "matchQuestionId", askTemplateQuestionSimilarity.getId(),
                                 "dictId", String.valueOf(askTemplateQuestionLibrary.getDictId())))));
             }
         }