diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/impl/VectorStoreServiceImpl.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/impl/VectorStoreServiceImpl.java
index 584b273a..58a55a2a 100644
--- a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/impl/VectorStoreServiceImpl.java
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/impl/VectorStoreServiceImpl.java
@@ -47,36 +47,40 @@ public class VectorStoreServiceImpl implements VectorStoreService {
     @Override
     @PostConstruct
     public void createSchema(String kid,String modelName) {
-        if(modelName.equals("weaviate")){
-            String protocol = configService.getConfigValue("weaviate", "protocol");
-            String host = configService.getConfigValue("weaviate", "host");
-            String className = configService.getConfigValue("weaviate", "classname");
-            this.embeddingStore = WeaviateEmbeddingStore.builder()
-                    .scheme(protocol)
-                    .host(host)
-                    .objectClass(className+kid)
-                    .scheme(protocol)
-                    .avoidDups(true)
-                    .consistencyLevel("ALL")
-                    .build();
-        }else if(modelName.equals("milvus")){
-            String uri = configService.getConfigValue("milvus", "host");
-            String collection = configService.getConfigValue("milvus", "collection");
-            String dimension = configService.getConfigValue("milvus", "dimension");
-            this.embeddingStore =  MilvusEmbeddingStore.builder()
-                    .uri(uri)
-                    .collectionName(collection+kid)
-                    .dimension(Integer.parseInt(dimension))
-                    .build();
-        }else if(modelName.equals("qdrant")){
-            String host = configService.getConfigValue("qdrant", "host");
-            String port = configService.getConfigValue("qdrant", "port");
-            String collectionName = configService.getConfigValue("qdrant", "collectionName");
-            this.embeddingStore = QdrantEmbeddingStore.builder()
-                            .host(host)
-                            .port(Integer.parseInt(port))
-                            .collectionName(collectionName)
-                            .build();
+        switch (modelName) {
+            case "weaviate" -> {
+                String protocol = configService.getConfigValue("weaviate", "protocol");
+                String host = configService.getConfigValue("weaviate", "host");
+                String className = configService.getConfigValue("weaviate", "classname");
+                this.embeddingStore = WeaviateEmbeddingStore.builder()
+                        .scheme(protocol)
+                        .host(host)
+                        .objectClass(className + kid)
+                        .scheme(protocol)
+                        .avoidDups(true)
+                        .consistencyLevel("ALL")
+                        .build();
+            }
+            case "milvus" -> {
+                String uri = configService.getConfigValue("milvus", "host");
+                String collection = configService.getConfigValue("milvus", "collection");
+                String dimension = configService.getConfigValue("milvus", "dimension");
+                this.embeddingStore = MilvusEmbeddingStore.builder()
+                        .uri(uri)
+                        .collectionName(collection + kid)
+                        .dimension(Integer.parseInt(dimension))
+                        .build();
+            }
+            case "qdrant" -> {
+                String host = configService.getConfigValue("qdrant", "host");
+                String port = configService.getConfigValue("qdrant", "port");
+                String collectionName = configService.getConfigValue("qdrant", "collectionName");
+                this.embeddingStore = QdrantEmbeddingStore.builder()
+                        .host(host)
+                        .port(Integer.parseInt(port))
+                        .collectionName(collectionName)
+                        .build();
+            }
         }
     }