[Dataset] [Tool call] Add When2Call single-turn tool-use dataset

docs/cli/response_regeneration.md

@@ -124,6 +124,7 @@ The text presets from the shared dataset registry (`DATASET_CONFIGS` in `specula
 | `magpie`            | `Magpie-Align/Magpie-Llama-3.1-Pro-300K-Filtered` | `train`       |
 | `nemotron`          | `nvidia/Llama-Nemotron-Post-Training-Dataset`     | `chat`        |
 | `open-perfectblend` | `mlabonne/open-perfectblend`                      | `train`       |
+| `when2call`         | `nvidia/When2Call`                                | `train`       |
 
 The registry's multimodal preset, `sharegpt4v_coco`, is **off-policy only** and `--dataset` rejects it. Its turns carry image content parts, which the Chat Completions API rejects, and the pre-tokenized output row has nowhere to keep pixel data. Use it with `prepare-data`.
 

src/speculators/data_generation/configs.py

@@ -49,6 +49,12 @@ def _normalize_nemotron(example: dict) -> dict:
     }
 
 
+def _normalize_when2call(example: dict) -> dict:
+    # When2Call already stores the dialogue as role/content ``messages``;
+    # the parallel ``tools`` column is preserved by the map for tool-aware runs.
+    return {"conversations": example["messages"]}
+
+
 def get_coco_dir():
     return os.getenv("COCO_DIR") or "coco/"
 
@@ -151,4 +157,11 @@ def _normalize_sharegpt4v_coco(example: dict) -> dict:
         hf_path="mlabonne/open-perfectblend",
         split="train",
     ),
+    "when2call": DatasetConfig(
+        name="when2call",
+        hf_path="nvidia/When2Call",
+        subset="train_sft",
+        split="train",
+        normalize_fn=_normalize_when2call,
+    ),
 }