mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-23 00:41:40 +00:00
Fix linting
This commit is contained in:
parent
bf17a6d1c9
commit
1a9b44ab1f
1 changed files with 3 additions and 2 deletions
|
|
@ -136,6 +136,9 @@ def rerank( # noqa: PLR0915
|
|||
api_key=optional_params.api_key,
|
||||
)
|
||||
|
||||
if isinstance(optional_params.timeout, str):
|
||||
optional_params.timeout = float(optional_params.timeout)
|
||||
|
||||
# Handle custom providers registered via custom_provider_map before
|
||||
# attempting to cast the provider string to the LlmProviders enum.
|
||||
if _custom_llm_provider in litellm._custom_providers:
|
||||
|
|
@ -191,8 +194,6 @@ def rerank( # noqa: PLR0915
|
|||
non_default_params=kwargs,
|
||||
)
|
||||
verbose_logger.info(f"optional_rerank_params: {optional_rerank_params}")
|
||||
if isinstance(optional_params.timeout, str):
|
||||
optional_params.timeout = float(optional_params.timeout)
|
||||
|
||||
model_response = RerankResponse()
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue