topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-11 12:07:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix milvus shutdown error (#1307)

Browse Source
This commit is contained in:
二毛
2024-07-01 14:55:40 +08:00
committed by GitHub
parent d2e9d1bf85
commit 7bb0f84bc3
4 changed files with 26 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
common/src/main/java/dev/langchain4j/milvus/spring/MilvusEmbeddingStoreFactory.java
View File

@@ -1,18 +1,19 @@
package dev.langchain4j.milvus.spring;
import dev.langchain4j.data.segment.TextSegment;
import dev.langchain4j.store.embedding.EmbeddingStore;
import dev.langchain4j.store.embedding.EmbeddingStoreFactory;
import dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore;
public class MilvusEmbeddingStoreFactory implements EmbeddingStoreFactory {
private Properties properties;
private final Properties properties;
public MilvusEmbeddingStoreFactory(Properties properties) {
this.properties = properties;
}
@Override
public EmbeddingStore create(String collectionName) {
public EmbeddingStore<TextSegment> create(String collectionName) {
EmbeddingStoreProperties embeddingStore = properties.getEmbeddingStore();
return MilvusEmbeddingStore.builder()
.host(embeddingStore.getHost())

4
common/src/main/java/dev/langchain4j/store/embedding/EmbeddingStoreFactory.java
View File

@@ -1,6 +1,8 @@
package dev.langchain4j.store.embedding;
import dev.langchain4j.data.segment.TextSegment;
public interface EmbeddingStoreFactory {
EmbeddingStore create(String collectionName);
EmbeddingStore<TextSegment> create(String collectionName);
}