Class AstraDbEmbeddingStore

java.lang.Object
dev.langchain4j.store.embedding.astradb.AstraDbEmbeddingStore
All Implemented Interfaces:
EmbeddingStore<TextSegment>
public class AstraDbEmbeddingStore extends Object implements EmbeddingStore<TextSegment>
Implementation of EmbeddingStore using AstraDB.
See Also:

  • Field Details

  • Constructor Details

    • AstraDbEmbeddingStore

      public AstraDbEmbeddingStore(@NonNull com.dtsx.astra.sdk.AstraDBCollection client)
      Initialization of the store with an EXISTING collection.
      Parameters:
      client - astra db collection client

    • AstraDbEmbeddingStore

      public AstraDbEmbeddingStore(@NonNull com.dtsx.astra.sdk.AstraDBCollection client, int itemsPerChunk, int concurrentThreads)
      Initialization of the store with an EXISTING collection.
      Parameters:
      client - astra db collection client
      itemsPerChunk - size of 1 chunk in between 1 and 20

  • Method Details

    • clear

      public void clear()
      Delete all records from the table.

    • add

      public String add(Embedding embedding)
      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 String add(Embedding embedding, TextSegment textSegment)
      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.

    • add

      public void add(String id, Embedding embedding)
      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.

    • addAll

      public List<String> addAll(List<Embedding> embeddings)
      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 void addAll(List<String> ids, List<Embedding> embeddingList, List<TextSegment> textSegmentList)
      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:
      ids - A list of IDs associated with the added embeddings.
      embeddingList - A list of embeddings to be added to the store.
      textSegmentList - A list of original contents that were embedded.

    • search

      public EmbeddingSearchResult<TextSegment> search(EmbeddingSearchRequest request)
      Description copied from interface: EmbeddingStore
      Searches for the most similar (closest in the embedding space) Embeddings.
      All search criteria are defined inside the EmbeddingSearchRequest.
      EmbeddingSearchRequest.filter() can be used to filter by various metadata entries (e.g., user/memory ID). Please note that not all EmbeddingStore implementations support Filtering.
      Specified by:
      search in interface EmbeddingStore<TextSegment>
      Parameters:
      request - A request to search in an EmbeddingStore. Contains all search criteria.
      Returns:
      An EmbeddingSearchResult containing all found Embeddings.

    • findRelevant

      public List<EmbeddingMatch<TextSegment>> findRelevant(Embedding referenceEmbedding, int maxResults, double minScore)

    • findRelevant

      public List<EmbeddingMatch<TextSegment>> findRelevant(Embedding referenceEmbedding, io.stargate.sdk.data.domain.query.Filter metaDatafilter, int maxResults, double minScore)
      Semantic search with metadata filtering.
      Parameters:
      referenceEmbedding - vector
      metaDatafilter - fileter for metadata
      maxResults - limit
      minScore - threshold
      Returns:
      records

    • astraDBCollection

      public com.dtsx.astra.sdk.AstraDBCollection astraDBCollection()

    • itemsPerChunk

      public int itemsPerChunk()

    • concurrentThreads

      public int concurrentThreads()