Phoenix21 commited on
Commit
d5e80d3
·
verified ·
1 Parent(s): 0d4fb22

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +3 -2
pipeline.py CHANGED
@@ -164,12 +164,13 @@ RATE_LIMIT_REQUESTS = 60
164
  CACHE_SIZE_LIMIT = 1000
165
 
166
  # Google Gemini (primary)
167
- GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
168
  gemini_llm = ChatGoogleGenerativeAI(
169
  model="gemini-2.0-flash",
170
  temperature=0.5,
 
 
171
  max_retries=2,
172
- google_api_key=GEMINI_API_KEY
173
  )
174
 
175
  # Fallback
 
164
  CACHE_SIZE_LIMIT = 1000
165
 
166
  # Google Gemini (primary)
167
+ GOOGLE_API_KEY = os.environ.get("GEMINI_API_KEY")
168
  gemini_llm = ChatGoogleGenerativeAI(
169
  model="gemini-2.0-flash",
170
  temperature=0.5,
171
+ max_tokens=None,
172
+ timeout=None,
173
  max_retries=2,
 
174
  )
175
 
176
  # Fallback