scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-11 08:38:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 280

fix(text-completion): keep body-only param split fully typed for the basedpyright gate

Browse source
This commit is contained in:
Prasanna Jain 2026-10-22 20:30:21 -04:00
parent 7836a54018
commit a4c5dfb055
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/llms/openai/completion/transformation.py
View file

@ -3,7 +3,7 @@ Support for gpt model family
"""
from collections.abc import Mapping
from typing import Final
from typing import Final, cast
from litellm.litellm_core_utils.hidden_params import set_hidden_param
from litellm.llms.base_llm.completion.transformation import BaseTextCompletionConfig
@ -161,10 +161,10 @@ class OpenAITextCompletionConfig(BaseTextCompletionConfig, OpenAIGPTConfig):
headers: dict,
) -> dict:
prompt: Final = transform_prompt(messages)
body_only: Final = {k: v for k, v in optional_params.items() if k in _BODY_ONLY_PARAMS}
params: Final = {k: v for k, v in optional_params.items() if k not in _BODY_ONLY_PARAMS}
params: Final = cast(dict[str, object], optional_params).copy()
body_only: Final = {key: params.pop(key) for key in _BODY_ONLY_PARAMS & params.keys()}
if body_only:
params["extra_body"] = {**(params.get("extra_body") or {}), **body_only}
params["extra_body"] = {**cast(Mapping[str, object], params.get("extra_body") or {}), **body_only}
return {
"model": model,
"prompt": prompt,