Merge pull request #522 from kikumoto/feature/add_suuport_thinking_for_sonnet4_opus4

feat: Add Claude-4 thinking support

Author: tjbck

examples/pipelines/providers/anthropic_manifold_pipeline.py

@@ -68,6 +68,14 @@ def get_anthropic_models(self):
             {"id": "claude-sonnet-4-20250514", "name": "claude-4-sonnet"},
         ]
 
+    def get_thinking_supported_models(self):
+        """Returns list of model identifiers that support extended thinking"""
+        return [
+            "claude-3-7",
+            "claude-sonnet-4",
+            "claude-opus-4"
+        ]
+
     async def on_startup(self):
         print(f"on_startup:{__name__}")
         pass
@@ -168,7 +176,7 @@ def pipe(
             }
 
             if body.get("stream", False):
-                supports_thinking = "claude-3-7" in model_id
+                supports_thinking = any(model in model_id for model in self.get_thinking_supported_models())
                 reasoning_effort = body.get("reasoning_effort", "none")
                 budget_tokens = REASONING_EFFORT_BUDGET_TOKEN_MAP.get(reasoning_effort)
 

examples/pipelines/providers/aws_bedrock_claude_pipeline.py

@@ -50,14 +50,6 @@ def __init__(self):
         # self.id = "openai_pipeline"
         self.name = "Bedrock: "
 
-        self.valves = self.Valves(
-            **{
-                "AWS_ACCESS_KEY": os.getenv("AWS_ACCESS_KEY", "your-aws-access-key-here"),
-                "AWS_SECRET_KEY": os.getenv("AWS_SECRET_KEY", "your-aws-secret-key-here"),
-                "AWS_REGION_NAME": os.getenv("AWS_REGION_NAME", "your-aws-region-name-here"),
-            }
-        )
-
         self.valves = self.Valves(
             **{
                 "AWS_ACCESS_KEY": os.getenv("AWS_ACCESS_KEY", ""),
@@ -72,6 +64,13 @@ def __init__(self):
 
         self.update_pipelines()
 
+    def get_thinking_supported_models(self):
+        """Returns list of model identifiers that support extended thinking"""
+        return [
+            "claude-3-7",
+            "claude-sonnet-4",
+            "claude-opus-4"
+        ]
 
     async def on_startup(self):
         # This function is called when the server is started.
@@ -183,13 +182,14 @@ def pipe(
                        }
 
             if body.get("stream", False):
-                supports_thinking = "claude-3-7" in model_id
+                supports_thinking = any(model in model_id for model in self.get_thinking_supported_models())
                 reasoning_effort = body.get("reasoning_effort", "none")
                 budget_tokens = REASONING_EFFORT_BUDGET_TOKEN_MAP.get(reasoning_effort)
 
                 # Allow users to input an integer value representing budget tokens
                 if (
                     not budget_tokens
+                    and reasoning_effort is not None
                     and reasoning_effort not in REASONING_EFFORT_BUDGET_TOKEN_MAP.keys()
                 ):
                     try: