DuckDB

https://duckdb.org/

Maven Dependency

<dependency>
    <groupId>dev.langchain4j</groupId>
    <artifactId>langchain4j-community-duckdb</artifactId>
    <version>1.0.0-beta4</version>
</dependency>

APIs

• DuckDBEmbeddingStore

Examples

// Init Model and Store
var embeddingStore = DuckDBEmbeddingStore.inMemory();
var embeddingModel = new AllMiniLmL6V2QuantizedEmbeddingModel();

//Create embeddings
Stream.of(
            "DuckDBは素晴らしいデータベースエンジンです！",
            "Pythonは型付けが本当に不足しています :D")
    .forEach(text -> {
        var segment = TextSegment.from(text);
        var embedding = embeddingModel.embed(segment).content();
        embeddingStore.add(embedding, segment);
    });


// Search request
var queryEmbedding = embeddingModel.embed("最高のデータベースエンジンは何ですか").content();
var request = EmbeddingSearchRequest.builder()
               .queryEmbedding(queryEmbedding)
               .maxResults(1)
               .build();

var relevant = embeddingStore.search(request);
EmbeddingMatch<TextSegment> embeddingMatch = relevant.matches().get(0);

// Show results
System.out.println(embeddingMatch.score()); // 0.8416415629618381
System.out.println(embeddingMatch.embedded().text()); // DuckDBは素晴らしいデータベースエンジンです！