diff --git a/tradingagents/graph/trading_graph.py b/tradingagents/graph/trading_graph.py
index cb649f53..41c0bdf1 100644
--- a/tradingagents/graph/trading_graph.py
+++ b/tradingagents/graph/trading_graph.py
@@ -147,11 +147,16 @@ class TradingAgentsGraph:
             reasoning_effort = self.config.get("openai_reasoning_effort")
             if reasoning_effort:
                 kwargs["reasoning_effort"] = reasoning_effort
-# Ton nouveau code pour Anthropic
-        elif self.provider == "anthropic" and self.config.get("anthropic_thinking_budget"):
-             kwargs["thinking_config"] = {"type": "enabled", "budget_tokens": int(self.config["anthropic_thinking_budget"])}
-        return kwargs
 
+        elif provider == "anthropic":
+            thinking_budget = self.config.get("anthropic_thinking_budget")
+            if thinking_budget:
+                kwargs["thinking_config"] = {
+                    "type": "enabled", 
+                    "budget_tokens": int(thinking_budget)
+                }
+
+        return kwargs
     def _create_tool_nodes(self) -> Dict[str, ToolNode]:
         """Create tool nodes for different data sources using abstract methods."""
         return {