From 405d0561df25ba6452a02e81f117f8b3d59bbabc Mon Sep 17 00:00:00 2001 From: kivvi Date: Thu, 26 Sep 2024 08:49:59 +0800 Subject: [PATCH] Fix: o1 input parameter must be max_completion_tokens --- backend/open_webui/apps/openai/main.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/backend/open_webui/apps/openai/main.py b/backend/open_webui/apps/openai/main.py index 93155063f..e0a40a1f5 100644 --- a/backend/open_webui/apps/openai/main.py +++ b/backend/open_webui/apps/openai/main.py @@ -415,6 +415,9 @@ async def generate_chat_completion( payload["max_tokens"] = payload["max_completion_tokens"] del payload["max_completion_tokens"] else: + if payload["model"].lower().startswith("o1-") and "max_tokens" in payload: + payload["max_completion_tokens"] = payload["max_tokens"] + del payload["max_tokens"] if "max_tokens" in payload and "max_completion_tokens" in payload: del payload["max_tokens"]