scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-04 07:27:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 195

fix(nvidia_nim): declare retrieval top_n as Final[object] for type gate

Browse source
This commit is contained in:
Ivan Konstantinov 2026-10-19 11:34:46 +03:00
parent 27a860bcc5
commit c56f093ccc
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/nvidia_nim/rerank/transformation.py
View file

@ -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: object = optional_rerank_params.get("top_n")
top_n: Final[object] = optional_rerank_params.get("top_n")
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}")