|
@@ -617,6 +617,10 @@ class OpenAILargeLanguageModel(_CommonOpenAI, LargeLanguageModel):
|
|
# o1 compatibility
|
|
# o1 compatibility
|
|
block_as_stream = False
|
|
block_as_stream = False
|
|
if model.startswith("o1"):
|
|
if model.startswith("o1"):
|
|
|
|
+ if "max_tokens" in model_parameters:
|
|
|
|
+ model_parameters["max_completion_tokens"] = model_parameters["max_tokens"]
|
|
|
|
+ del model_parameters["max_tokens"]
|
|
|
|
+
|
|
if stream:
|
|
if stream:
|
|
block_as_stream = True
|
|
block_as_stream = True
|
|
stream = False
|
|
stream = False
|