fix(gemini): token count (#3039)

* fix(gemini): token count

* fix to include tool call tokens
This commit is contained in:
Vikhyath Mondreti
2026-01-27 19:16:54 -08:00
committed by GitHub
parent b4a389a71f
commit c8ffda1616
3 changed files with 16 additions and 8 deletions

View File

@@ -936,8 +936,12 @@ export class AgentBlockHandler implements BlockHandler {
systemPrompt: validMessages ? undefined : inputs.systemPrompt,
context: validMessages ? undefined : stringifyJSON(messages),
tools: formattedTools,
temperature: inputs.temperature != null ? Number(inputs.temperature) : undefined,
maxTokens: inputs.maxTokens != null ? Number(inputs.maxTokens) : undefined,
temperature:
inputs.temperature != null && inputs.temperature !== ''
? Number(inputs.temperature)
: undefined,
maxTokens:
inputs.maxTokens != null && inputs.maxTokens !== '' ? Number(inputs.maxTokens) : undefined,
apiKey: inputs.apiKey,
azureEndpoint: inputs.azureEndpoint,
azureApiVersion: inputs.azureApiVersion,

View File

@@ -14,8 +14,8 @@ export interface AgentInputs {
slidingWindowSize?: string // For message-based sliding window
slidingWindowTokens?: string // For token-based sliding window
// LLM parameters
temperature?: number
maxTokens?: number
temperature?: string
maxTokens?: string
apiKey?: string
azureEndpoint?: string
azureApiVersion?: string