package com.example;

import dev.langchain4j.data.segment.TextSegment;
import dev.langchain4j.model.embedding.EmbeddingModel;
import dev.langchain4j.model.openai.OpenAiEmbeddingModel;
import dev.langchain4j.store.embedding.EmbeddingStore;
import dev.langchain4j.store.embedding.chroma.ChromaEmbeddingStore;

public class Chroma {

    public static final EmbeddingStore<TextSegment> embeddingStore =
            ChromaEmbeddingStore.builder()
                    .baseUrl("http://localhost:8000/")
                    .collectionName("my-collection")
                    .build();

    public static final EmbeddingModel embeddingModel =
            OpenAiEmbeddingModel.builder()
                    .apiKey("API_KEY")
                    .modelName("text-embedding-ada-002")
                    .build();
}

[((412, 579), 'dev.langchain4j.store.embedding.chroma.ChromaEmbeddingStore.builder'), ((412, 550), 'dev.langchain4j.store.embedding.chroma.ChromaEmbeddingStore.builder'), ((412, 497), 'dev.langchain4j.store.embedding.chroma.ChromaEmbeddingStore.builder'), ((650, 805), 'dev.langchain4j.model.openai.OpenAiEmbeddingModel.builder'), ((650, 776), 'dev.langchain4j.model.openai.OpenAiEmbeddingModel.builder'), ((650, 719), 'dev.langchain4j.model.openai.OpenAiEmbeddingModel.builder')]