Class AzureCosmosDbNoSqlEmbeddingStore
java.lang.Object
dev.langchain4j.store.embedding.azure.cosmos.nosql.AzureCosmosDbNoSqlEmbeddingStore
- All Implemented Interfaces:
EmbeddingStore<TextSegment>
Implementation of
EmbeddingStore
that uses Azure Cosmos DB NoSQL API for storing and retrieving embeddings.
This store provides vector search capabilities using Cosmos DB's vector search functionality.
You can read more about vector search using Azure Cosmos DB NoSQL here.
-
Nested Class Summary
Nested ClassesModifier and TypeClassDescriptionstatic class
Builder for creatingAzureCosmosDbNoSqlEmbeddingStore
instances. -
Constructor Summary
ConstructorsConstructorDescriptionAzureCosmosDbNoSqlEmbeddingStore
(com.azure.cosmos.CosmosClient cosmosClient, String databaseName, String containerName, com.azure.cosmos.models.CosmosVectorEmbeddingPolicy cosmosVectorEmbeddingPolicy, List<com.azure.cosmos.models.CosmosVectorIndexSpec> cosmosVectorIndexes, com.azure.cosmos.models.CosmosContainerProperties containerProperties) Creates a new instance ofAzureCosmosDbNoSqlEmbeddingStore
. -
Method Summary
Modifier and TypeMethodDescriptionAdds a given embedding to the store.add
(Embedding embedding, TextSegment textSegment) Adds a given embedding and the corresponding content that has been embedded to the store.void
Adds a given embedding to the store.Adds multiple embeddings to the store.void
Adds multiple embeddings and their corresponding contents that have been embedded to the store.builder()
Creates a new builder forAzureCosmosDbNoSqlEmbeddingStore
.findRelevant
(Embedding referenceEmbedding, int maxResults, double minScore) Finds embeddings relevant to the reference embedding.search
(EmbeddingSearchRequest request) Searches for the most similar (closest in the embedding space)Embedding
s.Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
Methods inherited from interface dev.langchain4j.store.embedding.EmbeddingStore
addAll, generateIds, remove, removeAll, removeAll, removeAll
-
Constructor Details
-
AzureCosmosDbNoSqlEmbeddingStore
public AzureCosmosDbNoSqlEmbeddingStore(com.azure.cosmos.CosmosClient cosmosClient, String databaseName, String containerName, com.azure.cosmos.models.CosmosVectorEmbeddingPolicy cosmosVectorEmbeddingPolicy, List<com.azure.cosmos.models.CosmosVectorIndexSpec> cosmosVectorIndexes, com.azure.cosmos.models.CosmosContainerProperties containerProperties) Creates a new instance ofAzureCosmosDbNoSqlEmbeddingStore
.- Parameters:
cosmosClient
- The Cosmos DB clientdatabaseName
- The name of the databasecontainerName
- The name of the containercosmosVectorEmbeddingPolicy
- The vector embedding policycosmosVectorIndexes
- The vector indexescontainerProperties
- The container properties- Throws:
IllegalArgumentException
- if any of the required parameters is null or empty
-
-
Method Details
-
builder
Creates a new builder forAzureCosmosDbNoSqlEmbeddingStore
.- Returns:
- a new builder instance
-
add
Description copied from interface:EmbeddingStore
Adds a given embedding to the store.- Specified by:
add
in interfaceEmbeddingStore<TextSegment>
- Parameters:
embedding
- The embedding to be added to the store.- Returns:
- The auto-generated ID associated with the added embedding.
-
add
Description copied from interface:EmbeddingStore
Adds a given embedding to the store.- Specified by:
add
in interfaceEmbeddingStore<TextSegment>
- Parameters:
id
- The unique identifier for the embedding to be added.embedding
- The embedding to be added to the store.
-
add
Description copied from interface:EmbeddingStore
Adds a given embedding and the corresponding content that has been embedded to the store.- Specified by:
add
in interfaceEmbeddingStore<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
Description copied from interface:EmbeddingStore
Adds multiple embeddings to the store.- Specified by:
addAll
in interfaceEmbeddingStore<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.
-
search
Description copied from interface:EmbeddingStore
Searches for the most similar (closest in the embedding space)Embedding
s.
All search criteria are defined inside theEmbeddingSearchRequest
.
EmbeddingSearchRequest.filter()
can be used to filter by various metadata entries (e.g., user/memory ID). Please note that not allEmbeddingStore
implementations supportFilter
ing.- Specified by:
search
in interfaceEmbeddingStore<TextSegment>
- Parameters:
request
- A request to search in anEmbeddingStore
. Contains all search criteria.- Returns:
- An
EmbeddingSearchResult
containing all foundEmbedding
s.
-
findRelevant
public List<EmbeddingMatch<TextSegment>> findRelevant(Embedding referenceEmbedding, int maxResults, double minScore) Finds embeddings relevant to the reference embedding.- Parameters:
referenceEmbedding
- The reference embedding to compare againstmaxResults
- The maximum number of results to returnminScore
- The minimum similarity score required- Returns:
- A list of embedding matches ordered by relevance
-
addAll
Description copied from interface:EmbeddingStore
Adds multiple embeddings and their corresponding contents that have been embedded to the store.- Specified by:
addAll
in interfaceEmbeddingStore<TextSegment>
- Parameters:
ids
- A list of IDs associated with the added embeddings.embeddings
- A list of embeddings to be added to the store.embedded
- A list of original contents that were embedded.
-