io.quarkiverse.langchain4j.deployment.OpenNlpProcessor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quarkus-langchain4j-core-deployment Show documentation

There is a newer version: 0.21.0

Show newest version

package io.quarkiverse.langchain4j.deployment;

import static io.quarkiverse.langchain4j.deployment.JarResourceUtil.determineJarLocation;
import static io.quarkiverse.langchain4j.deployment.JarResourceUtil.matchingJarEntries;

import java.nio.file.Path;
import java.util.List;
import java.util.stream.Collectors;
import java.util.zip.ZipEntry;

import dev.langchain4j.data.document.splitter.DocumentBySentenceSplitter;
import io.quarkiverse.langchain4j.deployment.items.InProcessEmbeddingBuildItem;
import io.quarkus.deployment.annotations.BuildProducer;
import io.quarkus.deployment.annotations.BuildStep;
import io.quarkus.deployment.builditem.nativeimage.NativeImageResourceBuildItem;
import io.quarkus.deployment.builditem.nativeimage.ReflectiveClassBuildItem;

public class OpenNlpProcessor {

    @BuildStep
    void nativeResources(
            List inProcessEmbeddingBuildItems,
            BuildProducer reflectionProducer,
            BuildProducer nativeImageResourceProducer) {
        if (!inProcessEmbeddingBuildItems.isEmpty()) {
            reflectionProducer
                    .produce(ReflectiveClassBuildItem.builder("opennlp.tools.sentdetect.SentenceDetectorFactory").build());
        }
        Path langChain4jJar = determineJarLocation(DocumentBySentenceSplitter.class);
        List names = matchingJarEntries(langChain4jJar, e -> e.getName().endsWith(".bin")).stream().map(
                ZipEntry::getName).collect(Collectors.toList());
        if (!names.isEmpty()) {
            nativeImageResourceProducer.produce(new NativeImageResourceBuildItem(names));
        }
    }

}