Hariprita/nl2sql-openenv

1name: sql-agent-env2version: "1.0.0"3description: "Real-world SQL query generation environment for AI agent training"4author: "your-hf-username"5tags:6  - sql7  - natural-language-to-sql8  - real-world9  - database10  - text2sql11tasks:12  - id: simple_select13    name: "Simple SELECT with filter"14    difficulty: easy15    description: "Filter customers by country and return name + city"16    max_score: 1.017  - id: join_aggregation18    name: "JOIN with GROUP BY aggregation"19    difficulty: medium20    description: "Join order_items with products, compute revenue per category"21    max_score: 1.022  - id: window_ranking23    name: "Window function ranking"24    difficulty: hard25    description: "Rank customers by spending using window functions or subqueries"26    max_score: 1.027action_space:28  type: text29  description: "A SQL SELECT query string"30observation_space:31  type: object32  fields:33    schema: "Database DDL as text"34    question: "Natural language question"35    result: "Query execution result or error"36    reward: "Float 0.0 to 1.0"37    done: "Boolean"38    feedback: "Reward explanation"39    task_id: "Current task identifier"40    task_difficulty: "easy/medium/hard"41    attempt: "Current attempt number"42    max_attempts: "Maximum attempts per task (3)"43    hint: "Optional hint for the current task"44reward:45  type: float46  range: [0.0, 1.0]47  description: "Graded by query result accuracy vs expected output"48