• Home
• io.thomasvitale.langchain4j
• langchain4j-spring-weaviate
• 0.12.0
• source code
• WeaviateAdapters.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.thomasvitale.langchain4j.spring.weaviate.WeaviateAdapters Maven / Gradle / Ivy

package io.thomasvitale.langchain4j.spring.weaviate;

import java.util.List;
import java.util.Map;

import dev.langchain4j.data.embedding.Embedding;
import dev.langchain4j.data.segment.TextSegment;
import dev.langchain4j.store.embedding.EmbeddingMatch;

import org.springframework.util.StringUtils;

import static io.thomasvitale.langchain4j.spring.weaviate.WeaviateEmbeddingStore.ADDITIONAL_CERTAINTY_FIELD_NAME;
import static io.thomasvitale.langchain4j.spring.weaviate.WeaviateEmbeddingStore.ADDITIONAL_FIELD_NAME;
import static io.thomasvitale.langchain4j.spring.weaviate.WeaviateEmbeddingStore.ADDITIONAL_ID_FIELD_NAME;
import static io.thomasvitale.langchain4j.spring.weaviate.WeaviateEmbeddingStore.ADDITIONAL_VECTOR_FIELD_NAME;
import static io.thomasvitale.langchain4j.spring.weaviate.WeaviateEmbeddingStore.CONTENT_FIELD_NAME;

/**
 * Utility class to convert between Weaviate and Langchain4j types.
 * 

 * Based on the original LangChain4j implementation.
 */
public class WeaviateAdapters {

    public static EmbeddingMatch toEmbeddingMatch(Map item) {
        Map additional = (Map) item.get(ADDITIONAL_FIELD_NAME);
        double certainty = (Double) additional.get(ADDITIONAL_CERTAINTY_FIELD_NAME);
        String id = (String) additional.get(ADDITIONAL_ID_FIELD_NAME);
        List embedding = ((List) additional.get(ADDITIONAL_VECTOR_FIELD_NAME)).stream()
                .map(Double::floatValue).toList();

        String content = (String) item.get(CONTENT_FIELD_NAME);

        return new EmbeddingMatch<>(certainty, id, Embedding.from(embedding), StringUtils.hasText(content) ? TextSegment.from(content) : null);
    }

}