Update model_config.py

model_config.py

@@ -1,17 +1,15 @@
 PROVIDER_MODEL_MAP = {
-    "openai": ["gpt-4.1-mini"],
-    "anthropic": [
-        "anthropic.claude-3-sonnet-20240229-v1:0",
-        "anthropic.claude-3-5-haiku-20241022-v1:0",
+    "openai": [
+        "gpt-4.1-mini",
+        "gpt-5-mini",
+        "gpt-5-nano",
     ],
-    "deepseek": ["us.deepseek.r1-v1:0"],
-    "meta": ["meta.llama3-8b-instruct-v1:0", "meta.llama3-70b-instruct-v1:0"],
-    "mistral": [
-        "mistral.mixtral-8x7b-instruct-v0:1",
-        "mistral.mistral-7b-instruct-v0:2",
+    "anthropic": [
+        "anthropic.claude-3-5-sonnet-20241022-v2:0",
+        "anthropic.claude-3-haiku-20240307-v1:0",
     ],
 }
-# "meta.llama4-maverick-17b-instruct-v1:0" (not currently working)
+
 
 MODEL_TO_PROVIDER = {
     model: provider
@@ -19,4 +17,13 @@ MODEL_TO_PROVIDER = {
     for model in models
 }
 
+# Model display names with categories
+MODEL_DISPLAY_NAMES = {
+    "gpt-4.1-mini": "OpenAI - GPT-4.1 Mini",
+    "gpt-5-mini": "OpenAI - GPT-5 Mini",
+    "gpt-5-nano": "OpenAI - GPT-5 Nano",
+    "anthropic.claude-3-5-sonnet-20241022-v2:0": "AWS Bedrock - Anthropic - Claude 3.5 Sonnet",
+    "anthropic.claude-3-haiku-20240307-v1:0": "AWS Bedrock - Anthropic - Claude 3 Haiku",
+}
+
 MODEL_CHOICES = [model for models in PROVIDER_MODEL_MAP.values() for model in models]