nl2sql/executor.py

import time
from nl2sql.types import StageResult, StageTrace
from adapters.db.base import DBAdapter


class Executor:
    name = "executor"

    def __init__(self, db: DBAdapter):
        self.db = db

    def run(self, sql: str) -> StageResult:
        t0 = time.perf_counter()
        try:
            rows, cols = self.db.execute(sql)
            trace = StageTrace(
                stage=self.name,
                duration_ms=(time.perf_counter() - t0) * 1000,
                notes={"row_count": len(rows), "col_count": len(cols)},
            )
            return StageResult(
                ok=True, data={"rows": rows, "columns": cols}, trace=trace
            )
        except Exception as e:
            trace = StageTrace(
                stage=self.name,
                duration_ms=(time.perf_counter() - t0) * 1000,
                notes={"error": str(e)},
            )
            return StageResult(ok=False, data=None, trace=trace, error=[str(e)])