Package org.springframework.ai.vectorstore.cosmosdb

Class CosmosDBVectorStore

java.lang.Object

org.springframework.ai.vectorstore.observation.AbstractObservationVectorStore

org.springframework.ai.vectorstore.cosmosdb.CosmosDBVectorStore

All Implemented Interfaces:

AutoCloseable, Consumer<List<Document>>, DocumentWriter, VectorStore, VectorStoreRetriever

public class CosmosDBVectorStore
extends AbstractObservationVectorStore
implements AutoCloseable

Cosmos DB implementation.

Since:

1.0.0

Author:

Theo van Kraay, Soby Chacko, Thomas Vitale

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	CosmosDBVectorStore.Builder	Builder class for creating CosmosDBVectorStore instances.

Field Summary

Fields inherited from class org.springframework.ai.vectorstore.observation.AbstractObservationVectorStore

batchingStrategy, embeddingModel

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	CosmosDBVectorStore(CosmosDBVectorStore.Builder builder)	Protected constructor that accepts a builder instance.

Method Summary

Modifier and Type	Method	Description
static CosmosDBVectorStore.Builder	builder(com.azure.cosmos.CosmosAsyncClient cosmosClient, EmbeddingModel embeddingModel)
void	close()
VectorStoreObservationContext.Builder	createObservationContextBuilder(String operationName)	Create a new VectorStoreObservationContext.Builder instance.
void	doAdd(List<Document> documents)	Perform the actual add operation.
void	doDelete(List<String> idList)	Perform the actual delete operation.
List<Document>	doSimilaritySearch(SearchRequest request)	Perform the actual similarity search operation.
<T> Optional<T>	getNativeClient()	Returns the native client if available in this vector store implementation.
List<Document>	similaritySearch(String query)	Retrieves documents by query embedding similarity using the default SearchRequest's search criteria.

Methods inherited from class org.springframework.ai.vectorstore.observation.AbstractObservationVectorStore

add, delete, delete, doDelete, similaritySearch

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.util.function.Consumer

andThen

Methods inherited from interface org.springframework.ai.document.DocumentWriter

write

Methods inherited from interface org.springframework.ai.vectorstore.VectorStore

accept, delete, getName

Constructor Details

CosmosDBVectorStore

protected CosmosDBVectorStore(CosmosDBVectorStore.Builder builder)

Protected constructor that accepts a builder instance. This is the preferred way to create new CosmosDBVectorStore instances.

Parameters:

builder - the configured builder instance

Method Details

builder

public static CosmosDBVectorStore.Builder builder(com.azure.cosmos.CosmosAsyncClient cosmosClient,
 EmbeddingModel embeddingModel)

close

public void close()

Specified by:

close in interface AutoCloseable

doAdd

public void doAdd(List<Document> documents)

Description copied from class: AbstractObservationVectorStore

Perform the actual add operation.

Specified by:

doAdd in class AbstractObservationVectorStore

Parameters:

documents - the documents to add

doDelete

public void doDelete(List<String> idList)

Description copied from class: AbstractObservationVectorStore

Perform the actual delete operation.

Specified by:

doDelete in class AbstractObservationVectorStore

Parameters:

idList - the list of document IDs to delete

similaritySearch

public List<Document> similaritySearch(String query)

Description copied from interface: VectorStoreRetriever

Retrieves documents by query embedding similarity using the default SearchRequest's search criteria.

Specified by:

similaritySearch in interface VectorStoreRetriever

Parameters:

query - Text to use for embedding similarity comparison.

Returns:

Returns a list of documents that have embeddings similar to the query text embedding.

doSimilaritySearch

public List<Document> doSimilaritySearch(SearchRequest request)

Description copied from class: AbstractObservationVectorStore

Perform the actual similarity search operation.

Specified by:

doSimilaritySearch in class AbstractObservationVectorStore

Parameters:

request - the search request

Returns:

the list of documents that match the query request conditions

createObservationContextBuilder

public VectorStoreObservationContext.Builder createObservationContextBuilder(String operationName)

Description copied from class: AbstractObservationVectorStore

Create a new VectorStoreObservationContext.Builder instance.

Specified by:

createObservationContextBuilder in class AbstractObservationVectorStore

Parameters:

operationName - the operation name

Returns:

the observation context builder

getNativeClient

public <T> Optional<T> getNativeClient()

Description copied from interface: VectorStore

Returns the native client if available in this vector store implementation. Note on usage: 1. Returns empty Optional when no native client is available 2. Due to Java type erasure, runtime type checking is not possible Example usage: When working with implementation with known native client: Optional client = vectorStore.getNativeClient(); Note: Using Optionalinvalid input: '<'?> will return the native client if one exists, rather than an empty Optional. For type safety, prefer using the specific client type.

Specified by:

getNativeClient in interface VectorStore

Type Parameters:

T - The type of the native client

Returns:

Optional containing native client if available, empty Optional otherwise