浏览代码

fix: title generation issue

Timothy Jaeryang Baek 4 月之前
父节点
当前提交
23bf71022e
共有 1 个文件被更改,包括 2 次插入3 次删除
  1. 2 3
      backend/open_webui/routers/openai.py

+ 2 - 3
backend/open_webui/routers/openai.py

@@ -613,9 +613,8 @@ async def generate_chat_completion(
             payload["max_tokens"] = payload["max_completion_tokens"]
             del payload["max_completion_tokens"]
 
-    # TODO: check if below is needed
-    # if "max_tokens" in payload and "max_completion_tokens" in payload:
-    #     del payload["max_tokens"]
+    if "max_tokens" in payload and "max_completion_tokens" in payload:
+        del payload["max_tokens"]
 
     # Convert the modified body back to JSON
     payload = json.dumps(payload)