Debug Smarter, Not Harder: AI Agents for Error Resolution in Computational Notebooks

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Grotov, Konstantin, Borzilov, Artem, Krivobok, Maksim, Bryksin, Timofey, Zharov, Yaroslav
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929550163705856
author Grotov, Konstantin
Borzilov, Artem
Krivobok, Maksim
Bryksin, Timofey
Zharov, Yaroslav
author_facet Grotov, Konstantin
Borzilov, Artem
Krivobok, Maksim
Bryksin, Timofey
Zharov, Yaroslav
contents Computational notebooks became indispensable tools for research-related development, offering unprecedented interactivity and flexibility in the development process. However, these benefits come at the cost of reproducibility and an increased potential for bugs. With the rise of code-fluent Large Language Models empowered with agentic techniques, smart bug-fixing tools with a high level of autonomy have emerged. However, those tools are tuned for classical script programming and still struggle with non-linear computational notebooks. In this paper, we present an AI agent designed specifically for error resolution in a computational notebook. We have developed an agentic system capable of exploring a notebook environment by interacting with it -- similar to how a user would -- and integrated the system into the JetBrains service for collaborative data science called Datalore. We evaluate our approach against the pre-existing single-action solution by comparing costs and conducting a user study. Users rate the error resolution capabilities of the agentic system higher but experience difficulties with UI. We share the results of the study and consider them valuable for further improving user-agent collaboration.
format Preprint
id arxiv_https___arxiv_org_abs_2410_14393
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Debug Smarter, Not Harder: AI Agents for Error Resolution in Computational Notebooks
Grotov, Konstantin
Borzilov, Artem
Krivobok, Maksim
Bryksin, Timofey
Zharov, Yaroslav
Machine Learning
Artificial Intelligence
Computational notebooks became indispensable tools for research-related development, offering unprecedented interactivity and flexibility in the development process. However, these benefits come at the cost of reproducibility and an increased potential for bugs. With the rise of code-fluent Large Language Models empowered with agentic techniques, smart bug-fixing tools with a high level of autonomy have emerged. However, those tools are tuned for classical script programming and still struggle with non-linear computational notebooks. In this paper, we present an AI agent designed specifically for error resolution in a computational notebook. We have developed an agentic system capable of exploring a notebook environment by interacting with it -- similar to how a user would -- and integrated the system into the JetBrains service for collaborative data science called Datalore. We evaluate our approach against the pre-existing single-action solution by comparing costs and conducting a user study. Users rate the error resolution capabilities of the agentic system higher but experience difficulties with UI. We share the results of the study and consider them valuable for further improving user-agent collaboration.
title Debug Smarter, Not Harder: AI Agents for Error Resolution in Computational Notebooks
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2410.14393