optimizeqamapper

February 4, 2026 · View on GitHub

Mapper to optimize question-answer pairs.

This operator refines and enhances the quality of question-answer pairs. It uses a Hugging Face model to generate more detailed and accurate questions and answers. The input is formatted using a template, and the output is parsed using a regular expression. The system prompt, input template, and output pattern can be customized. If VLLM is enabled, the operator accelerates inference on CUDA devices.

优化问答对的映射器。

该算子改进和提高问答对的质量。它使用Hugging Face模型生成更详细和准确的问题和答案。输入使用模板格式化,输出通过正则表达式解析。可以自定义系统提示、输入模板和输出模式。如果启用了VLLM,算子将在CUDA设备上加速推理。

Type 算子类型: mapper

Tags 标签: gpu, vllm, hf, api

🔧 Parameter Configuration 参数配置

name 参数名type 类型default 默认值desc 说明
api_or_hf_model<class 'str'>'Qwen/Qwen2.5-7B-Instruct'API or huggingface model name.
is_hf_model<class 'bool'>TrueIf true, use huggingface model. Otherwise, use API.
api_endpointtyping.Optional[str]NoneURL endpoint for the API.
response_pathtyping.Optional[str]NonePath to extract content from the API response. Defaults to 'choices.0.message.content'.
system_prompttyping.Optional[str]NoneSystem prompt for guiding the optimization task.
input_templatetyping.Optional[str]NoneTemplate for building the input for the model. Please make sure the template contains one placeholder '{}', which corresponds to the question and answer pair generated by param qa_pair_template.
qa_pair_templatetyping.Optional[str]NoneTemplate for formatting the question and answer pair. Please make sure the template contains two '{}' to format question and answer.
output_patterntyping.Optional[str]NoneRegular expression pattern to extract question and answer from model response.
try_numtyping.Annotated[int, Gt(gt=0)]3The number of retry attempts when there is an API call error or output parsing error.
enable_vllm<class 'bool'>FalseWhether to use VLLM for inference acceleration.
model_paramstyping.Optional[typing.Dict]NoneParameters for initializing the model.
sampling_paramstyping.Optional[typing.Dict]NoneSampling parameters for text generation (e.g., {'temperature': 0.9, 'top_p': 0.95}).
kwargs''Extra keyword arguments.