From 346410e34e1f4b8874e18f0c38c1fbaabfbc1cd6 Mon Sep 17 00:00:00 2001 From: Raymond Lucke Date: Thu, 10 Jul 2025 18:21:44 -0700 Subject: [PATCH] Use ModelCapabilities for Grok thinking mode support check Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --- providers/xai.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/providers/xai.py b/providers/xai.py index 98dbd55..aca75cb 100644 --- a/providers/xai.py +++ b/providers/xai.py @@ -148,6 +148,10 @@ class XAIModelProvider(OpenAICompatibleProvider): def supports_thinking_mode(self, model_name: str) -> bool: """Check if the model supports extended thinking mode.""" - # Grok-4 supports extended thinking/reasoning mode - resolved_name = self._resolve_model_name(model_name) - return resolved_name == "grok-4-0709" + # Check capabilities to determine thinking mode support + try: + capabilities = self.get_capabilities(model_name) + return capabilities.supports_extended_thinking + except ValueError: + # If the model is not supported, it doesn't support thinking mode. + return False