Skip to content

Commit 9e85e3e

Browse files
authored
Drop unsupported params for litellm models (mlflow#20045)
Signed-off-by: SomtochiUmeh <[email protected]>
1 parent 5e0fa71 commit 9e85e3e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

mlflow/genai/simulators/simulator.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -182,6 +182,8 @@ def _invoke_model_without_tracing(
182182
if inference_params:
183183
kwargs.update(inference_params)
184184

185+
# Drop unsupported params (e.g., temperature=0 for gpt-5)
186+
litellm.drop_params = True
185187
response = litellm.completion(**kwargs)
186188
return response.choices[0].message.content
187189

0 commit comments

Comments
 (0)