Browse Source

fix: logit_bias in workspace models

dannyl1u 2 months ago
parent
commit
3c6a834ff4
1 changed files with 4 additions and 0 deletions
  1. 4 0
      backend/open_webui/routers/openai.py

+ 4 - 0
backend/open_webui/routers/openai.py

@@ -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