From 083cd38c29e3cf3b37d0b1b4fa4ca6124ac9c920 Mon Sep 17 00:00:00 2001
From: liu <liujiatong112@163.com>
Date: Thu, 6 Jun 2024 14:53:04 +0800
Subject: [PATCH] bugfix

---
 .../config/VectorSimilarityConfiguration.java |  6 +++---
 .../controller/TestController.java            | 20 ++++++++++---------
 2 files changed, 14 insertions(+), 12 deletions(-)

diff --git a/virtual-patient-common/src/main/java/com/supervision/config/VectorSimilarityConfiguration.java b/virtual-patient-common/src/main/java/com/supervision/config/VectorSimilarityConfiguration.java
index 90186aa2..9030e03c 100644
--- a/virtual-patient-common/src/main/java/com/supervision/config/VectorSimilarityConfiguration.java
+++ b/virtual-patient-common/src/main/java/com/supervision/config/VectorSimilarityConfiguration.java
@@ -26,16 +26,16 @@ public class VectorSimilarityConfiguration {
                 .withURI(redisVectorProperties.getUri())
                 .withPrefix(redisVectorProperties.getPrefix())
                 .withIndexName(redisVectorProperties.getIndexName())
-                // 定义搜索过滤器使用的元数据字段
+                // 定义搜索过滤器使用的元数据字段(!!!!!!!!千万重要,数据类型一定和实际插入的一致,否则会导致查询不到!!!!!!!!)
                 .withMetadataFields(
                         // 问题的ID
                         RedisVectorStore.MetadataField.tag("questionId"),
-                        // 关联字典ID
+                        //关联字典ID
                         RedisVectorStore.MetadataField.tag("dictId"),
                         // 标准问ID
                         RedisVectorStore.MetadataField.tag("standardQuestionId"),
                         // 类型 1标准问 2相似问 3自定义
-                        RedisVectorStore.MetadataField.tag("type"))
+                        RedisVectorStore.MetadataField.numeric("type"))
                 .build();
         return new RedisVectorStore(config, vectorEmbeddingClient);
     }
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 1137071e..ec739a39 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
@@ -82,16 +82,18 @@ public class TestController {
         List<AskTemplateQuestionLibrary> list = askTemplateQuestionLibraryService.list();
         for (AskTemplateQuestionLibrary askTemplateQuestionLibrary : list) {
             String description = askTemplateQuestionLibrary.getDescription();
-            redisVectorStore.add(List.of(new Document(description, Map.of("type", "1",
-                    "standardQuestionId", askTemplateQuestionLibrary.getId(),
-                    "questionId", askTemplateQuestionLibrary.getId(),
-                    "dictId", askTemplateQuestionLibrary.getDictId()))));
+            redisVectorStore.add(List.of(new Document(description,
+                    Map.of("type", "1",
+                            "standardQuestionId", askTemplateQuestionLibrary.getId(),
+                            "questionId", askTemplateQuestionLibrary.getId(),
+                            "dictId", askTemplateQuestionLibrary.getDictId()))));
             List<String> question = askTemplateQuestionLibrary.getQuestion();
             for (String s : question) {
-                redisVectorStore.add(List.of(new Document(s, Map.of("type", "2",
-                        "standardQuestionId", askTemplateQuestionLibrary.getId(),
-                        "questionId", askTemplateQuestionLibrary.getId(),
-                        "dictId", askTemplateQuestionLibrary.getDictId()))));
+                redisVectorStore.add(List.of(new Document(s,
+                        Map.of("type", "2",
+                                "standardQuestionId", askTemplateQuestionLibrary.getId(),
+                                "questionId", askTemplateQuestionLibrary.getId(),
+                                "dictId", askTemplateQuestionLibrary.getDictId()))));
             }
         }
     }
@@ -101,7 +103,7 @@ public class TestController {
     public List<Document> testQuestion(String question) {
         List<Document> documents = redisVectorStore.similaritySearch(SearchRequest
                 .query(question)
-                .withTopK(5).withSimilarityThreshold(0.5));
+                .withTopK(5));
 //        return documents.stream().map(document -> Map.of("content", document.getContent(), "id", document.getId())).collect(Collectors.toList());
         documents.forEach(e -> {
             double v = Double.parseDouble(String.valueOf(e.getMetadata().get("vector_score")));