OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-06-26 18:15:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: Add Claude-4 thinking support

Browse Source 
- Support thinking for claude-sonnet-4 and claude-opus-4
- Add get_thinking_supported_models() for better maintainability
This commit is contained in:
Takahiro Kikumoto 2025-05-31 16:49:37 +09:00
parent adec65727e
commit 25ddc603e8
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
examples/pipelines/providers/anthropic_manifold_pipeline.py
View File

@ -68,6 +68,14 @@ class Pipeline:
{"id": "claude-sonnet-4-20250514", "name": "claude-4-sonnet"}, {"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): async def on_startup(self):
print(f"on_startup:{__name__}") print(f"on_startup:{__name__}")
pass pass
@ -168,7 +176,7 @@ class Pipeline:
} }
if body.get("stream", False): 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") reasoning_effort = body.get("reasoning_effort", "none")
budget_tokens = REASONING_EFFORT_BUDGET_TOKEN_MAP.get(reasoning_effort) budget_tokens = REASONING_EFFORT_BUDGET_TOKEN_MAP.get(reasoning_effort)