mlsanityData to 7 Oct 2026

kRAIG: A Natural Language-Driven Agent for Automated DataOps Pipeline Generation

arXiv 2603.20311 ↗First posted 19 Mar 2026cs.SE cs.AI cs.CLcited by 3

On the map: LLM agents and tool use → Web, search, and research agents

Abstract (arXiv)

Modern machine learning systems rely on complex data engineering workflows to extract, transform, and load (ELT) data into production pipelines. However, constructing these pipelines remains time-consuming and requires substantial expertise in data infrastructure and orchestration frameworks. Recent advances in large language model (LLM) agents offer a potential path toward automating these workflows, but existing approaches struggle with under-specified user intent, unreliable tool generation, and limited guarantees of executable outputs. We introduce kRAIG, an AI agent that translates natural language specifications into production-ready Kubeflow Pipelines (KFP). To resolve ambiguity in user intent, we propose ReQuesAct (Reason, Question, Act), an interaction framework that explicitly clarifies intent prior to pipeline synthesis. The system orchestrates end-to-end data movement from diverse sources and generates task-specific transformation components through a retrieval-augmented tool synthesis process. To ensure data quality and safety, kRAIG incorporates LLM-based validation stages that verify pipeline integrity prior to execution. Our framework achieves a 3x improvement in extraction and loading success and a 25 percent increase in transformation accuracy compared to state-of-the-art agentic baselines. These improvements demonstrate that structured agent workflows with explicit intent clarification and validation significantly enhance the reliability and executability of automated data engineering pipelines.

Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Web, search, and research agents, one of 16 directions in LLM agents and tool use. Builds on 10 of its 15 references that are in the map. First posted on arXiv 19 Mar 2026.

Builds on

10 of 15 references.

Closest ancestors · 1

Same direction, newest first

  1. 2025ELT-Bench: An End-to-End Benchmark for Evaluating AI Agents on ELT Pipelines7 Apr

Other references · 9

  1. 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,355
  2. 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,072
  3. 2024SWE-agent: Agent-Computer Interfaces Enable Automated Software Engineeringcited by 1,628
  4. 2020Experience Grounds Languagecited by 241
  5. 2024Spider 2.0: Evaluating Language Models on Real-World Enterprise Text-to-SQL Workflowscited by 223
  6. 2024STaR-GATE: Teaching Language Models to Ask Clarifying Questionscited by 99
  7. 2025QuestBench: Can LLMs ask the right question to acquire information in reasoning tasks?cited by 56
  8. 2023A Survey of the Evolution of Language Model-Based Dialogue Systems: Data, Task and Modelscited by 23
  9. 2025When LLMs Meet API Documentation: Can Retrieval Augmentation Aid Code Generation Just as It Helps Developers?cited by 12

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025DataFlow: An LLM-Driven Framework for Unified Data Preparation and Workflow Automation in the Era of Data-Centric AI
  2. 2025Text-to-Pipeline: Bridging Natural Language and Data Preparation Pipelines
  3. 2026DataFlow-Harness: A Grounded Code-Agent Platform for Constructing Editable LLM Data Pipelines
  4. 2025Data Agent: A Holistic Architecture for Orchestrating Data+AI Ecosystems
  5. 2025AutoStreamPipe: LLM Assisted Automatic Generation of Data Stream Processing Pipelines
  6. 2024AutoFlow: Automated Workflow Generation for Large Language Model Agents
  7. 2026From Research Question to Scientific Workflow: Leveraging Agentic AI for Science Automation
  8. 2026DataEvolver: Automatic Data Preparation for Large Language Models through Multi-Level Self-Evolving