dev.langchain4j.image_to_diagram.ImageToDiagram Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of langgraph4j-image-to-diagram Show documentation
There is a newer version: 1.0-20241113
Show newest version
package dev.langchain4j.image_to_diagram;

import dev.langchain4j.model.input.PromptTemplate;
import dev.langchain4j.model.openai.OpenAiChatModel;
import net.sourceforge.plantuml.ErrorUmlType;
import org.bsc.async.AsyncGenerator;
import org.bsc.langgraph4j.NodeOutput;
import org.bsc.langgraph4j.state.AgentState;
import org.bsc.langgraph4j.state.AppendableValue;

import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.Map;
import java.util.Optional;

import static java.util.Optional.ofNullable;

public interface ImageToDiagram {

    class State extends AgentState {

        public State(Map initData) {
            super(initData);
        }

        public Optional diagram() {
            return value("diagram");
        }
        public AppendableValue diagramCode() {
            return appendableValue("diagramCode");
        }
        public Optional evaluationResult() {
            return value("evaluationResult" );
        }
        public Optional evaluationError() {
            return value("evaluationError" );
        }
        public Optional evaluationErrorType() {
            return value("evaluationErrorType" );
        }

        public boolean isExecutionError() {
            return evaluationErrorType()
                    .map( type -> type == ErrorUmlType.EXECUTION_ERROR )
                    .orElse(false);
        }

        public boolean lastTwoDiagramsAreEqual() {
            if( diagramCode().size() < 2 ) return false;

            String last = diagramCode().last()
                    .map(String::trim)
                    .orElseThrow( () -> new IllegalStateException( "last() is null!" ) );
            String prev = diagramCode().lastMinus(1)
                    .map(String::trim)
                    .orElseThrow( () -> new IllegalStateException( "last(-1) is null!" ) );

            return last.equals(prev);
        }

    }

    enum EvaluationResult {
        OK,
        ERROR,
        UNKNOWN
    }

    OpenAiChatModel getLLM();

    default PromptTemplate loadPromptTemplate(String resourceName ) throws Exception {
        final ClassLoader classLoader = getClass().getClassLoader();
        final InputStream inputStream = classLoader.getResourceAsStream(resourceName);
        if (inputStream == null) {
            throw new IllegalArgumentException("File not found: " + resourceName);
        }
        try( final BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream), 4*1024) ) {
            StringBuilder result = new StringBuilder();
            String line;
            while ((line = reader.readLine()) != null) {
                result.append(line).append('\n');
            }
            return PromptTemplate.from( result.toString() );
        }
    }

    AsyncGenerator> execute(Map inputs ) throws Exception;


}