From ff802a17553a84fa722f90f153b51aefec5de5a4 Mon Sep 17 00:00:00 2001 From: yumoqing Date: Sat, 21 Jun 2025 11:22:35 +0800 Subject: [PATCH] bugfix --- llmengine/base_reranker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llmengine/base_reranker.py b/llmengine/base_reranker.py index 8ef86a2..39a0570 100644 --- a/llmengine/base_reranker.py +++ b/llmengine/base_reranker.py @@ -34,7 +34,7 @@ class BaseReranker: def build_sys_prompt(self, sys_prompt): return f"<|im_start|>system\n{sys_prompt}\n<|im_end|>" - def build_user_prompt(self, query, docs, instruct=''): + def build_user_prompt(self, query, doc, instruct=''): return f'<|im_start|>user\n: {instruct}\n:{query}\n:\n{doc}<|im_end|>' def build_assistant_prompt(self):