Class AstraDbEmbeddingStore

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

  • Field Summary

    Fields
    Modifier and Type
    Field
    Description
    static final String
    KEY_ATTRIBUTES_BLOB
    Saving the text chunk as an attribut.
    static final String
    KEY_SIMILARITY
    Metadata used for similarity.

  • Constructor Summary

    Constructors
    Constructor
    Description
    AstraDbEmbeddingStore(@NonNull com.dtsx.astra.sdk.AstraDBCollection client)
    Initialization of the store with an EXISTING collection.
    AstraDbEmbeddingStore(@NonNull com.dtsx.astra.sdk.AstraDBCollection client, int itemsPerChunk, int concurrentThreads)
    Initialization of the store with an EXISTING collection.

  • Method Summary

    Modifier and Type
    Method
    Description
    String
    add(dev.langchain4j.data.embedding.Embedding embedding)
    String
    add(dev.langchain4j.data.embedding.Embedding embedding, dev.langchain4j.data.segment.TextSegment textSegment)
    void
    add(String id, dev.langchain4j.data.embedding.Embedding embedding)
    List<String>
    addAll(List<dev.langchain4j.data.embedding.Embedding> embeddings)
    void
    addAll(List<String> ids, List<dev.langchain4j.data.embedding.Embedding> embeddingList, List<dev.langchain4j.data.segment.TextSegment> textSegmentList)
     
    com.dtsx.astra.sdk.AstraDBCollection
    astraDBCollection()
     
    void
    clear()
    Delete all records from the table.
    int
    concurrentThreads()
     
    List<dev.langchain4j.store.embedding.EmbeddingMatch<dev.langchain4j.data.segment.TextSegment>>
    findRelevant(dev.langchain4j.data.embedding.Embedding referenceEmbedding, int maxResults, double minScore)
     
    List<dev.langchain4j.store.embedding.EmbeddingMatch<dev.langchain4j.data.segment.TextSegment>>
    findRelevant(dev.langchain4j.data.embedding.Embedding referenceEmbedding, io.stargate.sdk.data.domain.query.Filter metaDatafilter, int maxResults, double minScore)
    Semantic search with metadata filtering.
    int
    itemsPerChunk()
     
    dev.langchain4j.store.embedding.EmbeddingSearchResult<dev.langchain4j.data.segment.TextSegment>
    search(dev.langchain4j.store.embedding.EmbeddingSearchRequest request)
     

    Methods inherited from class Object

    clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

    Methods inherited from interface dev.langchain4j.store.embedding.EmbeddingStore

    addAll, addListener, addListeners, generateIds, remove, removeAll, removeAll, removeAll

  • 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(dev.langchain4j.data.embedding.Embedding embedding)
      Specified by:
      add in interface dev.langchain4j.store.embedding.EmbeddingStore<dev.langchain4j.data.segment.TextSegment>

    • add

      public String add(dev.langchain4j.data.embedding.Embedding embedding, dev.langchain4j.data.segment.TextSegment textSegment)
      Specified by:
      add in interface dev.langchain4j.store.embedding.EmbeddingStore<dev.langchain4j.data.segment.TextSegment>

    • add

      public void add(String id, dev.langchain4j.data.embedding.Embedding embedding)
      Specified by:
      add in interface dev.langchain4j.store.embedding.EmbeddingStore<dev.langchain4j.data.segment.TextSegment>

    • addAll

      public List<String> addAll(List<dev.langchain4j.data.embedding.Embedding> embeddings)
      Specified by:
      addAll in interface dev.langchain4j.store.embedding.EmbeddingStore<dev.langchain4j.data.segment.TextSegment>

    • addAll

      public void addAll(List<String> ids, List<dev.langchain4j.data.embedding.Embedding> embeddingList, List<dev.langchain4j.data.segment.TextSegment> textSegmentList)
      Specified by:
      addAll in interface dev.langchain4j.store.embedding.EmbeddingStore<dev.langchain4j.data.segment.TextSegment>

    • search

      public dev.langchain4j.store.embedding.EmbeddingSearchResult<dev.langchain4j.data.segment.TextSegment> search(dev.langchain4j.store.embedding.EmbeddingSearchRequest request)
      Specified by:
      search in interface dev.langchain4j.store.embedding.EmbeddingStore<dev.langchain4j.data.segment.TextSegment>

    • findRelevant

      public List<dev.langchain4j.store.embedding.EmbeddingMatch<dev.langchain4j.data.segment.TextSegment>> findRelevant(dev.langchain4j.data.embedding.Embedding referenceEmbedding, int maxResults, double minScore)

    • findRelevant

      public List<dev.langchain4j.store.embedding.EmbeddingMatch<dev.langchain4j.data.segment.TextSegment>> findRelevant(dev.langchain4j.data.embedding.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()