Package com.mulesoft.modules.agent.broker.internal.extension.connection.gemini

Class GeminiSession

java.lang.Object

com.mulesoft.modules.agent.broker.internal.extension.connection.BaseLLMSession

com.mulesoft.modules.agent.broker.internal.extension.connection.gemini.GeminiSession

All Implemented Interfaces:

LLMClient.LLMSession

public class GeminiSession
extends BaseLLMSession

Field Summary

Fields

Modifier and Type	Field	Description
static final String	LLM_ERROR_STRING
static final String	LLM_ROLE_MODEL
static final String	PROVIDE_STRUCTURED_ANSWER

Fields inherited from class com.mulesoft.modules.agent.broker.internal.extension.connection.BaseLLMSession

llmRequest, modelName

Constructor Summary

Constructors

Constructor	Description
GeminiSession(com.google.genai.Client client, LLMRequest llmRequest, GeminiSettings settings, List<com.google.genai.types.Tool> tools, Set<String> a2aToolSet)

Method Summary

Modifier and Type	Method	Description
void	addIteration(Iteration iteration)
void	addToolResponse(ToolResponse toolResponse)
protected CompletableFuture<org.mule.runtime.api.functional.Either<LLMOutput,ToolSelection>>	doGetNext(Map<String,String> additionalHeaders)	MG says: implementation hint.

Methods inherited from class com.mulesoft.modules.agent.broker.internal.extension.connection.BaseLLMSession

buildSystemPrompt, getInitialRequest, getNext

Methods inherited from class java.lang.Object

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

Field Details

PROVIDE_STRUCTURED_ANSWER

public static final String PROVIDE_STRUCTURED_ANSWER

See Also:

• Constant Field Values

LLM_ERROR_STRING

public static final String LLM_ERROR_STRING

See Also:

• Constant Field Values

LLM_ROLE_MODEL

public static final String LLM_ROLE_MODEL

See Also:

• Constant Field Values

Constructor Details

GeminiSession

public GeminiSession(com.google.genai.Client client,
 LLMRequest llmRequest,
 GeminiSettings settings,
 List<com.google.genai.types.Tool> tools,
 Set<String> a2aToolSet)

Method Details

doGetNext

protected CompletableFuture<org.mule.runtime.api.functional.Either<LLMOutput,ToolSelection>> doGetNext(Map<String,String> additionalHeaders)

MG says: implementation hint. Gemini doesn't support function calls and structured output at the same time. Define an LLMOutput equivalent function to compensate. Go to master branch and see the (now defunct) FunctionBasedSession class for reference

Specified by:

doGetNext in class BaseLLMSession

addIteration

public void addIteration(Iteration iteration)

addToolResponse

public void addToolResponse(ToolResponse toolResponse)