Source code for data_juicer.ops.mapper.calibrate_query_mapper

from data_juicer.ops.base_op import OPERATORS
from data_juicer.ops.mapper.calibrate_qa_mapper import CalibrateQAMapper

OP_NAME = 'calibrate_query_mapper'


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