From 26dc34bf840c3c8eb8e8fea273f3320edf4f6058 Mon Sep 17 00:00:00 2001
From: Umesh Yadav <git@umesh.dev>
Date: Thu, 28 Aug 2025 19:53:42 +0530
Subject: [PATCH] language_models: Fix tool calling for all grok-code model via
 OpenRouter

---
 crates/language_models/src/provider/open_router.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/crates/language_models/src/provider/open_router.rs b/crates/language_models/src/provider/open_router.rs
index 8f2abfce35852c..aaa0bd620ccf1a 100644
--- a/crates/language_models/src/provider/open_router.rs
+++ b/crates/language_models/src/provider/open_router.rs
@@ -381,7 +381,7 @@ impl LanguageModel for OpenRouterLanguageModel {
 
     fn tool_input_format(&self) -> LanguageModelToolSchemaFormat {
         let model_id = self.model.id().trim().to_lowercase();
-        if model_id.contains("gemini") || model_id.contains("grok-4") {
+        if model_id.contains("gemini") || model_id.contains("grok") {
             LanguageModelToolSchemaFormat::JsonSchemaSubset
         } else {
             LanguageModelToolSchemaFormat::JsonSchema