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
Files changed (2) hide show
  1. llm.py +1 -1
  2. 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-3.0-pro', 'gemini-2.5-flash', 'gemini-1.5-pro', 'gemini-1.5-flash'"
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-3.0-pro",
15
- "gemini-2.5-flash",
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-3.0-pro": "Google - Gemini 3.0 Pro",
36
- "gemini-2.5-flash": "Google - Gemini 2.5 Flash",
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]