public class dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore extends java.lang.Object implements dev.langchain4j.store.embedding.EmbeddingStore
{
static final java.lang.String ID_FIELD_NAME;
static final java.lang.String TEXT_FIELD_NAME;
static final java.lang.String VECTOR_FIELD_NAME;
private final io.milvus.client.MilvusServiceClient milvusClient;
private final java.lang.String collectionName;
private final io.milvus.param.MetricType metricType;
private final io.milvus.common.clientenum.ConsistencyLevelEnum consistencyLevel;
private final boolean retrieveEmbeddingsOnSearch;
public void <init>(java.lang.String, java.lang.Integer, java.lang.String, java.lang.Integer, io.milvus.param.IndexType, io.milvus.param.MetricType, java.lang.String, java.lang.String, java.lang.String, java.lang.String, io.milvus.common.clientenum.ConsistencyLevelEnum, java.lang.Boolean, java.lang.String)
{
java.lang.Integer v, v, v;
boolean v, v;
io.milvus.param.MetricType v, v, v;
dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore v;
io.milvus.param.IndexType v, v;
io.milvus.client.MilvusServiceClient v, v, v, v;
io.milvus.param.ConnectParam$Builder v, v, v, v, v, v;
int v, v;
io.milvus.common.clientenum.ConsistencyLevelEnum v, v;
java.lang.Boolean v, v;
java.lang.String v, v, v, v, v, v, v, v, v, v;
io.milvus.param.ConnectParam v;
java.lang.Object v, v, v, v, v, v, v, v;
v := @this: dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore;
v := @parameter: java.lang.String;
v := @parameter: java.lang.Integer;
v := @parameter: java.lang.String;
v := @parameter: java.lang.Integer;
v := @parameter: io.milvus.param.IndexType;
v := @parameter: io.milvus.param.MetricType;
v := @parameter: java.lang.String;
v := @parameter: java.lang.String;
v := @parameter: java.lang.String;
v := @parameter: java.lang.String;
v := @parameter: io.milvus.common.clientenum.ConsistencyLevelEnum;
v := @parameter: java.lang.Boolean;
v := @parameter: java.lang.String;
specialinvoke v.<java.lang.Object: void <init>()>();
v = staticinvoke <io.milvus.param.ConnectParam: io.milvus.param.ConnectParam$Builder newBuilder()>();
v = staticinvoke <dev.langchain4j.internal.Utils: java.lang.Object getOrDefault(java.lang.Object,java.lang.Object)>(v, "localhost");
v = virtualinvoke v.<io.milvus.param.ConnectParam$Builder: io.milvus.param.ConnectParam$Builder withHost(java.lang.String)>(v);
v = staticinvoke <java.lang.Integer: java.lang.Integer valueOf(int)>(19530);
v = staticinvoke <dev.langchain4j.internal.Utils: java.lang.Object getOrDefault(java.lang.Object,java.lang.Object)>(v, v);
v = virtualinvoke v.<java.lang.Integer: int intValue()>();
v = virtualinvoke v.<io.milvus.param.ConnectParam$Builder: io.milvus.param.ConnectParam$Builder withPort(int)>(v);
v = virtualinvoke v.<io.milvus.param.ConnectParam$Builder: io.milvus.param.ConnectParam$Builder withUri(java.lang.String)>(v);
v = virtualinvoke v.<io.milvus.param.ConnectParam$Builder: io.milvus.param.ConnectParam$Builder withToken(java.lang.String)>(v);
v = virtualinvoke v.<io.milvus.param.ConnectParam$Builder: io.milvus.param.ConnectParam$Builder withAuthorization(java.lang.String,java.lang.String)>(v, v);
if v == null goto label;
virtualinvoke v.<io.milvus.param.ConnectParam$Builder: io.milvus.param.ConnectParam$Builder withDatabaseName(java.lang.String)>(v);
label:
v = new io.milvus.client.MilvusServiceClient;
v = virtualinvoke v.<io.milvus.param.ConnectParam$Builder: io.milvus.param.ConnectParam build()>();
specialinvoke v.<io.milvus.client.MilvusServiceClient: void <init>(io.milvus.param.ConnectParam)>(v);
v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.client.MilvusServiceClient milvusClient> = v;
v = staticinvoke <dev.langchain4j.internal.Utils: java.lang.Object getOrDefault(java.lang.Object,java.lang.Object)>(v, "default");
v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: java.lang.String collectionName> = v;
v = <io.milvus.param.MetricType: io.milvus.param.MetricType COSINE>;
v = staticinvoke <dev.langchain4j.internal.Utils: java.lang.Object getOrDefault(java.lang.Object,java.lang.Object)>(v, v);
v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.param.MetricType metricType> = v;
v = <io.milvus.common.clientenum.ConsistencyLevelEnum: io.milvus.common.clientenum.ConsistencyLevelEnum EVENTUALLY>;
v = staticinvoke <dev.langchain4j.internal.Utils: java.lang.Object getOrDefault(java.lang.Object,java.lang.Object)>(v, v);
v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.common.clientenum.ConsistencyLevelEnum consistencyLevel> = v;
v = staticinvoke <java.lang.Boolean: java.lang.Boolean valueOf(boolean)>(0);
v = staticinvoke <dev.langchain4j.internal.Utils: java.lang.Object getOrDefault(java.lang.Object,java.lang.Object)>(v, v);
v = virtualinvoke v.<java.lang.Boolean: boolean booleanValue()>();
v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: boolean retrieveEmbeddingsOnSearch> = v;
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.client.MilvusServiceClient milvusClient>;
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: java.lang.String collectionName>;
v = staticinvoke <dev.langchain4j.store.embedding.milvus.CollectionOperationsExecutor: boolean hasCollection(io.milvus.client.MilvusServiceClient,java.lang.String)>(v, v);
if v != 0 goto label;
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.client.MilvusServiceClient milvusClient>;
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: java.lang.String collectionName>;
v = staticinvoke <dev.langchain4j.internal.ValidationUtils: java.lang.Object ensureNotNull(java.lang.Object,java.lang.String)>(v, "dimension");
v = virtualinvoke v.<java.lang.Integer: int intValue()>();
staticinvoke <dev.langchain4j.store.embedding.milvus.CollectionOperationsExecutor: void createCollection(io.milvus.client.MilvusServiceClient,java.lang.String,int)>(v, v, v);
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.client.MilvusServiceClient milvusClient>;
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: java.lang.String collectionName>;
v = <io.milvus.param.IndexType: io.milvus.param.IndexType FLAT>;
v = staticinvoke <dev.langchain4j.internal.Utils: java.lang.Object getOrDefault(java.lang.Object,java.lang.Object)>(v, v);
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.param.MetricType metricType>;
staticinvoke <dev.langchain4j.store.embedding.milvus.CollectionOperationsExecutor: void createIndex(io.milvus.client.MilvusServiceClient,java.lang.String,io.milvus.param.IndexType,io.milvus.param.MetricType)>(v, v, v, v);
label:
return;
}
public java.lang.String add(dev.langchain4j.data.embedding.Embedding)
{
dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore v;
java.lang.String v;
dev.langchain4j.data.embedding.Embedding v;
v := @this: dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore;
v := @parameter: dev.langchain4j.data.embedding.Embedding;
v = staticinvoke <dev.langchain4j.internal.Utils: java.lang.String randomUUID()>();
virtualinvoke v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: void add(java.lang.String,dev.langchain4j.data.embedding.Embedding)>(v, v);
return v;
}
public void add(java.lang.String, dev.langchain4j.data.embedding.Embedding)
{
dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore v;
java.lang.String v;
dev.langchain4j.data.embedding.Embedding v;
v := @this: dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore;
v := @parameter: java.lang.String;
v := @parameter: dev.langchain4j.data.embedding.Embedding;
specialinvoke v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: void addInternal(java.lang.String,dev.langchain4j.data.embedding.Embedding,dev.langchain4j.data.segment.TextSegment)>(v, v, null);
return;
}
public java.lang.String add(dev.langchain4j.data.embedding.Embedding, dev.langchain4j.data.segment.TextSegment)
{
dev.langchain4j.data.embedding.Embedding v;
dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore v;
java.lang.String v;
dev.langchain4j.data.segment.TextSegment v;
v := @this: dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore;
v := @parameter: dev.langchain4j.data.embedding.Embedding;
v := @parameter: dev.langchain4j.data.segment.TextSegment;
v = staticinvoke <dev.langchain4j.internal.Utils: java.lang.String randomUUID()>();
specialinvoke v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: void addInternal(java.lang.String,dev.langchain4j.data.embedding.Embedding,dev.langchain4j.data.segment.TextSegment)>(v, v, v);
return v;
}
public java.util.List addAll(java.util.List)
{
java.util.List v, v;
int v;
dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore v;
v := @this: dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore;
v := @parameter: java.util.List;
v = interfaceinvoke v.<java.util.List: int size()>();
v = staticinvoke <dev.langchain4j.store.embedding.milvus.Generator: java.util.List generateRandomIds(int)>(v);
specialinvoke v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: void addAllInternal(java.util.List,java.util.List,java.util.List)>(v, v, null);
return v;
}
public java.util.List addAll(java.util.List, java.util.List)
{
java.util.List v, v, v;
int v;
dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore v;
v := @this: dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore;
v := @parameter: java.util.List;
v := @parameter: java.util.List;
v = interfaceinvoke v.<java.util.List: int size()>();
v = staticinvoke <dev.langchain4j.store.embedding.milvus.Generator: java.util.List generateRandomIds(int)>(v);
specialinvoke v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: void addAllInternal(java.util.List,java.util.List,java.util.List)>(v, v, v);
return v;
}
public java.util.List findRelevant(dev.langchain4j.data.embedding.Embedding, int, double)
{
io.milvus.client.MilvusServiceClient v, v, v;
int v;
io.milvus.common.clientenum.ConsistencyLevelEnum v, v;
io.milvus.response.SearchResultsWrapper v;
java.lang.String v, v, v;
java.util.stream.Collector v;
boolean v;
double v;
dev.langchain4j.data.embedding.Embedding v;
io.milvus.param.MetricType v;
java.util.function.Predicate v;
dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore v;
io.milvus.param.dml.SearchParam v;
java.util.List v, v;
java.util.stream.Stream v, v;
java.lang.Object v;
v := @this: dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore;
v := @parameter: dev.langchain4j.data.embedding.Embedding;
v := @parameter: int;
v := @parameter: double;
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.client.MilvusServiceClient milvusClient>;
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: java.lang.String collectionName>;
staticinvoke <dev.langchain4j.store.embedding.milvus.CollectionOperationsExecutor: void loadCollectionInMemory(io.milvus.client.MilvusServiceClient,java.lang.String)>(v, v);
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: java.lang.String collectionName>;
v = virtualinvoke v.<dev.langchain4j.data.embedding.Embedding: java.util.List vectorAsList()>();
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.param.MetricType metricType>;
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.common.clientenum.ConsistencyLevelEnum consistencyLevel>;
v = staticinvoke <dev.langchain4j.store.embedding.milvus.CollectionRequestBuilder: io.milvus.param.dml.SearchParam buildSearchRequest(java.lang.String,java.util.List,int,io.milvus.param.MetricType,io.milvus.common.clientenum.ConsistencyLevelEnum)>(v, v, v, v, v);
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.client.MilvusServiceClient milvusClient>;
v = staticinvoke <dev.langchain4j.store.embedding.milvus.CollectionOperationsExecutor: io.milvus.response.SearchResultsWrapper search(io.milvus.client.MilvusServiceClient,io.milvus.param.dml.SearchParam)>(v, v);
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.client.MilvusServiceClient milvusClient>;
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: java.lang.String collectionName>;
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.common.clientenum.ConsistencyLevelEnum consistencyLevel>;
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: boolean retrieveEmbeddingsOnSearch>;
v = staticinvoke <dev.langchain4j.store.embedding.milvus.Mapper: java.util.List toEmbeddingMatches(io.milvus.client.MilvusServiceClient,io.milvus.response.SearchResultsWrapper,java.lang.String,io.milvus.common.clientenum.ConsistencyLevelEnum,boolean)>(v, v, v, v, v);
v = interfaceinvoke v.<java.util.List: java.util.stream.Stream stream()>();
v = staticinvoke <dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore$lambda_findRelevant_0__4: java.util.function.Predicate bootstrap$(double)>(v);
v = interfaceinvoke v.<java.util.stream.Stream: java.util.stream.Stream filter(java.util.function.Predicate)>(v);
v = staticinvoke <java.util.stream.Collectors: java.util.stream.Collector toList()>();
v = interfaceinvoke v.<java.util.stream.Stream: java.lang.Object collect(java.util.stream.Collector)>(v);
return v;
}
private void addInternal(java.lang.String, dev.langchain4j.data.embedding.Embedding, dev.langchain4j.data.segment.TextSegment)
{
java.util.List v, v, v;
dev.langchain4j.data.embedding.Embedding v;
dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore v;
java.lang.String v;
dev.langchain4j.data.segment.TextSegment v;
v := @this: dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore;
v := @parameter: java.lang.String;
v := @parameter: dev.langchain4j.data.embedding.Embedding;
v := @parameter: dev.langchain4j.data.segment.TextSegment;
v = staticinvoke <java.util.Collections: java.util.List singletonList(java.lang.Object)>(v);
v = staticinvoke <java.util.Collections: java.util.List singletonList(java.lang.Object)>(v);
if v != null goto label;
v = null;
goto label;
label:
v = staticinvoke <java.util.Collections: java.util.List singletonList(java.lang.Object)>(v);
label:
specialinvoke v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: void addAllInternal(java.util.List,java.util.List,java.util.List)>(v, v, v);
return;
}
private void addAllInternal(java.util.List, java.util.List, java.util.List)
{
io.milvus.client.MilvusServiceClient v, v;
dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore v;
java.util.ArrayList v;
java.util.List v, v, v, v, v;
int v;
io.milvus.param.dml.InsertParam$Field v, v, v;
java.lang.String v, v;
v := @this: dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore;
v := @parameter: java.util.List;
v := @parameter: java.util.List;
v := @parameter: java.util.List;
v = new java.util.ArrayList;
specialinvoke v.<java.util.ArrayList: void <init>()>();
v = new io.milvus.param.dml.InsertParam$Field;
specialinvoke v.<io.milvus.param.dml.InsertParam$Field: void <init>(java.lang.String,java.util.List)>("id", v);
interfaceinvoke v.<java.util.List: boolean add(java.lang.Object)>(v);
v = new io.milvus.param.dml.InsertParam$Field;
v = interfaceinvoke v.<java.util.List: int size()>();
v = staticinvoke <dev.langchain4j.store.embedding.milvus.Mapper: java.util.List toScalars(java.util.List,int)>(v, v);
specialinvoke v.<io.milvus.param.dml.InsertParam$Field: void <init>(java.lang.String,java.util.List)>("text", v);
interfaceinvoke v.<java.util.List: boolean add(java.lang.Object)>(v);
v = new io.milvus.param.dml.InsertParam$Field;
v = staticinvoke <dev.langchain4j.store.embedding.milvus.Mapper: java.util.List toVectors(java.util.List)>(v);
specialinvoke v.<io.milvus.param.dml.InsertParam$Field: void <init>(java.lang.String,java.util.List)>("vector", v);
interfaceinvoke v.<java.util.List: boolean add(java.lang.Object)>(v);
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.client.MilvusServiceClient milvusClient>;
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: java.lang.String collectionName>;
staticinvoke <dev.langchain4j.store.embedding.milvus.CollectionOperationsExecutor: void insert(io.milvus.client.MilvusServiceClient,java.lang.String,java.util.List)>(v, v, v);
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: io.milvus.client.MilvusServiceClient milvusClient>;
v = v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore: java.lang.String collectionName>;
staticinvoke <dev.langchain4j.store.embedding.milvus.CollectionOperationsExecutor: void flush(io.milvus.client.MilvusServiceClient,java.lang.String)>(v, v);
return;
}
public static dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore$Builder builder()
{
dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore$Builder v;
v = new dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore$Builder;
specialinvoke v.<dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore$Builder: void <init>()>();
return v;
}
}