File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -335,7 +335,7 @@ def get_llm_and_embeddings(
335335 provider = provider .lower ()
336336 if provider == "google" :
337337 if llm_model_name is None :
338- llm_model_name = "gemini-2.5-flash"
338+ llm_model_name = "gemini-2.5-flash-lite" #"gemini-2.0-flash" # gemini-2.0-flash-lite-001" #"gemini-2.5-flash "
339339 if embedding_model_name is None :
340340 embedding_model_name = "models/embedding-001"
341341
@@ -356,7 +356,7 @@ def get_llm_and_embeddings(
356356 elif provider == "openai" :
357357 if llm_model_name is None :
358358 # Use a powerful and cost-effective model like gpt-4o
359- llm_model_name = "gpt-4o"
359+ llm_model_name = "gpt-4o-mini" #"gpt-4o "
360360 if embedding_model_name is None :
361361 # Use OpenAI's latest embedding model
362362 embedding_model_name = "text-embedding-3-small"
You can’t perform that action at this time.
0 commit comments