Spaces:
Build error
Build error
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
|