All Downloads are FREE. Search and download functionalities are using the official Maven repository.

apoc.ml.aws.BedrockInvokeConfig Maven / Gradle / Ivy

package apoc.ml.aws;

import apoc.util.Util;

import java.util.Map;

public class BedrockInvokeConfig extends AWSConfig {
    public static final String MODEL = "model";
    public static final String OPEN_AI_COMPATIBLE = "openAICompatible";

    private final boolean openAICompatible;

    public BedrockInvokeConfig(Map config) {
        super(config);
        this.openAICompatible = Util.toBoolean(config.get(OPEN_AI_COMPATIBLE));
    }

    @Override
    String getDefaultEndpoint(Map config) {
        String modelId = (String) config.get(MODEL);
        return modelId == null
                ? null
                : String.format("https://bedrock-runtime.%s.amazonaws.com/model/%s/invoke", getRegion(), modelId);
    }

    @Override
    String getDefaultMethod() {
        return "POST";
    }

    public boolean isOpenAICompatible() {
        return openAICompatible;
    }
}




© 2015 - 2025 Weber Informatics LLC | Privacy Policy