From 4a66777761b85a7cc40af1a07ddbf4b29feb450a Mon Sep 17 00:00:00 2001
From: Chenyme <118253778+chenyme@users.noreply.github.com>
Date: Thu, 5 Feb 2026 15:15:01 +0800
Subject: [PATCH] feat: support grok 4 heavy models

---
 app/services/grok/model.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/app/services/grok/model.py b/app/services/grok/model.py
index 74bd2c5..e85d694 100644
--- a/app/services/grok/model.py
+++ b/app/services/grok/model.py
@@ -83,6 +83,14 @@ class ModelService:
             cost=Cost.LOW,
             display_name="GROK-4-THINKING",
         ),
+        ModelInfo(
+            model_id="grok-4-heavy",
+            grok_model="grok-4",
+            model_mode="MODEL_MODE_HEAVY",
+            cost=Cost.HIGH,
+            tier=Tier.SUPER,
+            display_name="GROK-4-HEAVY",
+        ),
         ModelInfo(
             model_id="grok-4.1-mini",
             grok_model="grok-4-1-thinking-1129",