Francesco-A commited on
Commit
d02a7bb
·
1 Parent(s): 9b507b3

Reduced max_tokens for Gemini agent

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -169,7 +169,7 @@ class GeminiAgent:
169
  api_key = GOOGLE_API_KEY,
170
  temperature = 0.0,
171
  top_p = 1.0,
172
- max_tokens = 4098,
173
  timeout = 120 # Add timeout to prevent hanging
174
  )
175
  # If native, we can optionally remove image_to_text from tools to prevent the agent from getting confused
 
169
  api_key = GOOGLE_API_KEY,
170
  temperature = 0.0,
171
  top_p = 1.0,
172
+ max_tokens = 2048,
173
  timeout = 120 # Add timeout to prevent hanging
174
  )
175
  # If native, we can optionally remove image_to_text from tools to prevent the agent from getting confused