CoolFace
Apppublic

Kalletlamadhav/sql_optimized_env_new

sourceHugging Faceupdated 5mo agoView on Hugging Face
0likes
App README

๐Ÿ—„๏ธ SQL Query Optimization RL Environment

An OpenEnv-compliant Reinforcement Learning environment for training agents to automatically optimize SQL queries on real-world Indian government-scale datasets.


๐Ÿš€ Why This Project?

Modern databases often suffer from:

  • โ€”Slow queries due to poor indexing
  • โ€”Inefficient joins and anti-patterns
  • โ€”High latency on large-scale datasets

This environment enables: ๐Ÿ‘‰ Training AI agents to detect and fix SQL inefficiencies automatically ๐Ÿ‘‰ Benchmarking query optimization strategies in a realistic setting


๐Ÿ›๏ธ Data Domains (Realistic Scale)

DomainTablesDescription
GSTgst_invoice_records, gst_invoice_itemsB2B invoices, HSN items, tax splits
PDSration_card_beneficiaries, pds_allotmentsRation distribution system
Railwayrailway_trains, railway_pnr_bookingsTicket booking workloads
MGNREGAmgnrega_workers, mgnrega_attendance, mgnrega_paymentsRural employment data

๐ŸŽฏ Task Design (9 Tasks ร— 5 Levels)

LevelTasksAnti-Patterns
1 Intropds_select_starSELECT_STAR
2 Easygst_missing_index, railway_simple_filterMISSING_INDEX
3 Mediumgst_n_plus_one, pds_cartesian, mgnrega_wildcardN+1, CARTESIAN, WILDCARD
4 Hardgst_multi_join, railway_tatkal_workloadMulti-pattern
5 Expertmgnrega_schema_eUNBOUNDED_AGG + CAST

โš™๏ธ System Architecture

text
Agent (LLM / RL Policy)
        โ†“
   OpenEnv API
        โ†“
SQL Execution Engine (SQLite)
        โ†“
Execution Plan + Metrics
        โ†“
Reward Function (5 factors)