Spaces:
Sleeping
Sleeping
Update pipeline.py
Browse files- 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 |
-
|
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
|