Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -278,7 +278,7 @@ def setup_vector_store() -> SupabaseVectorStore:
|
|
| 278 |
# LLM Provider Setup
|
| 279 |
# ===================
|
| 280 |
|
| 281 |
-
def get_llm(provider: str = "
|
| 282 |
"""
|
| 283 |
Initialize and return an LLM based on the specified provider.
|
| 284 |
|
|
@@ -293,7 +293,7 @@ def get_llm(provider: str = "groq"):
|
|
| 293 |
"""
|
| 294 |
if provider == "google":
|
| 295 |
logger.info("Using Google Gemini as LLM provider")
|
| 296 |
-
return ChatGoogleGenerativeAI(model="gemini-2.
|
| 297 |
|
| 298 |
elif provider == "groq":
|
| 299 |
logger.info("Using Groq as LLM provider with qwen-qwq-32b model")
|
|
|
|
| 278 |
# LLM Provider Setup
|
| 279 |
# ===================
|
| 280 |
|
| 281 |
+
def get_llm(provider: str = "google"):
|
| 282 |
"""
|
| 283 |
Initialize and return an LLM based on the specified provider.
|
| 284 |
|
|
|
|
| 293 |
"""
|
| 294 |
if provider == "google":
|
| 295 |
logger.info("Using Google Gemini as LLM provider")
|
| 296 |
+
return ChatGoogleGenerativeAI(model="gemini-2.5-flash-preview-04-17", temperature=0)
|
| 297 |
|
| 298 |
elif provider == "groq":
|
| 299 |
logger.info("Using Groq as LLM provider with qwen-qwq-32b model")
|