From 3c16e0495043b3c0981c17b7c288e92d64ff9ece Mon Sep 17 00:00:00 2001
From: Yang Yang <geraint0923@gmail.com>
Date: Sat, 18 Apr 2026 22:14:31 -0700
Subject: [PATCH] fix(xai): correctly resolve the canonical names and alias for
 xai models

---
 ...odel_prices_and_context_window_backup.json | 29 +++++-------------
 model_prices_and_context_window.json          | 30 +++++--------------
 2 files changed, 16 insertions(+), 43 deletions(-)

diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json
index 5cccd5f00af..df17f7ea1fb 100644
--- a/litellm/model_prices_and_context_window_backup.json
+++ b/litellm/model_prices_and_context_window_backup.json
@@ -34515,7 +34515,7 @@
         "supports_vision": true,
         "supports_web_search": true
     },
-    "xai/grok-4.20-multi-agent-beta-0309": {
+    "xai/grok-4.20-multi-agent-0309": {
         "cache_read_input_token_cost": 2e-07,
         "input_cost_per_token": 2e-06,
         "litellm_provider": "xai",
@@ -34529,23 +34529,8 @@
         "supports_reasoning": true,
         "supports_tool_choice": true,
         "supports_vision": true,
-        "supports_web_search": true
-    },
-    "xai/grok-4.20-beta-0309-reasoning": {
-        "cache_read_input_token_cost": 2e-07,
-        "input_cost_per_token": 2e-06,
-        "litellm_provider": "xai",
-        "max_input_tokens": 2000000,
-        "max_output_tokens": 2000000,
-        "max_tokens": 2000000,
-        "mode": "chat",
-        "output_cost_per_token": 6e-06,
-        "source": "https://docs.x.ai/docs/models",
-        "supports_function_calling": true,
-        "supports_reasoning": true,
-        "supports_tool_choice": true,
-        "supports_vision": true,
-        "supports_web_search": true
+        "supports_web_search": true,
+        "aliases": ["xai/grok-4.20-multi-agent-beta-0309"]
     },
     "xai/grok-4.20-0309-reasoning": {
         "cache_read_input_token_cost": 2e-07,
@@ -34561,9 +34546,10 @@
         "supports_reasoning": true,
         "supports_tool_choice": true,
         "supports_vision": true,
-        "supports_web_search": true
+        "supports_web_search": true,
+        "aliases": ["xai/grok-4.20-beta-0309-reasoning"]
     },
-    "xai/grok-4.20-beta-0309-non-reasoning": {
+    "xai/grok-4.20-0309-non-reasoning": {
         "cache_read_input_token_cost": 2e-07,
         "input_cost_per_token": 2e-06,
         "litellm_provider": "xai",
@@ -34576,7 +34562,8 @@
         "supports_function_calling": true,
         "supports_tool_choice": true,
         "supports_vision": true,
-        "supports_web_search": true
+        "supports_web_search": true,
+        "aliases": ["xai/grok-4.20-beta-0309-non-reasoning"]
     },
     "xai/grok-beta": {
         "input_cost_per_token": 5e-06,
diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json
index 4d8f3a984f7..08ef1408ad0 100644
--- a/model_prices_and_context_window.json
+++ b/model_prices_and_context_window.json
@@ -34590,7 +34590,7 @@
         "supports_vision": true,
         "supports_web_search": true
     },
-    "xai/grok-4.20-multi-agent-beta-0309": {
+    "xai/grok-4.20-multi-agent-0309": {
         "cache_read_input_token_cost": 2e-07,
         "input_cost_per_token": 2e-06,
         "litellm_provider": "xai",
@@ -34605,24 +34605,8 @@
         "supports_reasoning": true,
         "supports_tool_choice": true,
         "supports_vision": true,
-        "supports_web_search": true
-    },
-    "xai/grok-4.20-beta-0309-reasoning": {
-        "cache_read_input_token_cost": 2e-07,
-        "input_cost_per_token": 2e-06,
-        "litellm_provider": "xai",
-        "max_input_tokens": 2000000,
-        "max_output_tokens": 2000000,
-        "max_tokens": 2000000,
-        "mode": "chat",
-        "output_cost_per_token": 6e-06,
-        "source": "https://docs.x.ai/docs/models",
-        "supports_function_calling": true,
-        "supports_prompt_caching": true,
-        "supports_reasoning": true,
-        "supports_tool_choice": true,
-        "supports_vision": true,
-        "supports_web_search": true
+        "supports_web_search": true,
+        "aliases": ["xai/grok-4.20-multi-agent-beta-0309"]
     },
     "xai/grok-4.20-0309-reasoning": {
         "cache_read_input_token_cost": 2e-07,
@@ -34638,9 +34622,10 @@
         "supports_reasoning": true,
         "supports_tool_choice": true,
         "supports_vision": true,
-        "supports_web_search": true
+        "supports_web_search": true,
+        "aliases": ["xai/grok-4.20-beta-0309-reasoning"]
     },
-    "xai/grok-4.20-beta-0309-non-reasoning": {
+    "xai/grok-4.20-0309-non-reasoning": {
         "cache_read_input_token_cost": 2e-07,
         "input_cost_per_token": 2e-06,
         "litellm_provider": "xai",
@@ -34654,7 +34639,8 @@
         "supports_prompt_caching": true,
         "supports_tool_choice": true,
         "supports_vision": true,
-        "supports_web_search": true
+        "supports_web_search": true,
+        "aliases": ["xai/grok-4.20-beta-0309-non-reasoning"]
     },
     "xai/grok-beta": {
         "input_cost_per_token": 5e-06,