Package dev.langchain4j.store.embedding.infinispan

Class InfinispanEmbeddingStore

java.lang.Object
dev.langchain4j.store.embedding.infinispan.InfinispanEmbeddingStore
All Implemented Interfaces:
EmbeddingStore<TextSegment>
public class InfinispanEmbeddingStore extends Object implements EmbeddingStore<TextSegment>
Infinispan Embedding Store

  • Constructor Details

    • InfinispanEmbeddingStore

      public InfinispanEmbeddingStore(org.infinispan.client.hotrod.RemoteCacheManager remoteCacheManager, InfinispanStoreConfiguration storeConfiguration)
      Creates an Infinispan embedding store from a RemoteCacheManager Assumes marshalling configuration is already provided by the RemoteCacheManager instance.
      Parameters:
      remoteCacheManager - , the already configured remote cache manager
      storeConfiguration - , the store configuration

    • InfinispanEmbeddingStore

      public InfinispanEmbeddingStore(org.infinispan.client.hotrod.configuration.ConfigurationBuilder builder, InfinispanStoreConfiguration storeConfiguration)
      Creates an instance of InfinispanEmbeddingStore

  • Method Details

    • getRemoteCache

      public org.infinispan.client.hotrod.RemoteCache<String,LangChainInfinispanItem> getRemoteCache()
      Gets the underlying Infinispan remote cache
      Returns:
      RemoteCache

    • add

      public String add(Embedding embedding)
      Description copied from interface: EmbeddingStore
      Adds a given embedding to the store.
      Specified by:
      add in interface EmbeddingStore<TextSegment>
      Parameters:
      embedding - The embedding to be added to the store.
      Returns:
      The auto-generated ID associated with the added embedding.

    • add

      public void add(String id, Embedding embedding)
      Description copied from interface: EmbeddingStore
      Adds a given embedding to the store.
      Specified by:
      add in interface EmbeddingStore<TextSegment>
      Parameters:
      id - The unique identifier for the embedding to be added.
      embedding - The embedding to be added to the store.

    • add

      public String add(Embedding embedding, TextSegment textSegment)
      Description copied from interface: EmbeddingStore
      Adds a given embedding and the corresponding content that has been embedded to the store.
      Specified by:
      add in interface EmbeddingStore<TextSegment>
      Parameters:
      embedding - The embedding to be added to the store.
      textSegment - Original content that was embedded.
      Returns:
      The auto-generated ID associated with the added embedding.

    • addAll

      public List<String> addAll(List<Embedding> embeddings)
      Description copied from interface: EmbeddingStore
      Adds multiple embeddings to the store.
      Specified by:
      addAll in interface EmbeddingStore<TextSegment>
      Parameters:
      embeddings - A list of embeddings to be added to the store.
      Returns:
      A list of auto-generated IDs associated with the added embeddings.

    • addAll

      public List<String> addAll(List<Embedding> embeddings, List<TextSegment> embedded)
      Description copied from interface: EmbeddingStore
      Adds multiple embeddings and their corresponding contents that have been embedded to the store.
      Specified by:
      addAll in interface EmbeddingStore<TextSegment>
      Parameters:
      embeddings - A list of embeddings to be added to the store.
      embedded - A list of original contents that were embedded.
      Returns:
      A list of auto-generated IDs associated with the added embeddings.

    • findRelevant

      public List<EmbeddingMatch<TextSegment>> findRelevant(Embedding referenceEmbedding, int maxResults, double minScore)
      Description copied from interface: EmbeddingStore
      Finds the most relevant (closest in space) embeddings to the provided reference embedding.
      Specified by:
      findRelevant in interface EmbeddingStore<TextSegment>
      Parameters:
      referenceEmbedding - The embedding used as a reference. Returned embeddings should be relevant (closest) to this one.
      maxResults - The maximum number of embeddings to be returned.
      minScore - The minimum relevance score, ranging from 0 to 1 (inclusive). Only embeddings with a score of this value or higher will be returned.
      Returns:
      A list of embedding matches. Each embedding match includes a relevance score (derivative of cosine distance), ranging from 0 (not relevant) to 1 (highly relevant).

    • builder

      public static InfinispanEmbeddingStore.Builder builder()

    • remoteCache

      public org.infinispan.client.hotrod.RemoteCache<String,LangChainInfinispanItem> remoteCache()

    • clearCache

      public void clearCache()