dev.langchain4j.image_to_diagram.DiagramOutputParser 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 com.google.gson.Gson;
import lombok.var;

import java.util.regex.Pattern;

public class DiagramOutputParser {
    public Diagram.Element parse(String s) {
        // String pattern = "```json\n(.*?)\n```";
        String pattern = "```json\n(.*?)\n```";

        // Create a Pattern object
        var jsonPattern = Pattern.compile(pattern, Pattern.DOTALL | Pattern.MULTILINE);

        // Create a Matcher object
        var matcher = jsonPattern.matcher(s);

        // Check if a match is found
        if (!matcher.find()) {
            throw new IllegalArgumentException("no diagram provided! in text\n" + s);
        }

        var gson = new Gson();
        return gson.fromJson(matcher.group(1), Diagram.Element.class);
    }
}