GuglielmoTor commited on
Commit
5506915
·
verified ·
1 Parent(s): c9f7ea0

Update eb_agent_module.py

Browse files
Files changed (1) hide show
  1. eb_agent_module.py +2 -2
eb_agent_module.py CHANGED
@@ -40,8 +40,8 @@ except ImportError:
40
 
41
  # --- Configuration ---
42
  GEMINI_API_KEY = os.getenv('GEMINI_API_KEY', "") # Use your environment variable
43
- LLM_MODEL_NAME = "gemini-1.5-flash-latest"
44
- GEMINI_EMBEDDING_MODEL_NAME = "models/embedding-001" # Standard embedding model
45
 
46
  # Generation configuration for the LLM
47
  GENERATION_CONFIG_PARAMS = {
 
40
 
41
  # --- Configuration ---
42
  GEMINI_API_KEY = os.getenv('GEMINI_API_KEY', "") # Use your environment variable
43
+ LLM_MODEL_NAME = "gemini-2.0-flash"
44
+ GEMINI_EMBEDDING_MODEL_NAME = "gemini-embedding-exp-03-07" # Standard embedding model
45
 
46
  # Generation configuration for the LLM
47
  GENERATION_CONFIG_PARAMS = {