adapters/llm/base.py

# adapters/llm/base.py
from __future__ import annotations
from typing import Tuple, List, Dict, Any, Protocol

class LLMProvider(Protocol):
    provider_id: str

    def plan(self, *, user_query: str, schema_preview: str) -> Tuple[str, int, int, float]:
        """Return (plan_text, token_in, token_out, cost_usd)."""

    def generate_sql(self, *, user_query: str, schema_preview: str, plan_text: str,
                     clarify_answers: Dict[str, Any] | None = None) -> Tuple[str, str, int, int, float]:
        """Return (sql, rationale, token_in, token_out, cost_usd)."""

    def repair(self, *, sql: str, error_msg: str, schema_preview: str) -> Tuple[str, int, int, float]:
        """Return (patched_sql, token_in, token_out, cost_usd)."""