phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: fix linting error

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-13 14:33:19 -07:00
parent a2414d09c1
commit 1cd57e95aa
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/utils.py
View file

@ -2451,8 +2451,11 @@ def get_optional_params_image_gen(
config_class = ( config_class = (
litellm.AmazonStability3Config litellm.AmazonStability3Config
if litellm.AmazonStability3Config._is_stability_3_model(model=model) if litellm.AmazonStability3Config._is_stability_3_model(model=model)
else litellm.AmazonNovaCanvasConfig if litellm.AmazonNovaCanvasConfig._is_nova_model(model=model) else (
else litellm.AmazonStabilityConfig litellm.AmazonNovaCanvasConfig
if litellm.AmazonNovaCanvasConfig._is_nova_model(model=model)
else litellm.AmazonStabilityConfig
)
) )
supported_params = config_class.get_supported_openai_params(model=model) supported_params = config_class.get_supported_openai_params(model=model)
_check_valid_arg(supported_params=supported_params) _check_valid_arg(supported_params=supported_params)
@ -3948,7 +3951,9 @@ def _count_characters(text: str) -> int:
def get_response_string(response_obj: Union[ModelResponse, ModelResponseStream]) -> str: def get_response_string(response_obj: Union[ModelResponse, ModelResponseStream]) -> str:
_choices: List[Union[Choices, StreamingChoices]] = response_obj.choices _choices: Union[List[Union[Choices, StreamingChoices]], List[StreamingChoices]] = (
response_obj.choices
)
response_str = "" response_str = ""
for choice in _choices: for choice in _choices: