Spaces:
Running
Running
Anirudh Esthuri
commited on
Commit
·
1f9a06d
1
Parent(s):
61fbcaf
Update Gemini model names to use available models (gemini-1.5-pro and gemini-2.0-flash-exp)
Browse files- llm.py +1 -1
- model_config.py +4 -4
llm.py
CHANGED
|
@@ -275,7 +275,7 @@ def chat(messages, persona):
|
|
| 275 |
f"Invalid Gemini model ID: '{MODEL_STRING}'. "
|
| 276 |
f"Error: {error_msg}. "
|
| 277 |
"Please verify the model ID is correct. "
|
| 278 |
-
"Common Gemini model IDs: 'gemini-
|
| 279 |
) from e
|
| 280 |
raise
|
| 281 |
elif provider == "deepseek":
|
|
|
|
| 275 |
f"Invalid Gemini model ID: '{MODEL_STRING}'. "
|
| 276 |
f"Error: {error_msg}. "
|
| 277 |
"Please verify the model ID is correct. "
|
| 278 |
+
"Common Gemini model IDs: 'gemini-1.5-pro', 'gemini-1.5-flash', 'gemini-2.0-flash-exp', 'gemini-pro'"
|
| 279 |
) from e
|
| 280 |
raise
|
| 281 |
elif provider == "deepseek":
|
model_config.py
CHANGED
|
@@ -11,8 +11,8 @@ PROVIDER_MODEL_MAP = {
|
|
| 11 |
"anthropic.claude-opus-4-20250514-v1:0",
|
| 12 |
],
|
| 13 |
"google": [
|
| 14 |
-
"gemini-
|
| 15 |
-
"gemini-
|
| 16 |
],
|
| 17 |
}
|
| 18 |
|
|
@@ -32,8 +32,8 @@ MODEL_DISPLAY_NAMES = {
|
|
| 32 |
"anthropic.claude-haiku-4-5-20251001-v1:0": "AWS Bedrock - Anthropic - Claude Haiku 4.5",
|
| 33 |
"anthropic.claude-sonnet-4-5-20250929-v1:0": "AWS Bedrock - Anthropic - Claude Sonnet 4.5",
|
| 34 |
"anthropic.claude-opus-4-20250514-v1:0": "AWS Bedrock - Anthropic - Claude Opus 4",
|
| 35 |
-
"gemini-
|
| 36 |
-
"gemini-
|
| 37 |
}
|
| 38 |
|
| 39 |
MODEL_CHOICES = [model for models in PROVIDER_MODEL_MAP.values() for model in models]
|
|
|
|
| 11 |
"anthropic.claude-opus-4-20250514-v1:0",
|
| 12 |
],
|
| 13 |
"google": [
|
| 14 |
+
"gemini-2.0-flash-exp",
|
| 15 |
+
"gemini-1.5-pro",
|
| 16 |
],
|
| 17 |
}
|
| 18 |
|
|
|
|
| 32 |
"anthropic.claude-haiku-4-5-20251001-v1:0": "AWS Bedrock - Anthropic - Claude Haiku 4.5",
|
| 33 |
"anthropic.claude-sonnet-4-5-20250929-v1:0": "AWS Bedrock - Anthropic - Claude Sonnet 4.5",
|
| 34 |
"anthropic.claude-opus-4-20250514-v1:0": "AWS Bedrock - Anthropic - Claude Opus 4",
|
| 35 |
+
"gemini-2.0-flash-exp": "Google - Gemini 2.0 Flash (Experimental)",
|
| 36 |
+
"gemini-1.5-pro": "Google - Gemini 1.5 Pro",
|
| 37 |
}
|
| 38 |
|
| 39 |
MODEL_CHOICES = [model for models in PROVIDER_MODEL_MAP.values() for model in models]
|