public class CassandraEmbeddingStore extends Object
EmbeddingStore using Cassandra AstraDB.EmbeddingStore,
MetadataVectorCassandraTable| Modifier and Type | Class and Description |
|---|---|
static class |
CassandraEmbeddingStore.Builder
Syntax Sugar Builder.
|
| Modifier and Type | Field and Description |
|---|---|
protected com.dtsx.astra.sdk.cassio.MetadataVectorCassandraTable |
embeddingTable
Represents an embedding table in Cassandra, it is a table with a vector column.
|
| Constructor and Description |
|---|
CassandraEmbeddingStore(CassandraEmbeddingConfiguration config)
Build the store from the configuration.
|
| Modifier and Type | Method and Description |
|---|---|
String |
add(@NonNull dev.langchain4j.data.embedding.Embedding embedding)
Add a new embedding to the store.
|
String |
add(@NonNull dev.langchain4j.data.embedding.Embedding embedding,
dev.langchain4j.data.segment.TextSegment textSegment)
Add a new embedding to the store.
|
void |
add(@NonNull String rowId,
@NonNull dev.langchain4j.data.embedding.Embedding embedding)
Add a new embedding to the store.
|
List<String> |
addAll(List<dev.langchain4j.data.embedding.Embedding> embeddingList)
Add multiple embeddings as a single action.
|
List<String> |
addAll(List<dev.langchain4j.data.embedding.Embedding> embeddingList,
List<dev.langchain4j.data.segment.TextSegment> textSegmentList)
Add multiple embeddings as a single action.
|
static CassandraEmbeddingStore.Builder |
builder() |
List<EmbeddingMatch<dev.langchain4j.data.segment.TextSegment>> |
findRelevant(dev.langchain4j.data.embedding.Embedding embedding,
int maxResults,
double minScore)
Search for relevant.
|
List<EmbeddingMatch<dev.langchain4j.data.segment.TextSegment>> |
findRelevant(dev.langchain4j.data.embedding.Embedding embedding,
int maxResults,
double minScore,
Metadata metadata)
Similarity Search ANN based on the embedding.
|
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, waitfindRelevantprotected com.dtsx.astra.sdk.cassio.MetadataVectorCassandraTable embeddingTable
public CassandraEmbeddingStore(CassandraEmbeddingConfiguration config)
config - configurationpublic static CassandraEmbeddingStore.Builder builder()
public String add(@NonNull @NonNull dev.langchain4j.data.embedding.Embedding embedding)
add in interface EmbeddingStore<dev.langchain4j.data.segment.TextSegment>embedding - representation of the list of floatspublic String add(@NonNull @NonNull dev.langchain4j.data.embedding.Embedding embedding, dev.langchain4j.data.segment.TextSegment textSegment)
add in interface EmbeddingStore<dev.langchain4j.data.segment.TextSegment>embedding - representation of the list of floatstextSegment - text content and metadatapublic void add(@NonNull
@NonNull String rowId,
@NonNull
@NonNull dev.langchain4j.data.embedding.Embedding embedding)
add in interface EmbeddingStore<dev.langchain4j.data.segment.TextSegment>rowId - the row idembedding - representation of the list of floatspublic List<String> addAll(List<dev.langchain4j.data.embedding.Embedding> embeddingList)
addAll in interface EmbeddingStore<dev.langchain4j.data.segment.TextSegment>embeddingList - embeddings listpublic List<String> addAll(List<dev.langchain4j.data.embedding.Embedding> embeddingList, List<dev.langchain4j.data.segment.TextSegment> textSegmentList)
addAll in interface EmbeddingStore<dev.langchain4j.data.segment.TextSegment>embeddingList - embeddingstextSegmentList - text segmentspublic List<EmbeddingMatch<dev.langchain4j.data.segment.TextSegment>> findRelevant(dev.langchain4j.data.embedding.Embedding embedding, int maxResults, double minScore)
findRelevant in interface EmbeddingStore<dev.langchain4j.data.segment.TextSegment>embedding - current embeddingsmaxResults - max number of resultminScore - thresholdpublic List<EmbeddingMatch<dev.langchain4j.data.segment.TextSegment>> findRelevant(dev.langchain4j.data.embedding.Embedding embedding, int maxResults, double minScore, Metadata metadata)
embedding - vectormaxResults - max number of resultsminScore - score minScoremetadata - map key-value to build a metadata filterCopyright © 2023. All rights reserved.