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):