AIAny
Icon for item

DataFlow-Harness: A Grounded Code-Agent Platform for Constructing Editable LLM Data Pipelines

Guides an LLM agent to build persistent, editable DAG-based data pipelines via typed, incremental mutations instead of free-form scripts. Combines DataFlow-Skills, a Model Context Protocol exposing live operator registry and pipeline state, and a synchronized Web UI; achieves 93.3% end-to-end pass rate on a 12-task benchmark while cutting cost and latency versus script baselines.

Introduction

Why this matters

LLM-driven coding agents often emit throwaway scripts that are hard to maintain, compose, or inspect — a gap this paper calls the NL2Pipeline gap. DataFlow-Harness reframes pipeline construction as a sequence of typed, platform-grounded mutations so the agent produces native, editable DAG artifacts rather than ephemeral code blobs. That shift targets the two practical barriers preventing LLMs from reliably engineering production data workflows: artifact persistence and live platform grounding.

Key Findings
  • Grounded construction workflow: By exposing a Model Context Protocol (MCP) that surfaces the live operator registry and current pipeline state, the system lets an LLM issue small, typed edits to a DAG. This reduces brittleness versus free-form script generation, so the artifact is directly editable in-platform.
  • Skills and procedural priors matter: DataFlow-Skills provide procedural guidance (common sequences and domain knowledge) that meaningfully improve success when construction depends on implicit operational steps; tasks needing procedural knowledge see the largest gains.
  • Strong empirical results with cost/latency trade-offs: On a 12-task data-engineering benchmark the platform attained a 93.3% observed end-to-end pass rate. Compared to a vanilla script-generation baseline, it reduced measured monetary cost by ~72.5% and generation latency by ~49.9%, while staying within ~0.9 percentage points of a context-aware model baseline at lower cost.
Who It's For and Trade-offs

Great fit if you build systems where LLMs must produce maintainable, inspectable data workflows (data engineering teams, ML platform engineers, or product teams integrating agentic automation into pipelines). The platform is especially useful when you need persistent artifacts, visual editing, and lower operational cost for agent-driven construction.

Look elsewhere if your goal is raw peak-performance automation where free-form script generation and maximum model creativity are acceptable; DataFlow-Harness trades some generative freedom and complexity for structured, platform-native artifacts and lower downstream maintenance cost. It also presumes integration with a platform that can represent DAGs and support the MCP — adopting it requires platform-side instrumentation.

Where It Fits

This paper sits at the intersection of AI agents, MLOps, and software engineering for data pipelines. It should be read alongside work on agent tool-use, program synthesis for production systems, and pipeline orchestration, as it prioritizes artifact editability and live grounding over unconstrained code generation.

How it works (brief)

The system combines three pieces: typed mutation primitives (so edits are small and verifiable), a Skills library encoding common procedural patterns, and an MCP layer that provides the model with live, queryable platform state. A synchronized Web UI mirrors the conversational authoring into a visual DAG editor so human reviewers can inspect and iteratively edit artifacts produced by the agent.

Information

  • Websitearxiv.org
  • AuthorsRunming He, Zhen Hao Wong, Hao Liang, Zimo Meng, Chengyu Shen, Xiaochen Ma, Wentao Zhang
  • Published date2026/07/18

More Items

A PyTorch-native training framework for agentic reinforcement learning research that keeps researcher-facing code compact and editable. Uses an asynchronous loop to train multimodal and mixture-of-experts policies while never training on tokens the agent didn't generate; matches Megatron-style stacks under a comparable protocol and ships recipes and containers on GitHub.

Presents Skill Self-Play (Skill-SP), a co-evolutionary training loop where a proposer, solver, and dynamic skill controller generate, solve, and verify tasks conditioned on reusable skills — balancing verifiable execution with open-ended task diversity to boost LLM tool-use and reasoning.

Alternates targeted research and constraint-wise audits to recursively improve long-horizon answers: an inner loop gathers evidence and drafts solutions, an outer loop audits unresolved claims and launches focused follow-ups. Trains 4B dense and 122B-A10B MoE agents with long-horizon RL and agentic mid-training, outperforming comparable-scale baselines on multi-step research benchmarks.