Spaces:
Sleeping
Sleeping
Update eb_agent_module.py
Browse files- 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-
|
75 |
# For client.models.generate_content, it might need "models/gemini-1.5-flash-latest"
|
76 |
-
GEMINI_EMBEDDING_MODEL_NAME = "
|
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,
|