> For clean Markdown of any page, append .md to the page URL.
> For a complete documentation index, see https://fr.nvidia-localization.ferndocs.com/fr/nemo/rl/latest/nemo-rl-data-hf-datasets-oai-format-dataset/llms.txt.
> For AI client integration (Claude Code, Cursor, etc.), connect to the MCP server at https://fr.nvidia-localization.ferndocs.com/_mcp/server.

# nemo\_rl.data.hf\_datasets.oai\_format\_dataset

## Contenu du module

### Classes

| Nom                                                                               | Description                                                                    |
| --------------------------------------------------------------------------------- | ------------------------------------------------------------------------------ |
| [`OpenAIFormatDataset`](#nemorldatahfdatasetsoaiformatdatasetopenaiformatdataset) | Cette classe est utilisée pour charger un jeu de données SFT au format OpenAI. |

### API

```python
class nemo_rl.data.hf_datasets.oai_format_dataset.OpenAIFormatDataset(train_ds_path: str, val_ds_path: str, chat_key: str = 'messages', system_key: str | None = None, system_prompt: str | None = None)
```

Cette classe est utilisée pour charger un jeu de données SFT au format OpenAI.

Le jeu de données doit être au format suivant :
\{
"messages": \[
\{"role": "system", "content": "Vous êtes un assistant utile."},
\{"role": "user", "content": "Quelle est la capitale de la France ?"},
\{"role": "assistant", "content": "La capitale de la France est Paris."}
]
}
system\_key et system\_prompt sont facultatifs. S'ils sont fournis, ils seront ajoutés au
début du jeu de données.
chat\_key doit être la clé de la liste des messages. Les conversations à plusieurs tours
sont prises en charge.
Le dernier message de la conversation doit provenir de l'assistant.

```python
add_messages_key(example: dict[str, typing.Any]) -> dict[str, list[dict[str, typing.Any]]]
```