Eval v6: restore _fetch_first_rows

eval_securecoder.py

@@ -178,6 +178,23 @@ def _messages_from_any(row, kind):
                 messages.append({"role": "user", "content": str(content)})
                 break
     return messages, tools
+def _fetch_first_rows(repo: str, config: str | None, split: str, n: int) -> list[dict]:
+    """Stream up to ``n`` rows from a Hub dataset. ``config`` is the dataset config
+    name (e.g. ``func_calling`` for Hermes FC); pass ``None`` for datasets that
+    have only the default config."""
+    from datasets import load_dataset
+    kwargs: dict[str, Any] = {"split": split, "streaming": True}
+    if config:
+        kwargs["name"] = config
+    ds = load_dataset(repo, token=os.environ.get("HF_TOKEN"), **kwargs)
+    out = []
+    for row in ds:
+        out.append(dict(row))
+        if len(out) >= n:
+            break
+    return out
+
+
 def _build_tool_prompts(rows: list[dict]) -> list[dict]:
     """Build (prompt, tools) tuples via the shared _messages_from_any helper.
     Hermes rows use ``conversations`` (from/value); OpenAI-style use ``messages``