Empirical evaluation of LLMs capabilities for data pipeline generation on Databricks platform

Read full text
Authors: Tobia Martina, Motaz Saad, Chiara Rucco, Antonella Longo
Cited by: 1
References: 5

Abstract

Building production-ready data pipelines on enterprise platforms requires expertise spanning distributed computing, data integration, and platform-specific tooling — a process that Large Language Models (LLMs) may help accelerate. This paper presents an empirical evaluation of seven leading LLMs — from the GPT-5, Claude 4, and Qwen3 families — on their ability to generate functional PySpark data pipelines for the Databricks platform. Each model was evaluated across three progressively complex ETL scenarios (data quality validation, temporal aggregation, and multi-source integration) using a fix-count metric measuring iterative LLM-driven error corrections. Results show that Claude Opus 4 and GPT-5 mini were the most reliable, requiring zero or one fix across all prompts, while Claude 3.5 Haiku required up to eight fixes on the most complex scenario. Code style compliance scores (Pylint/PEP8) proved orthogonal to functional correctness, with some models producing fully functional pipelines despite low style scores. These findings provide practitioners with empirical guidance on model selection for LLM-assisted pipeline development, while identifying current capability boundaries that require human oversight.

Fields of Study & Subjects

Advanced Database Systems and Queries Cloud Computing and Resource Management Advanced Data Storage Technologies

Publication details

Journal / source
Future Generation Computer Systems
Publisher
Elsevier BV
Volume
183
Pages
108587–108587
Publication date
16 May 2026
Language
EN