io.quarkiverse.langchain4j.vertexai.runtime.gemini.VertexAiGeminiChatLanguageModel Maven / Gradle / Ivy
The newest version!
package io.quarkiverse.langchain4j.vertexai.runtime.gemini;
import java.net.URI;
import java.net.URISyntaxException;
import java.time.Duration;
import java.util.Collections;
import java.util.List;
import java.util.Optional;
import java.util.concurrent.TimeUnit;
import org.jboss.resteasy.reactive.client.api.LoggingScope;
import dev.langchain4j.data.message.AiMessage;
import dev.langchain4j.data.message.ChatMessage;
import dev.langchain4j.model.chat.ChatLanguageModel;
import dev.langchain4j.model.output.Response;
import io.quarkus.rest.client.reactive.QuarkusRestClientBuilder;
public class VertexAiGeminiChatLanguageModel implements ChatLanguageModel {
private final GenerationConfig generationConfig;
private final VertxAiGeminiRestApi.ApiMetadata apiMetadata;
private final VertxAiGeminiRestApi restApi;
private VertexAiGeminiChatLanguageModel(Builder builder) {
this.generationConfig = GenerationConfig.builder()
.maxOutputTokens(builder.maxOutputTokens)
.temperature(builder.temperature)
.topK(builder.topK)
.topP(builder.topP)
.build();
this.apiMetadata = VertxAiGeminiRestApi.ApiMetadata
.builder()
.modelId(builder.modelId)
.location(builder.location)
.projectId(builder.projectId)
.publisher(builder.publisher)
.build();
try {
String baseUrl = builder.baseUrl.orElse(String.format("https://%s-aiplatform.googleapis.com", builder.location));
var restApiBuilder = QuarkusRestClientBuilder.newBuilder()
.baseUri(new URI(baseUrl))
.connectTimeout(builder.timeout.toSeconds(), TimeUnit.SECONDS)
.readTimeout(builder.timeout.toSeconds(), TimeUnit.SECONDS);
if (builder.logRequests || builder.logResponses) {
restApiBuilder.loggingScope(LoggingScope.REQUEST_RESPONSE);
restApiBuilder.clientLogger(new VertxAiGeminiRestApi.VertxAiClientLogger(builder.logRequests,
builder.logResponses));
}
restApi = restApiBuilder.build(VertxAiGeminiRestApi.class);
} catch (URISyntaxException e) {
throw new RuntimeException(e);
}
}
@Override
public Response generate(List messages) {
GenerateContentRequest request = ContentMapper.map(messages, Collections.emptyList(), generationConfig);
GenerateContentResponse response = restApi.generateContent(request, apiMetadata);
return Response.from(
AiMessage.from(GenerateContentResponseHandler.getText(response)),
GenerateContentResponseHandler.getTokenUsage(response.usageMetadata()),
FinishReasonMapper.map(GenerateContentResponseHandler.getFinishReason(response)));
}
public static Builder builder() {
return new Builder();
}
@SuppressWarnings("OptionalUsedAsFieldOrParameterType")
public static final class Builder {
private Optional baseUrl = Optional.empty();
private String projectId;
private String location;
private String modelId;
private String publisher;
private Double temperature;
private Integer maxOutputTokens;
private Integer topK;
private Double topP;
private Duration timeout = Duration.ofSeconds(10);
private Boolean logRequests = false;
private Boolean logResponses = false;
public Builder baseUrl(Optional baseUrl) {
this.baseUrl = baseUrl;
return this;
}
public Builder projectId(String projectId) {
this.projectId = projectId;
return this;
}
public Builder location(String location) {
this.location = location;
return this;
}
public Builder modelId(String modelId) {
this.modelId = modelId;
return this;
}
public Builder publisher(String publisher) {
this.publisher = publisher;
return this;
}
public Builder temperature(Double temperature) {
this.temperature = temperature;
return this;
}
public Builder maxOutputTokens(Integer maxOutputTokens) {
this.maxOutputTokens = maxOutputTokens;
return this;
}
public Builder topK(Integer topK) {
this.topK = topK;
return this;
}
public Builder topP(Double topP) {
this.topP = topP;
return this;
}
public Builder timeout(Duration timeout) {
this.timeout = timeout;
return this;
}
public Builder logRequests(boolean logRequests) {
this.logRequests = logRequests;
return this;
}
public Builder logResponses(boolean logResponses) {
this.logResponses = logResponses;
return this;
}
public VertexAiGeminiChatLanguageModel build() {
return new VertexAiGeminiChatLanguageModel(this);
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy