public class dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore extends java.lang.Object implements dev.langchain4j.store.embedding.EmbeddingStore
{
private static final org.slf4j.Logger log;
private final org.infinispan.client.hotrod.RemoteCache remoteCache;
private final dev.langchain4j.store.embedding.infinispan.LangChainItemMarshaller itemMarshaller;
private static final java.lang.String DEFAULT_CACHE_CONFIG;
private static final java.lang.String PROTO;
public void <init>(org.infinispan.client.hotrod.configuration.ConfigurationBuilder, java.lang.String, java.lang.Integer)
{
org.infinispan.client.hotrod.RemoteCache v, v;
java.lang.Integer v;
org.infinispan.client.hotrod.RemoteCacheManager v;
java.lang.String v, v, v, v, v, v, v, v, v;
dev.langchain4j.store.embedding.infinispan.LangChainItemMarshaller v, v, v;
org.infinispan.protostream.FileDescriptorSource v;
org.infinispan.client.hotrod.configuration.Configuration v;
org.infinispan.client.hotrod.configuration.ConfigurationBuilder v;
org.infinispan.client.hotrod.configuration.RemoteCacheConfigurationBuilder v;
dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore v;
org.infinispan.commons.marshall.ProtoStreamMarshaller v;
org.infinispan.protostream.SerializationContext v;
v := @this: dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore;
v := @parameter: org.infinispan.client.hotrod.configuration.ConfigurationBuilder;
v := @parameter: java.lang.String;
v := @parameter: java.lang.Integer;
specialinvoke v.<java.lang.Object: void <init>()>();
staticinvoke <dev.langchain4j.internal.ValidationUtils: java.lang.Object ensureNotNull(java.lang.Object,java.lang.String)>(v, "builder");
staticinvoke <dev.langchain4j.internal.ValidationUtils: java.lang.String ensureNotBlank(java.lang.String,java.lang.String)>(v, "name");
staticinvoke <dev.langchain4j.internal.ValidationUtils: java.lang.Object ensureNotNull(java.lang.Object,java.lang.String)>(v, "dimension");
v = new dev.langchain4j.store.embedding.infinispan.LangChainItemMarshaller;
specialinvoke v.<dev.langchain4j.store.embedding.infinispan.LangChainItemMarshaller: void <init>(java.lang.Integer)>(v);
v.<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: dev.langchain4j.store.embedding.infinispan.LangChainItemMarshaller itemMarshaller> = v;
v = virtualinvoke v.<org.infinispan.client.hotrod.configuration.ConfigurationBuilder: org.infinispan.client.hotrod.configuration.RemoteCacheConfigurationBuilder remoteCache(java.lang.String)>(v);
v = "<distributed-cache name=\"CACHE_NAME\">\n<indexing storage=\"local-heap\">\n<indexed-entities>\n<indexed-entity>LANGCHAINITEM</indexed-entity>\n</indexed-entities>\n</indexing>\n</distributed-cache>";
v = virtualinvoke v.<java.lang.String: java.lang.String replace(java.lang.CharSequence,java.lang.CharSequence)>("CACHE_NAME", v);
v = v.<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: dev.langchain4j.store.embedding.infinispan.LangChainItemMarshaller itemMarshaller>;
v = virtualinvoke v.<dev.langchain4j.store.embedding.infinispan.LangChainItemMarshaller: java.lang.String getTypeName()>();
v = virtualinvoke v.<java.lang.String: java.lang.String replace(java.lang.CharSequence,java.lang.CharSequence)>("LANGCHAINITEM", v);
virtualinvoke v.<org.infinispan.client.hotrod.configuration.RemoteCacheConfigurationBuilder: org.infinispan.client.hotrod.configuration.RemoteCacheConfigurationBuilder configuration(java.lang.String)>(v);
v = new org.infinispan.commons.marshall.ProtoStreamMarshaller;
specialinvoke v.<org.infinispan.commons.marshall.ProtoStreamMarshaller: void <init>()>();
v = virtualinvoke v.<org.infinispan.commons.marshall.ProtoStreamMarshaller: org.infinispan.protostream.SerializationContext getSerializationContext()>();
v = new org.infinispan.protostream.FileDescriptorSource;
specialinvoke v.<org.infinispan.protostream.FileDescriptorSource: void <init>()>();
v = dynamicinvoke "makeConcatWithConstants" <java.lang.String (java.lang.Integer)>(v) <java.lang.invoke.StringConcatFactory: java.lang.invoke.CallSite makeConcatWithConstants(java.lang.invoke.MethodHandles$Lookup,java.lang.String,java.lang.invoke.MethodType,java.lang.String,java.lang.Object[])>("langchain_dimension_\u.proto");
v = "syntax = \"proto2\";\n\n/**\n * @Indexed\n */\nmessage LangChainItemDIMENSION {\n   \n   /**\n    * @Keyword\n    */\n   optional string id = 1;\n   \n   /**\n    * @Vector(dimension=DIMENSION, similarity=COSINE)\n    */\n   repeated float embedding = 2;\n   \n   optional string text = 3;\n   \n   repeated string metadataKeys = 4;\n   \n   repeated string metadataValues = 5;\n}\n";
v = virtualinvoke v.<java.lang.Integer: java.lang.String toString()>();
v = virtualinvoke v.<java.lang.String: java.lang.String replace(java.lang.CharSequence,java.lang.CharSequence)>("DIMENSION", v);
virtualinvoke v.<org.infinispan.protostream.FileDescriptorSource: org.infinispan.protostream.FileDescriptorSource addProtoFile(java.lang.String,java.lang.String)>(v, v);
interfaceinvoke v.<org.infinispan.protostream.SerializationContext: void registerProtoFiles(org.infinispan.protostream.FileDescriptorSource)>(v);
v = v.<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: dev.langchain4j.store.embedding.infinispan.LangChainItemMarshaller itemMarshaller>;
interfaceinvoke v.<org.infinispan.protostream.SerializationContext: void registerMarshaller(org.infinispan.protostream.BaseMarshaller)>(v);
virtualinvoke v.<org.infinispan.client.hotrod.configuration.ConfigurationBuilder: org.infinispan.client.hotrod.configuration.ConfigurationBuilder marshaller(org.infinispan.commons.marshall.Marshaller)>(v);
v = new org.infinispan.client.hotrod.RemoteCacheManager;
v = virtualinvoke v.<org.infinispan.client.hotrod.configuration.ConfigurationBuilder: org.infinispan.client.hotrod.configuration.Configuration build()>();
specialinvoke v.<org.infinispan.client.hotrod.RemoteCacheManager: void <init>(org.infinispan.client.hotrod.configuration.Configuration)>(v);
v = virtualinvoke v.<org.infinispan.client.hotrod.RemoteCacheManager: org.infinispan.client.hotrod.RemoteCache getCache(java.lang.String)>("___protobuf_metadata");
interfaceinvoke v.<org.infinispan.client.hotrod.RemoteCache: java.lang.Object put(java.lang.Object,java.lang.Object)>(v, v);
v = virtualinvoke v.<org.infinispan.client.hotrod.RemoteCacheManager: org.infinispan.client.hotrod.RemoteCache getCache(java.lang.String)>(v);
v.<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: org.infinispan.client.hotrod.RemoteCache remoteCache> = v;
return;
}
public java.lang.String add(dev.langchain4j.data.embedding.Embedding)
{
dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore v;
java.lang.String v;
dev.langchain4j.data.embedding.Embedding v;
v := @this: dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore;
v := @parameter: dev.langchain4j.data.embedding.Embedding;
v = staticinvoke <dev.langchain4j.internal.Utils: java.lang.String randomUUID()>();
virtualinvoke v.<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: 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.infinispan.InfinispanEmbeddingStore v;
java.lang.String v;
dev.langchain4j.data.embedding.Embedding v;
v := @this: dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore;
v := @parameter: java.lang.String;
v := @parameter: dev.langchain4j.data.embedding.Embedding;
specialinvoke v.<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: 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.store.embedding.infinispan.InfinispanEmbeddingStore v;
dev.langchain4j.data.embedding.Embedding v;
java.lang.String v;
dev.langchain4j.data.segment.TextSegment v;
v := @this: dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore;
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.infinispan.InfinispanEmbeddingStore: 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)
{
dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore v;
java.util.List v;
java.util.stream.Stream v, v;
java.lang.Object v;
java.util.stream.Collector v;
java.util.function.Function v;
v := @this: dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore;
v := @parameter: java.util.List;
v = interfaceinvoke v.<java.util.List: java.util.stream.Stream stream()>();
v = staticinvoke <dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore$lambda_addAll_0__1: java.util.function.Function bootstrap$()>();
v = interfaceinvoke v.<java.util.stream.Stream: java.util.stream.Stream map(java.util.function.Function)>(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);
specialinvoke v.<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: 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.function.Function v;
dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore v;
java.util.List v, v;
java.util.stream.Stream v, v;
java.lang.Object v;
java.util.stream.Collector v;
v := @this: dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore;
v := @parameter: java.util.List;
v := @parameter: java.util.List;
v = interfaceinvoke v.<java.util.List: java.util.stream.Stream stream()>();
v = staticinvoke <dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore$lambda_addAll_1__2: java.util.function.Function bootstrap$()>();
v = interfaceinvoke v.<java.util.stream.Stream: java.util.stream.Stream map(java.util.function.Function)>(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);
specialinvoke v.<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: 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)
{
org.infinispan.commons.api.query.Query v, v;
float[] v;
org.infinispan.client.hotrod.RemoteCache v;
java.util.function.Function v;
int v;
java.lang.String v, v, v;
java.util.stream.Collector v;
dev.langchain4j.store.embedding.infinispan.LangChainItemMarshaller v;
double v;
dev.langchain4j.data.embedding.Embedding v;
java.util.function.Predicate v;
dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore v;
java.util.List v;
java.util.stream.Stream v, v, v;
java.lang.Object v;
v := @this: dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore;
v := @parameter: dev.langchain4j.data.embedding.Embedding;
v := @parameter: int;
v := @parameter: double;
v = v.<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: org.infinispan.client.hotrod.RemoteCache remoteCache>;
v = v.<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: dev.langchain4j.store.embedding.infinispan.LangChainItemMarshaller itemMarshaller>;
v = virtualinvoke v.<dev.langchain4j.store.embedding.infinispan.LangChainItemMarshaller: java.lang.String getTypeName()>();
v = virtualinvoke v.<dev.langchain4j.data.embedding.Embedding: float[] vector()>();
v = staticinvoke <java.util.Arrays: java.lang.String toString(float[])>(v);
v = dynamicinvoke "makeConcatWithConstants" <java.lang.String (java.lang.String,java.lang.String)>(v, v) <java.lang.invoke.StringConcatFactory: java.lang.invoke.CallSite makeConcatWithConstants(java.lang.invoke.MethodHandles$Lookup,java.lang.String,java.lang.invoke.MethodType,java.lang.String,java.lang.Object[])>("select i, score(i) from \u i where i.embedding <-> \u0001~3");
v = interfaceinvoke v.<org.infinispan.client.hotrod.RemoteCache: org.infinispan.commons.api.query.Query query(java.lang.String)>(v);
v = interfaceinvoke v.<org.infinispan.commons.api.query.Query: org.infinispan.commons.api.query.Query maxResults(int)>(v);
v = interfaceinvoke v.<org.infinispan.commons.api.query.Query: java.util.List list()>();
v = interfaceinvoke v.<java.util.List: java.util.stream.Stream stream()>();
v = staticinvoke <dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore$lambda_findRelevant_2__3: java.util.function.Function bootstrap$(double)>(v);
v = interfaceinvoke v.<java.util.stream.Stream: java.util.stream.Stream map(java.util.function.Function)>(v);
v = staticinvoke <dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore$nonNull__4: java.util.function.Predicate bootstrap$()>();
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)
{
dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore v;
java.util.List v, v, v;
dev.langchain4j.data.embedding.Embedding v;
java.lang.String v;
dev.langchain4j.data.segment.TextSegment v;
v := @this: dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore;
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.infinispan.InfinispanEmbeddingStore: 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)
{
org.infinispan.client.hotrod.RemoteCache v;
java.util.Map v;
dev.langchain4j.data.document.Metadata v;
dev.langchain4j.store.embedding.infinispan.LangChainInfinispanItem v, v;
boolean v, v, v, v;
java.util.Set v;
dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore v;
java.util.List v, v, v;
float[] v, v;
java.util.HashMap v;
java.util.ArrayList v, v;
int v, v, v, v, v, v, v, v;
java.lang.String v;
org.slf4j.Logger v;
java.util.function.Consumer v;
java.lang.Object v, v, v;
v := @this: dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore;
v := @parameter: java.util.List;
v := @parameter: java.util.List;
v := @parameter: java.util.List;
v = staticinvoke <dev.langchain4j.internal.Utils: boolean isNullOrEmpty(java.util.Collection)>(v);
if v != 0 goto label;
v = staticinvoke <dev.langchain4j.internal.Utils: boolean isNullOrEmpty(java.util.Collection)>(v);
if v == 0 goto label;
label:
v = <dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: org.slf4j.Logger log>;
interfaceinvoke v.<org.slf4j.Logger: void info(java.lang.String)>("do not add empty embeddings to infinispan");
return;
label:
v = interfaceinvoke v.<java.util.List: int size()>();
v = interfaceinvoke v.<java.util.List: int size()>();
if v != v goto label;
v = 1;
goto label;
label:
v = 0;
label:
staticinvoke <dev.langchain4j.internal.ValidationUtils: void ensureTrue(boolean,java.lang.String)>(v, "ids size is not equal to embeddings size");
if v == null goto label;
v = interfaceinvoke v.<java.util.List: int size()>();
v = interfaceinvoke v.<java.util.List: int size()>();
if v != v goto label;
label:
v = 1;
goto label;
label:
v = 0;
label:
staticinvoke <dev.langchain4j.internal.ValidationUtils: void ensureTrue(boolean,java.lang.String)>(v, "embeddings size is not equal to embedded size");
v = interfaceinvoke v.<java.util.List: int size()>();
v = new java.util.HashMap;
specialinvoke v.<java.util.HashMap: void <init>(int)>(v);
v = 0;
label:
if v >= v goto label;
v = interfaceinvoke v.<java.util.List: java.lang.Object get(int)>(v);
v = interfaceinvoke v.<java.util.List: java.lang.Object get(int)>(v);
if v == null goto label;
v = interfaceinvoke v.<java.util.List: java.lang.Object get(int)>(v);
label:
if v == null goto label;
v = virtualinvoke v.<dev.langchain4j.data.segment.TextSegment: dev.langchain4j.data.document.Metadata metadata()>();
v = virtualinvoke v.<dev.langchain4j.data.document.Metadata: java.util.Map asMap()>();
v = new java.util.ArrayList;
v = interfaceinvoke v.<java.util.Map: int size()>();
specialinvoke v.<java.util.ArrayList: void <init>(int)>(v);
v = new java.util.ArrayList;
v = interfaceinvoke v.<java.util.Map: int size()>();
specialinvoke v.<java.util.ArrayList: void <init>(int)>(v);
v = interfaceinvoke v.<java.util.Map: java.util.Set entrySet()>();
v = staticinvoke <dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore$lambda_addAllInternal_3__5: java.util.function.Consumer bootstrap$(java.util.List,java.util.List)>(v, v);
interfaceinvoke v.<java.util.Set: void forEach(java.util.function.Consumer)>(v);
v = new dev.langchain4j.store.embedding.infinispan.LangChainInfinispanItem;
v = virtualinvoke v.<dev.langchain4j.data.embedding.Embedding: float[] vector()>();
v = virtualinvoke v.<dev.langchain4j.data.segment.TextSegment: java.lang.String text()>();
specialinvoke v.<dev.langchain4j.store.embedding.infinispan.LangChainInfinispanItem: void <init>(java.lang.String,float[],java.lang.String,java.util.List,java.util.List)>(v, v, v, v, v);
interfaceinvoke v.<java.util.Map: java.lang.Object put(java.lang.Object,java.lang.Object)>(v, v);
goto label;
label:
v = new dev.langchain4j.store.embedding.infinispan.LangChainInfinispanItem;
v = virtualinvoke v.<dev.langchain4j.data.embedding.Embedding: float[] vector()>();
specialinvoke v.<dev.langchain4j.store.embedding.infinispan.LangChainInfinispanItem: void <init>(java.lang.String,float[],java.lang.String,java.util.List,java.util.List)>(v, v, null, null, null);
interfaceinvoke v.<java.util.Map: java.lang.Object put(java.lang.Object,java.lang.Object)>(v, v);
label:
v = v + 1;
goto label;
label:
v = v.<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: org.infinispan.client.hotrod.RemoteCache remoteCache>;
interfaceinvoke v.<org.infinispan.client.hotrod.RemoteCache: void putAll(java.util.Map)>(v);
return;
}
public static dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore$Builder builder()
{
dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore$Builder v;
v = new dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore$Builder;
specialinvoke v.<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore$Builder: void <init>()>();
return v;
}
public org.infinispan.client.hotrod.RemoteCache remoteCache()
{
dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore v;
org.infinispan.client.hotrod.RemoteCache v;
v := @this: dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore;
v = v.<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: org.infinispan.client.hotrod.RemoteCache remoteCache>;
return v;
}
public void clearCache()
{
dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore v;
org.infinispan.client.hotrod.RemoteCache v;
v := @this: dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore;
v = v.<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: org.infinispan.client.hotrod.RemoteCache remoteCache>;
interfaceinvoke v.<org.infinispan.client.hotrod.RemoteCache: void clear()>();
return;
}
static void <clinit>()
{
org.slf4j.Logger v;
v = staticinvoke <org.slf4j.LoggerFactory: org.slf4j.Logger getLogger(java.lang.Class)>(class "Ldev/langchain4j/store/embedding/infinispan/InfinispanEmbeddingStore;");
<dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore: org.slf4j.Logger log> = v;
return;
}
}