raaec commited on
Commit
7fdc0b2
·
verified ·
1 Parent(s): 8f6ee9b

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -278,7 +278,7 @@ def setup_vector_store() -> SupabaseVectorStore:
278
  # LLM Provider Setup
279
  # ===================
280
 
281
- def get_llm(provider: str = "groq"):
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.0-flash", temperature=0)
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")