nvidia rerank missed top n

langgenius · Sep 10, 2024 · 5809d96 · 5809d96
1 parent 72df152
commit 5809d96
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/api/core/model_runtime/model_providers/nvidia/rerank/rerank.py b/api/core/model_runtime/model_providers/nvidia/rerank/rerank.py
@@ -52,7 +52,6 @@ def _invoke(self, model: str, credentials: dict,
             payload = {
                 "model": model,
                 "query": {"text": query},
-                "top_n": 1,
                 "passages": [{"text": doc} for doc in docs],
             }
             session = requests.Session()