get_sharegpt_messages

torchtune.data.get_sharegpt_messages(sample: Mapping[str, Any], train_on_input: bool = False) → List[Message]

Convert a chat sample adhering to the ShareGPT json structure to torchtune’s Message structure.

ShareGPT follows:

{
    "conversations": [
        {
            "from": <system|human|gpt>,
            "value": <message>,
        },
        ...
    ]
}

Message follows:

[
    {
        "role": <system|user|assistant>,
        "content": <message>,
    },
    ...
]

Parameters:

• sample (Mapping[str, Any]) – a single data sample with “conversations” field pointing to a list of dict messages.

• train_on_input (bool) – whether the prompt should remain unmasked. Default: False

Returns:

A list of messages with “role” and “content” fields.

Return type:

List[Message]