Merge pull request #11394 from dannyl1u/fix/logit-bias-workspace-models

fix: logit_bias bug in workspace models
This commit is contained in:
Timothy Jaeryang Baek 2025-03-08 12:11:53 -04:00 committed by GitHub
commit e4a79a0b0e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -36,6 +36,9 @@ from open_webui.utils.payload import (
apply_model_params_to_body_openai,
apply_model_system_prompt_to_body,
)
from open_webui.utils.misc import (
convert_logit_bias_input_to_json,
)
from open_webui.utils.auth import get_admin_user, get_verified_user
from open_webui.utils.access_control import has_access
@ -666,6 +669,7 @@ async def generate_chat_completion(
del payload["max_tokens"]
# Convert the modified body back to JSON
payload['logit_bias'] = json.loads(convert_logit_bias_input_to_json(payload['logit_bias']))
payload = json.dumps(payload)
r = None