Spaces:
Running
Running
Update eb_agent_module.py
Browse files- 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-
|
44 |
-
GEMINI_EMBEDDING_MODEL_NAME = "
|
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 = {
|