| | |
| | | @Override |
| | | @PostConstruct |
| | | public void createSchema(String kid,String modelName) { |
| | | if(modelName.equals("weaviate")){ |
| | | switch (modelName) { |
| | | case "weaviate" -> { |
| | | String protocol = configService.getConfigValue("weaviate", "protocol"); |
| | | String host = configService.getConfigValue("weaviate", "host"); |
| | | String className = configService.getConfigValue("weaviate", "classname"); |
| | |
| | | .avoidDups(true) |
| | | .consistencyLevel("ALL") |
| | | .build(); |
| | | }else if(modelName.equals("milvus")){ |
| | | } |
| | | case "milvus" -> { |
| | | String uri = configService.getConfigValue("milvus", "host"); |
| | | String collection = configService.getConfigValue("milvus", "collection"); |
| | | String dimension = configService.getConfigValue("milvus", "dimension"); |
| | |
| | | .collectionName(collection+kid) |
| | | .dimension(Integer.parseInt(dimension)) |
| | | .build(); |
| | | }else if(modelName.equals("qdrant")){ |
| | | } |
| | | case "qdrant" -> { |
| | | String host = configService.getConfigValue("qdrant", "host"); |
| | | String port = configService.getConfigValue("qdrant", "port"); |
| | | String collectionName = configService.getConfigValue("qdrant", "collectionName"); |
| | |
| | | .build(); |
| | | } |
| | | } |
| | | } |
| | | |
| | | @Override |
| | | public void storeEmbeddings(StoreEmbeddingBo storeEmbeddingBo) { |