Package dev.langchain4j.model.googleai

Class GoogleAiGeminiChatModel

java.lang.Object

dev.langchain4j.model.googleai.GoogleAiGeminiChatModel

All Implemented Interfaces:

dev.langchain4j.model.chat.ChatLanguageModel, dev.langchain4j.model.chat.TokenCountEstimator

public class GoogleAiGeminiChatModel
extends Object
implements dev.langchain4j.model.chat.ChatLanguageModel, dev.langchain4j.model.chat.TokenCountEstimator

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	GoogleAiGeminiChatModel.GoogleAiGeminiChatModelBuilder

Field Summary

Fields

Modifier and Type	Field	Description
protected final boolean	allowCodeExecution
protected final String	apiKey
protected final dev.langchain4j.model.googleai.GeminiService	geminiService
protected final boolean	includeCodeExecutionOutput
protected final List<dev.langchain4j.model.chat.listener.ChatModelListener>	listeners
protected final Integer	maxOutputTokens
protected final Integer	maxRetries
protected final String	modelName
protected final dev.langchain4j.model.chat.request.ResponseFormat	responseFormat
protected final List<GeminiSafetySetting>	safetySettings
protected final List<String>	stopSequences
protected final Double	temperature
protected final GeminiFunctionCallingConfig	toolConfig
protected final Integer	topK
protected final Double	topP

Constructor Summary

Constructors

Constructor

Description

GoogleAiGeminiChatModel(String apiKey, String modelName, Integer maxRetries, Double temperature, Integer topK, Double topP, Integer maxOutputTokens, Duration timeout, dev.langchain4j.model.chat.request.ResponseFormat responseFormat, List<String> stopSequences, GeminiFunctionCallingConfig toolConfig, Boolean allowCodeExecution, Boolean includeCodeExecutionOutput, Boolean logRequestsAndResponses, List<GeminiSafetySetting> safetySettings, List<dev.langchain4j.model.chat.listener.ChatModelListener> listeners)

Method Summary

Modifier and Type	Method	Description
dev.langchain4j.model.chat.response.ChatResponse	chat(dev.langchain4j.model.chat.request.ChatRequest chatRequest)
protected static String	computeMimeType(dev.langchain4j.model.chat.request.ResponseFormat responseFormat)
protected dev.langchain4j.model.chat.listener.ChatModelRequest	createChatModelRequest(List<dev.langchain4j.data.message.ChatMessage> messages, List<dev.langchain4j.agent.tool.ToolSpecification> toolSpecifications)
protected dev.langchain4j.model.googleai.GeminiGenerateContentRequest	createGenerateContentRequest(List<dev.langchain4j.data.message.ChatMessage> messages, List<dev.langchain4j.agent.tool.ToolSpecification> toolSpecifications, dev.langchain4j.model.chat.request.ResponseFormat responseFormat)
int	estimateTokenCount(List<dev.langchain4j.data.message.ChatMessage> messages)
dev.langchain4j.model.output.Response<dev.langchain4j.data.message.AiMessage>	generate(List<dev.langchain4j.data.message.ChatMessage> messages)
dev.langchain4j.model.output.Response<dev.langchain4j.data.message.AiMessage>	generate(List<dev.langchain4j.data.message.ChatMessage> messages, dev.langchain4j.agent.tool.ToolSpecification toolSpecification)
dev.langchain4j.model.output.Response<dev.langchain4j.data.message.AiMessage>	generate(List<dev.langchain4j.data.message.ChatMessage> messages, List<dev.langchain4j.agent.tool.ToolSpecification> toolSpecifications)
protected void	notifyListenersOnError(Exception exception, dev.langchain4j.model.chat.listener.ChatModelRequest request, ConcurrentHashMap<Object,Object> attributes)
protected void	notifyListenersOnRequest(dev.langchain4j.model.chat.listener.ChatModelRequestContext context)
protected void	notifyListenersOnResponse(dev.langchain4j.model.output.Response<dev.langchain4j.data.message.AiMessage> response, dev.langchain4j.model.chat.listener.ChatModelRequest request, ConcurrentHashMap<Object,Object> attributes)
Set<dev.langchain4j.model.chat.Capability>	supportedCapabilities()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface dev.langchain4j.model.chat.ChatLanguageModel

generate, generate

Methods inherited from interface dev.langchain4j.model.chat.TokenCountEstimator

estimateTokenCount, estimateTokenCount, estimateTokenCount, estimateTokenCount

Field Details

geminiService

protected final dev.langchain4j.model.googleai.GeminiService geminiService

apiKey

protected final String apiKey

modelName

protected final String modelName

temperature

protected final Double temperature

topK

protected final Integer topK

topP

protected final Double topP

maxOutputTokens

protected final Integer maxOutputTokens

stopSequences

protected final List<String> stopSequences

responseFormat

protected final dev.langchain4j.model.chat.request.ResponseFormat responseFormat

toolConfig

protected final GeminiFunctionCallingConfig toolConfig

allowCodeExecution

protected final boolean allowCodeExecution

includeCodeExecutionOutput

protected final boolean includeCodeExecutionOutput

safetySettings

protected final List<GeminiSafetySetting> safetySettings

listeners

protected final List<dev.langchain4j.model.chat.listener.ChatModelListener> listeners

maxRetries

protected final Integer maxRetries

Constructor Details

GoogleAiGeminiChatModel

public GoogleAiGeminiChatModel(String apiKey,
 String modelName,
 Integer maxRetries,
 Double temperature,
 Integer topK,
 Double topP,
 Integer maxOutputTokens,
 Duration timeout,
 dev.langchain4j.model.chat.request.ResponseFormat responseFormat,
 List<String> stopSequences,
 GeminiFunctionCallingConfig toolConfig,
 Boolean allowCodeExecution,
 Boolean includeCodeExecutionOutput,
 Boolean logRequestsAndResponses,
 List<GeminiSafetySetting> safetySettings,
 List<dev.langchain4j.model.chat.listener.ChatModelListener> listeners)

Method Details

generate

public dev.langchain4j.model.output.Response<dev.langchain4j.data.message.AiMessage> generate(List<dev.langchain4j.data.message.ChatMessage> messages)

Specified by:

generate in interface dev.langchain4j.model.chat.ChatLanguageModel

generate

public dev.langchain4j.model.output.Response<dev.langchain4j.data.message.AiMessage> generate(List<dev.langchain4j.data.message.ChatMessage> messages,
 dev.langchain4j.agent.tool.ToolSpecification toolSpecification)

Specified by:

generate in interface dev.langchain4j.model.chat.ChatLanguageModel

generate

public dev.langchain4j.model.output.Response<dev.langchain4j.data.message.AiMessage> generate(List<dev.langchain4j.data.message.ChatMessage> messages,
 List<dev.langchain4j.agent.tool.ToolSpecification> toolSpecifications)

Specified by:

generate in interface dev.langchain4j.model.chat.ChatLanguageModel

chat

public dev.langchain4j.model.chat.response.ChatResponse chat(dev.langchain4j.model.chat.request.ChatRequest chatRequest)

Specified by:

chat in interface dev.langchain4j.model.chat.ChatLanguageModel

estimateTokenCount

public int estimateTokenCount(List<dev.langchain4j.data.message.ChatMessage> messages)

Specified by:

estimateTokenCount in interface dev.langchain4j.model.chat.TokenCountEstimator

supportedCapabilities

public Set<dev.langchain4j.model.chat.Capability> supportedCapabilities()

Specified by:

supportedCapabilities in interface dev.langchain4j.model.chat.ChatLanguageModel

createGenerateContentRequest

protected dev.langchain4j.model.googleai.GeminiGenerateContentRequest createGenerateContentRequest(List<dev.langchain4j.data.message.ChatMessage> messages,
 List<dev.langchain4j.agent.tool.ToolSpecification> toolSpecifications,
 dev.langchain4j.model.chat.request.ResponseFormat responseFormat)

createChatModelRequest

protected dev.langchain4j.model.chat.listener.ChatModelRequest createChatModelRequest(List<dev.langchain4j.data.message.ChatMessage> messages,
 List<dev.langchain4j.agent.tool.ToolSpecification> toolSpecifications)

computeMimeType

protected static String computeMimeType(dev.langchain4j.model.chat.request.ResponseFormat responseFormat)

notifyListenersOnRequest

protected void notifyListenersOnRequest(dev.langchain4j.model.chat.listener.ChatModelRequestContext context)

notifyListenersOnResponse

protected void notifyListenersOnResponse(dev.langchain4j.model.output.Response<dev.langchain4j.data.message.AiMessage> response,
 dev.langchain4j.model.chat.listener.ChatModelRequest request,
 ConcurrentHashMap<Object,Object> attributes)

notifyListenersOnError

protected void notifyListenersOnError(Exception exception,
 dev.langchain4j.model.chat.listener.ChatModelRequest request,
 ConcurrentHashMap<Object,Object> attributes)