cs.CLMay 4, 2026

FlexSQL: Flexible Exploration and Execution Make Better Text-to-SQL Agents

Authors: Quang Hieu PhamYang HePing NieCanwen XuDavood RafieiYuepeng WangXi YeJocelyn Qiaochu Chen

Organizations: University of Alberta · Simon Fraser University · University of Waterloo · Snowflake

Abstract

Text-to-SQL over large analytical databases requires navigating complex schemas, resolving ambiguous queries, and grounding decisions in actual data. Most current systems follow a fixed pipeline where schema elements are retrieved once upfront and the database is only revisited for post-hoc repair, limiting recovery from early mistakes. We present FlexSQL, a text-to-SQL agent whose core design principle is flexible database interaction: the agent can explore schema structure, inspect data values, and run verification queries at any point during reasoning. FlexSQL generates diverse execution plans to cover multiple query interpretations, implements each plan in either SQL or Python depending on the task, and uses a two-tiered repair mechanism that can backtrack from code-level errors to plan-level revisions. On Spider2-Snow, using gpt-oss-120b, FlexSQL achieves a 65.4% score, outperforming strong open-source baselines that use stronger, larger models such as gpt-o3 and DeepSeek-R1. When integrated into a general-purpose coding agent (as skills in Claude Code), our approach yields over 10% relative improvement on Spider2-Snow. Further analysis shows that flexible exploration and flexible execution jointly contribute to the effectiveness of our approach, highlighting flexibility as a key design principle. Our code is available at: https://github.com/StringNLPLAB/FlexSQL

Explore similar work

CardsList
  1. PExA: Parallel Exploration Agent for Complex Text-to-SQL

    Apr 24, 2026Tanmay Parekh, Ella Hofmann-Coyle, Shuyi Wang +3Text-To-SqlSql

  2. DualSQL: Text-to-SQL with Multi-Agent Reinforcement Learning

    Sep 16, 2026Shijie Chen, Yu Gan, Yeounoh Chung +6Text-To-SqlSql