oumi.environments.examples

Submodules

oumi.environments.examples.nl2sql module

Read-only run_sql tool executor for the database environment.

oumi.environments.examples.nl2sql.run_sql(arguments: dict[str, Any], context: Connection) → ToolResult

Execute the query argument against the rollout’s bound SQLite connection.

Returns:

{"columns", "rows"} on success, {"error": <sqlite message>} if the query does not run, is unauthorized, or exceeds its work budget.