This commit is contained in:
yumoqing 2025-06-21 11:22:35 +08:00
parent 652349f4fa
commit ff802a1755

View File

@ -34,7 +34,7 @@ class BaseReranker:
def build_sys_prompt(self, sys_prompt): def build_sys_prompt(self, sys_prompt):
return f"<|im_start|>system\n{sys_prompt}\n<|im_end|>" 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>: {instruct}\n<Query>:{query}\n<Document>:\n{doc}<|im_end|>' return f'<|im_start|>user\n<Instruct>: {instruct}\n<Query>:{query}\n<Document>:\n{doc}<|im_end|>'
def build_assistant_prompt(self): def build_assistant_prompt(self):