diff --git a/llmengine/rerank.py b/llmengine/rerank.py index fc7c3c6..a817a6b 100644 --- a/llmengine/rerank.py +++ b/llmengine/rerank.py @@ -74,7 +74,7 @@ async def rerank(request, params_kw, *params, **kw): raise e if isinstance(documents, str): documents = [documents] - top_n = parmas_kw.top_n + top_n = params_kw.top_n arr = await f(query, params_kw.documents, top_n = top_n) debug(f'{arr=}, type(arr)') return arr