dev.langchain4j.store.embedding.chroma

Class ChromaEmbeddingStore

java.lang.Object

dev.langchain4j.store.embedding.chroma.ChromaEmbeddingStore

All Implemented Interfaces:

EmbeddingStore<TextSegment>

public class ChromaEmbeddingStore
extends Object
implements EmbeddingStore<TextSegment>

Represents a store for embeddings using the Chroma backend. Always uses cosine distance as the distance metric.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ChromaEmbeddingStore.Builder

Constructor Summary

Constructors

Constructor and Description
ChromaEmbeddingStore(String baseUrl, String collectionName, Duration timeout) Initializes a new instance of ChromaEmbeddingStore with the specified parameters.

Method Summary

Modifier and Type	Method and Description
String	add(Embedding embedding)
String	add(Embedding embedding, TextSegment textSegment)
void	add(String id, Embedding embedding)
List<String>	addAll(List<Embedding> embeddings)
List<String>	addAll(List<Embedding> embeddings, List<TextSegment> textSegments)
static ChromaEmbeddingStore.Builder	builder()
List<EmbeddingMatch<TextSegment>>	findRelevant(Embedding referenceEmbedding, int maxResults, double minScore)

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

findRelevant

Constructor Detail

ChromaEmbeddingStore

public ChromaEmbeddingStore(String baseUrl,
                            String collectionName,
                            Duration timeout)

Initializes a new instance of ChromaEmbeddingStore with the specified parameters.

Parameters:

baseUrl - The base URL of the Chroma service.

collectionName - The name of the collection in the Chroma service. If not specified, "default" will be used.

timeout - The timeout duration for the Chroma client. If not specified, 5 seconds will be used.

Method Detail

builder

public static ChromaEmbeddingStore.Builder builder()

add

public String add(Embedding embedding)

Specified by:

add in interface EmbeddingStore<TextSegment>

add

public void add(String id,
                Embedding embedding)

Specified by:

add in interface EmbeddingStore<TextSegment>

add

public String add(Embedding embedding,
                  TextSegment textSegment)

Specified by:

add in interface EmbeddingStore<TextSegment>

addAll

public List<String> addAll(List<Embedding> embeddings)

Specified by:

addAll in interface EmbeddingStore<TextSegment>

addAll

public List<String> addAll(List<Embedding> embeddings,
                           List<TextSegment> textSegments)

Specified by:

addAll in interface EmbeddingStore<TextSegment>

findRelevant

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

Specified by:

findRelevant in interface EmbeddingStore<TextSegment>