from data_juicer.ops.base_op import OPERATORS
from data_juicer.ops.mapper.calibrate_qa_mapper import CalibrateQAMapper
OP_NAME = 'calibrate_response_mapper'
# TODO: LLM-based inference.
[docs]
@OPERATORS.register_module(OP_NAME)
class CalibrateResponseMapper(CalibrateQAMapper):
"""
Mapper to calibrate response in question-answer pairs based on reference text.
""" # noqa: E501
DEFAULT_SYSTEM_PROMPT = '请根据提供的【参考信息】对问答对中的【回答】进行校准,\
使其更加详细、准确,且仍可以回答原问题。只输出校准后的回答,不要输出多余内容。'
[docs]
def parse_output(self, raw_output):
return None, raw_output.strip()