Skip to content

Commit 5e4057b

Browse files
Use max_completion_tokens for newer OpenAI models
Newer models (5.x+) reject max_tokens and require max_completion_tokens instead. Convert in _create(). Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent ef32bf1 commit 5e4057b

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

queries/util/openai_gpt.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,9 @@ def _create(cls, *args: Any, **kwargs: Any) -> ChatCompletion:
8282
"""
8383
Creates a new completion while handling formatting and parsing.
8484
"""
85+
# Newer OpenAI models use max_completion_tokens instead of max_tokens
86+
if "max_tokens" in kwargs:
87+
kwargs["max_completion_tokens"] = kwargs.pop("max_tokens")
8588
return client.chat.completions.create(*args, model=MODEL, **kwargs)
8689

8790
@classmethod

0 commit comments

Comments
 (0)