• Home
• io.quarkiverse.langchain4j
• quarkus-langchain4j-easy-rag
• 0.23.0.CR1
• source code
• EasyRagManualIngestion.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.quarkiverse.langchain4j.easyrag.EasyRagManualIngestion Maven / Gradle / Ivy

package io.quarkiverse.langchain4j.easyrag;

import jakarta.enterprise.context.ApplicationScoped;
import jakarta.enterprise.inject.spi.CDI;

import dev.langchain4j.data.segment.TextSegment;
import dev.langchain4j.model.embedding.EmbeddingModel;
import dev.langchain4j.store.embedding.EmbeddingStore;
import io.quarkiverse.langchain4j.easyrag.runtime.EasyRagConfig;
import io.quarkiverse.langchain4j.easyrag.runtime.EasyRagIngestor;
import io.quarkiverse.langchain4j.easyrag.runtime.EasyRagRecorder;
import io.quarkiverse.langchain4j.easyrag.runtime.IngestionStrategy;

@ApplicationScoped
public class EasyRagManualIngestion {

    public void ingest() {
        EasyRagConfig config = EasyRagRecorder.easyRagConfig;
        if (config.ingestionStrategy() != IngestionStrategy.MANUAL) {
            throw new IllegalStateException("Manual ingestion trigger called when " +
                    "`quarkus.langchain4j.easy-rag.ingestion-strategy` is not MANUAL");
        }
        EmbeddingModel embeddingModel = CDI.current().select(EmbeddingModel.class).get();
        EmbeddingStore embeddingStore = CDI.current().select(EmbeddingStore.class).get();
        EasyRagIngestor ingestor = new EasyRagIngestor(embeddingModel, embeddingStore, config);
        ingestor.ingest();
    }
}