Team Ai
Apppublic

Kalletlamadhav/sql-optimization-env

sourceHugging Faceupdated 6mo agoView on Hugging Face
0likes
openenv.yaml121 linesDownload Raw Back to root
1spec_version: 12name: sql-optimization-env3version: "1.0.0"4 5description: >6  RL environment for training agents to optimise SQL queries on 4 Indian7  government schemas: GST, PDS (Ration), IRCTC (Railway), MGNREGA.8  9 tasks across 5 curriculum levels. Real SQLite execution with SEED=42.9 10type: space11runtime: fastapi12app: server.main:app13port: 786014 15tasks:16  - id: pds_select_star17    level: 118    pattern: SELECT_STAR19 20  - id: gst_missing_index21    level: 222    pattern: MISSING_INDEX23 24  - id: railway_simple_filter25    level: 226    pattern: MISSING_INDEX27 28  - id: gst_n_plus_one29    level: 330    pattern: N_PLUS_ONE31 32  - id: pds_cartesian33    level: 334    pattern: CARTESIAN_PRODUCT35 36  - id: mgnrega_wildcard37    level: 338    pattern: LEADING_WILDCARD39 40  - id: gst_multi_join41    level: 442    pattern: N_PLUS_ONE43 44  - id: railway_tatkal_workload45    level: 446    pattern: N_PLUS_ONE47 48  - id: mgnrega_schema_e49    level: 550    pattern: UNBOUNDED_AGGREGATION51 52observation_space:53  type: object54  properties:55    task_id:56      type: string57    goal:58      type: string59    schema_ddl:60      type: string61    current_query:62      type: string63    execution_plan:64      type: object65    execution_time_ms:66      type: number67    curriculum_level:68      type: integer69      minimum: 170      maximum: 571 72action_space:73  type: object74  properties:75    optimized_query:76      type: string77    identified_pattern:78      type: string79      enum:80        - N_PLUS_ONE81        - CARTESIAN_PRODUCT82        - MISSING_INDEX83        - SELECT_STAR84        - LEADING_WILDCARD85        - IMPLICIT_CAST86        - UNBOUNDED_AGGREGATION87        - NONE88    explanation:89      type: string90    index_statements:91      type: array92      items:93        type: string94 95graders:96  - name: speedup97    weight: 0.3598    range: [0.0, 0.35]99 100  - name: equivalence101    weight: 0.25102    range: [0.0, 0.25]103 104  - name: pattern105    weight: 0.20106    range: [0.0, 0.20]107 108  - name: index109    weight: 0.10110    range: [0.0, 0.10]111 112  - name: simplicity113    weight: 0.10114    range: [0.0, 0.10]115 116curriculum:117  levels: 5118  graduate_threshold: 0.70119  graduate_consecutive: 3120  remediate_threshold: 0.40121  remediate_consecutive: 2