ConversationTemplate

从指令和响应生成对话模板。

输入 & 输出列

graph TD
    subgraph Dataset
        subgraph Columns
            ICOL0[instruction]
            ICOL1[response]
        end
        subgraph New columns
            OCOL0[conversation]
        end
    end

    subgraph ConversationTemplate
        StepInput[Input Columns: instruction, response]
        StepOutput[Output Columns: conversation]
    end

    ICOL0 --> StepInput
    ICOL1 --> StepInput
    StepOutput --> OCOL0
    StepInput --> StepOutput

输入

• instruction (str): 要在对话中使用的指令。

• response (str): 要在对话中使用的响应。

输出

• conversation (ChatType): 对话模板。

示例

从指令和响应创建对话

from distilabel.steps import ConversationTemplate

conv_template = ConversationTemplate()
conv_template.load()

result = next(
    conv_template.process(
        [
            {
                "instruction": "Hello",
                "response": "Hi",
            }
        ],
    )
)
# >>> result
# [{'instruction': 'Hello', 'response': 'Hi', 'conversation': [{'role': 'user', 'content': 'Hello'}, {'role': 'assistant', 'content': 'Hi'}]}]