ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-13 12:55:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-08-20 21:05:04 +04:00
parent d2d23f4abd
commit ba972ecd52
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
backend/open_webui/routers/openai.py
View file

@ -702,6 +702,10 @@ def get_azure_allowed_params(api_version: str) -> set[str]:
return allowed_params return allowed_params
def is_openai_reasoning_model(model: str) -> bool:
return model.lower().startswith(("o1", "o3", "o4", "gpt-5"))
def convert_to_azure_payload(url, payload: dict, api_version: str): def convert_to_azure_payload(url, payload: dict, api_version: str):
model = payload.get("model", "") model = payload.get("model", "")
@ -709,7 +713,7 @@ def convert_to_azure_payload(url, payload: dict, api_version: str):
allowed_params = get_azure_allowed_params(api_version) allowed_params = get_azure_allowed_params(api_version)
# Special handling for o-series models # Special handling for o-series models
if model.startswith("o") and model.endswith("-mini"): if is_openai_reasoning_model(model):
# Convert max_tokens to max_completion_tokens for o-series models # Convert max_tokens to max_completion_tokens for o-series models
if "max_tokens" in payload: if "max_tokens" in payload:
payload["max_completion_tokens"] = payload["max_tokens"] payload["max_completion_tokens"] = payload["max_tokens"]
@ -815,10 +819,7 @@ async def generate_chat_completion(
key = request.app.state.config.OPENAI_API_KEYS[idx] key = request.app.state.config.OPENAI_API_KEYS[idx]
# Check if model is a reasoning model that needs special handling # Check if model is a reasoning model that needs special handling
is_reasoning_model = ( if is_openai_reasoning_model(payload["model"]):
payload["model"].lower().startswith(("o1", "o3", "o4", "gpt-5"))
)
if is_reasoning_model:
payload = openai_reasoning_model_handler(payload) payload = openai_reasoning_model_handler(payload)
elif "api.openai.com" not in url: elif "api.openai.com" not in url:
# Remove "max_completion_tokens" from the payload for backward compatibility # Remove "max_completion_tokens" from the payload for backward compatibility