GuglielmoTor commited on
Commit
342bc51
·
verified ·
1 Parent(s): 4cd2f6c

Update eb_agent_module.py

Browse files
Files changed (1) hide show
  1. eb_agent_module.py +2 -2
eb_agent_module.py CHANGED
@@ -71,9 +71,9 @@ class AgentNotReadyError(Exception):
71
 
72
  # --- Configuration Constants ---
73
  GEMINI_API_KEY = os.getenv('GEMINI_API_KEY', "")
74
- LLM_MODEL_NAME = "gemini-1.5-flash-latest" # For google-generativeai, model name is directly used.
75
  # For client.models.generate_content, it might need "models/gemini-1.5-flash-latest"
76
- GEMINI_EMBEDDING_MODEL_NAME = "text-embedding-004" # Similarly, might need "models/text-embedding-004"
77
 
78
  GENERATION_CONFIG_PARAMS = {
79
  "temperature": 0.7,
 
71
 
72
  # --- Configuration Constants ---
73
  GEMINI_API_KEY = os.getenv('GEMINI_API_KEY', "")
74
+ LLM_MODEL_NAME = "gemini-2.5-flash-preview-05-20" # For google-generativeai, model name is directly used.
75
  # For client.models.generate_content, it might need "models/gemini-1.5-flash-latest"
76
+ GEMINI_EMBEDDING_MODEL_NAME = "gemini-embedding-exp-03-07" # Similarly, might need "models/text-embedding-004"
77
 
78
  GENERATION_CONFIG_PARAMS = {
79
  "temperature": 0.7,