Jelajahi Sumber

修改向量保存逻辑

feng 11 bulan lalu
induk
melakukan
f0906ffd31

+ 0 - 3
ruoyi-extend/ruoyi-mcp-server/src/main/java/org/ruoyi/mcpserve/service/impl/FastMCPServiceImpl.java

@@ -23,9 +23,6 @@ public class FastMCPServiceImpl implements IMCPService {
     public List<String> supportedTools() {
         return List.of("queryKnowledgeList", "hospitalActivity", "loginUser");
     }
-
-
-
     @Override
     public List<ToolSchema> getToolSchemas() {
         Map<String, Object> schema = Map.of(

+ 4 - 2
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/impl/VectorStoreServiceImpl.java

@@ -74,8 +74,10 @@ public class VectorStoreServiceImpl implements VectorStoreService {
                 throw new RuntimeException(e);
             }
             Embedding embedding = new Embedding(toFloatArray(vector));
-            TextSegment segment = TextSegment.from(text);
-            embeddingStore.add(embedding, segment);
+            if (text != null && !text.trim().isEmpty()) {
+                TextSegment segment = TextSegment.from(text);
+                embeddingStore.add(embedding, segment);
+            }
         }
     }