Upload environment.py

server/environment.py

@@ -223,13 +223,13 @@ class SQLQualityEnvironment:
         sql_stripped = re.sub(r"/\*.*?\*/", "", sql_stripped, flags=re.DOTALL)  # block comments
         first_keyword = sql_stripped.strip().split()[0].upper() if sql_stripped.strip() else ""
 
-        ALLOWED_READ_KEYWORDS = {"SELECT", "WITH", "EXPLAIN", "PRAGMA"}
+        # Allow all SQL types in query box for testing
+        # Only FIX action should be restricted
+        ALLOWED_READ_KEYWORDS = {"SELECT", "WITH", "EXPLAIN", "PRAGMA", "UPDATE", "INSERT", "DELETE", "CREATE", "DROP", "ALTER"}
         if first_keyword not in ALLOWED_READ_KEYWORDS:
             rejection_feedback = (
-                f"QUERY action only accepts read-only SQL. "
-                f"You submitted a '{first_keyword}' statement. "
-                f"To modify data, use the FIX action (action_type: 'fix'). "
-                f"To find the answer, use action_type: 'submit'."
+                f"Invalid SQL keyword: '{first_keyword}'. "
+                f"Allowed keywords: {', '.join(sorted(ALLOWED_READ_KEYWORDS))}"
             )
             # Add to trajectory as a step but with zero reward and clear feedback
             self._trajectory.append(