diff --git a/litellm/llms/nvidia_nim/rerank/transformation.py b/litellm/llms/nvidia_nim/rerank/transformation.py index c014064485b..37480ef1076 100644 --- a/litellm/llms/nvidia_nim/rerank/transformation.py +++ b/litellm/llms/nvidia_nim/rerank/transformation.py @@ -210,7 +210,7 @@ class NvidiaNimRerankConfig(BaseRerankConfig): if "documents" not in optional_rerank_params: raise ValueError("documents is required for Nvidia NIM rerank") - top_n: Final[object] = cast(object, optional_rerank_params.get("top_n")) # cast-ok: untyped provider parameter is validated below + top_n: Final[object] = cast(object, optional_rerank_params.get("top_n")) # cast-ok: value validated below if top_n is not None: if isinstance(top_n, bool) or not isinstance(top_n, int) or top_n < 1: raise ValueError(f"top_n must be a positive integer, got: {top_n!r}")