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")
|