📘 概述

ReasoningQuestionCategorySampleEvaluator 算子用于对用户问题进行多级分类（主分类和子分类）。通过大语言模型对输入问题进行语义分析，输出分类结果。

__init__函数

@prompt_restrict(
    MathQuestionCategoryPrompt
)

@OPERATOR_REGISTRY.register()
class ReasoningQuestionCategorySampleEvaluator(OperatorABC):
    def __init__(self, llm_serving: LLMServingABC = None):

init参数说明

参数名	类型	默认值	说明
llm_serving	LLMServingABC	None	大语言模型服务实例，用于执行推理与生成。

Prompt模板说明

Prompt 模板名称	主要用途	适用场景	特点说明
MathQuestionCategoryPrompt	问题多级分类	对用户问题进行主分类和子分类	输入问题，输出主分类和子分类

run函数

def run(self, storage: DataFlowStorage, input_key:str = "instruction", output_key:str="question_category")

参数

名称	类型	默认值	说明
storage	DataFlowStorage	必需	数据流存储实例，负责读取与写入数据。
input_key	str	"instruction"	输入列名，对应问题字段。
output_key	str	"question_category"	输出列名的基础键名，用于校验是否存在同名字段。

🧠 示例用法

from dataflow.operators.reasoning import ReasoningQuestionCategorySampleEvaluator
from dataflow.utils.storage import FileStorage
from dataflow.core import LLMServingABC
from dataflow.serving import APILLMServing_request

class ReasoningQuestionCategorySampleEvaluatorTest():
    def __init__(self, llm_serving: LLMServingABC = None):
        
        self.storage = FileStorage(
            first_entry_file_name="example.json",
            cache_path="./cache_local",
            file_name_prefix="dataflow_cache_step",
            cache_type="jsonl",
        )
        
        # use API server as LLM serving
        self.llm_serving = APILLMServing_request(
                    api_url="",
                    model_name="gpt-4o",
                    max_workers=30
        )
        
        self.evaluator = ReasoningQuestionCategorySampleEvaluator(llm_serving=self.llm_serving)
        
    def forward(self):
        self.evaluator.run(
            storage = self.storage.step(),
            input_key = "instruction",
            output_key = "category",
        )

if __name__ == "__main__":
    pl = ReasoningQuestionCategorySampleEvaluatorTest()
    pl.forward()

🧾 默认输出格式（Output Format）

字段	类型	说明
instruction	str	输入的问题文本。
primary_category	str	模型生成的主分类。
secondary_category	str	模型生成的子分类。