nao-core 0.0.38__py3-none-manylinux2014_aarch64.whl

nao_core/__init__.py

@@ -0,0 +1,2 @@
+# nao Core CLI package
+__version__ = "0.0.38"

nao_core/__init__.py.bak

@@ -0,0 +1,2 @@
+# nao Core CLI package
+__version__ = "0.0.37"

nao_core/bin/build-info.json

@@ -0,0 +1,5 @@
+{
+  "commit": "28e16dab7e1e09323857821960e354591afcf71a",
+  "commitShort": "28e16da",
+  "buildTime": "2026-01-30T16:33:50.944081+00:00"
+}

nao_core/bin/fastapi/main.py

@@ -0,0 +1,268 @@
+import os
+import sys
+from contextlib import asynccontextmanager
+from datetime import datetime
+from pathlib import Path
+
+import numpy as np
+import uvicorn
+from dotenv import load_dotenv
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+
+load_dotenv()
+
+cli_path = Path(__file__).parent.parent.parent / "cli"
+sys.path.insert(0, str(cli_path))
+
+from nao_core.config import NaoConfig
+from nao_core.context import get_context_provider
+
+port = int(os.environ.get("PORT", 8005))
+
+# Global scheduler instance
+scheduler = None
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """Manage application lifespan - setup scheduler on startup."""
+    global scheduler
+
+    # Setup periodic refresh if configured
+    refresh_schedule = os.environ.get("NAO_REFRESH_SCHEDULE")
+    if refresh_schedule:
+        from apscheduler.schedulers.asyncio import AsyncIOScheduler
+        from apscheduler.triggers.cron import CronTrigger
+
+        scheduler = AsyncIOScheduler()
+
+        try:
+            trigger = CronTrigger.from_crontab(refresh_schedule)
+            scheduler.add_job(
+                _refresh_context_task,
+                trigger,
+                id="context_refresh",
+                name="Periodic context refresh",
+            )
+            scheduler.start()
+            print(f"[Scheduler] Periodic refresh enabled: {refresh_schedule}")
+        except ValueError as e:
+            print(f"[Scheduler] Invalid cron expression '{refresh_schedule}': {e}")
+
+    yield
+
+    # Shutdown scheduler
+    if scheduler:
+        scheduler.shutdown(wait=False)
+
+
+async def _refresh_context_task():
+    """Background task for scheduled context refresh."""
+    try:
+        provider = get_context_provider()
+        updated = provider.refresh()
+        if updated:
+            print(f"[Scheduler] Context refreshed at {datetime.now().isoformat()}")
+        else:
+            print(f"[Scheduler] Context already up-to-date at {datetime.now().isoformat()}")
+    except Exception as e:
+        print(f"[Scheduler] Failed to refresh context: {e}")
+
+
+app = FastAPI(lifespan=lifespan)
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+
+# =============================================================================
+# Request/Response Models
+# =============================================================================
+
+
+class ExecuteSQLRequest(BaseModel):
+    sql: str
+    nao_project_folder: str
+    database_id: str | None = None
+
+
+class ExecuteSQLResponse(BaseModel):
+    data: list[dict]
+    row_count: int
+    columns: list[str]
+
+
+class RefreshResponse(BaseModel):
+    status: str
+    updated: bool
+    message: str
+
+
+class HealthResponse(BaseModel):
+    status: str
+    context_source: str
+    context_initialized: bool
+    refresh_schedule: str | None
+
+
+# =============================================================================
+# API Endpoints
+# =============================================================================
+
+
+@app.get("/health", response_model=HealthResponse)
+async def health_check():
+    """Health check endpoint with context status."""
+    try:
+        provider = get_context_provider()
+        context_source = os.environ.get("NAO_CONTEXT_SOURCE", "local")
+        return HealthResponse(
+            status="ok",
+            context_source=context_source,
+            context_initialized=provider.is_initialized(),
+            refresh_schedule=os.environ.get("NAO_REFRESH_SCHEDULE"),
+        )
+    except Exception:
+        return HealthResponse(
+            status="error",
+            context_source=os.environ.get("NAO_CONTEXT_SOURCE", "local"),
+            context_initialized=False,
+            refresh_schedule=os.environ.get("NAO_REFRESH_SCHEDULE"),
+        )
+
+
+@app.post("/api/refresh", response_model=RefreshResponse)
+async def refresh_context():
+    """Trigger a context refresh (git pull if using git source).
+
+    This endpoint can be called by:
+    - CI/CD pipelines after pushing new context
+    - Webhooks when data schemas change
+    - Manual triggers for immediate updates
+    """
+    try:
+        provider = get_context_provider()
+        updated = provider.refresh()
+
+        if updated:
+            return RefreshResponse(
+                status="ok",
+                updated=True,
+                message="Context updated successfully",
+            )
+        else:
+            return RefreshResponse(
+                status="ok",
+                updated=False,
+                message="Context already up-to-date",
+            )
+    except Exception as e:
+        raise HTTPException(
+            status_code=500,
+            detail=f"Failed to refresh context: {str(e)}",
+        )
+
+
+@app.post("/execute_sql", response_model=ExecuteSQLResponse)
+async def execute_sql(request: ExecuteSQLRequest):
+    try:
+        # Load the nao config from the project folder
+        project_path = Path(request.nao_project_folder)
+        os.chdir(project_path)
+        config = NaoConfig.try_load(project_path)
+
+        if config is None:
+            raise HTTPException(
+                status_code=400,
+                detail=f"Could not load nao_config.yaml from {request.nao_project_folder}",
+            )
+
+        if len(config.databases) == 0:
+            raise HTTPException(
+                status_code=400,
+                detail="No databases configured in nao_config.yaml",
+            )
+
+        # Determine which database to use
+        if len(config.databases) == 1:
+            db_config = config.databases[0]
+        elif request.database_id:
+            # Find the database by name
+            db_config = next(
+                (db for db in config.databases if db.name == request.database_id),
+                None,
+            )
+            if db_config is None:
+                available_databases = [db.name for db in config.databases]
+                raise HTTPException(
+                    status_code=400,
+                    detail={
+                        "message": f"Database '{request.database_id}' not found",
+                        "available_databases": available_databases,
+                    },
+                )
+        else:
+            # Multiple databases and no database_id specified
+            available_databases = [db.name for db in config.databases]
+            raise HTTPException(
+                status_code=400,
+                detail={
+                    "message": "Multiple databases configured. Please specify database_id.",
+                    "available_databases": available_databases,
+                },
+            )
+
+        connection = db_config.connect()
+
+        # Use raw_sql to execute arbitrary SQL (including CTEs)
+        cursor = connection.raw_sql(request.sql)
+
+        # Handle different cursor types from different backends
+        if hasattr(cursor, "fetchdf"):
+            # DuckDB returns a cursor with fetchdf()
+            df = cursor.fetchdf()
+        elif hasattr(cursor, "to_dataframe"):
+            # Some backends return cursors with to_dataframe()
+            df = cursor.to_dataframe()
+        else:
+            # Fallback: try to use pandas read_sql or fetchall
+            import pandas as pd
+
+            columns = [desc[0] for desc in cursor.description]
+            df = pd.DataFrame(cursor.fetchall(), columns=columns)
+
+        def convert_value(v):
+            if isinstance(v, (np.integer,)):
+                return int(v)
+            if isinstance(v, (np.floating,)):
+                return float(v)
+            if isinstance(v, np.ndarray):
+                return v.tolist()
+            if hasattr(v, "item"):  # numpy scalar
+                return v.item()
+            return v
+
+        data = [{k: convert_value(v) for k, v in row.items()} for row in df.to_dict(orient="records")]
+
+        return ExecuteSQLResponse(
+            data=data,
+            row_count=len(data),
+            columns=[str(c) for c in df.columns.tolist()],
+        )
+    except HTTPException:
+        raise
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+if __name__ == "__main__":
+    nao_project_folder = os.getenv("NAO_DEFAULT_PROJECT_PATH")
+    if nao_project_folder:
+        os.chdir(nao_project_folder)
+    uvicorn.run("main:app", host="0.0.0.0", port=port, reload=True)

nao_core/bin/fastapi/test_main.py

@@ -0,0 +1,156 @@
+import tempfile
+from pathlib import Path
+
+import pytest
+import yaml
+from fastapi.testclient import TestClient
+
+from main import app
+
+
+def assert_sql_result(data: dict, *, row_count: int, columns: list[str], expected_data: list[dict]):
+    """Assert that SQL response data matches expected values."""
+    assert data["row_count"] == row_count
+    assert data["columns"] == columns
+    assert len(data["data"]) == row_count
+    assert data["data"] == expected_data
+
+
+@pytest.fixture
+def duckdb_project_folder():
+    """Create a temporary project folder with a DuckDB config."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        config = {
+            "project_name": "test-project",
+            "databases": [
+                {
+                    "name": "test-duckdb",
+                    "type": "duckdb",
+                    "path": ":memory:",
+                }
+            ],
+        }
+        config_path = Path(tmpdir) / "nao_config.yaml"
+        with config_path.open("w") as f:
+            yaml.dump(config, f)
+        yield tmpdir
+
+
+def test_execute_sql_simple_duckdb(duckdb_project_folder):
+    """Test execute_sql endpoint with a DuckDB in-memory database."""
+    client = TestClient(app)
+
+    response = client.post(
+        "/execute_sql",
+        json={
+            "sql": "SELECT 1 AS id, 'hello' AS message",
+            "nao_project_folder": duckdb_project_folder,
+        },
+    )
+
+    assert response.status_code == 200
+    assert_sql_result(
+        response.json(),
+        row_count=1,
+        columns=["id", "message"],
+        expected_data=[{"id": 1, "message": "hello"}],
+    )
+
+
+def test_execute_sql_with_cte_duckdb(duckdb_project_folder):
+    """Test execute_sql endpoint with a DuckDB in-memory database."""
+    client = TestClient(app)
+
+    response = client.post(
+        "/execute_sql",
+        json={
+            "sql": "WITH test AS (SELECT 1 AS id, 'hello' AS message) SELECT * FROM test",
+            "nao_project_folder": duckdb_project_folder,
+        },
+    )
+
+    assert response.status_code == 200
+    assert_sql_result(
+        response.json(),
+        row_count=1,
+        columns=["id", "message"],
+        expected_data=[{"id": 1, "message": "hello"}],
+    )
+
+
+# BigQuery tests (requires SSO authentication)
+
+@pytest.fixture
+def bigquery_project_folder():
+    """Create a temporary project folder with a BigQuery config using SSO."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        config = {
+            "project_name": "test-project",
+            "databases": [
+                {
+                    "name": "nao-bigquery",
+                    "type": "bigquery",
+                    "project_id": "nao-corp",
+                    "sso": True,
+                }
+            ],
+        }
+        config_path = Path(tmpdir) / "nao_config.yaml"
+        with config_path.open("w") as f:
+            yaml.dump(config, f)
+        yield tmpdir
+
+
+def test_execute_sql_simple_bigquery(bigquery_project_folder):
+    """Test execute_sql endpoint with BigQuery using SSO."""
+    client = TestClient(app)
+
+    response = client.post(
+        "/execute_sql",
+        json={
+            "sql": "SELECT 1 AS id, 'hello' AS message",
+            "nao_project_folder": bigquery_project_folder,
+        },
+    )
+
+    assert response.status_code == 200
+    assert_sql_result(
+        response.json(),
+        row_count=1,
+        columns=["id", "message"],
+        expected_data=[{"id": 1, "message": "hello"}],
+    )
+
+
+def test_execute_sql_with_cte_bigquery(bigquery_project_folder):
+    """Test execute_sql endpoint with a CTE query on BigQuery."""
+    client = TestClient(app)
+
+    cte_sql = """
+    WITH users AS (
+        SELECT 1 AS id, 'Alice' AS name
+        UNION ALL SELECT 2, 'Bob'
+        UNION ALL SELECT 3, 'Charlie'
+    )
+    SELECT * FROM users
+    """
+
+    response = client.post(
+        "/execute_sql",
+        json={
+            "sql": cte_sql,
+            "nao_project_folder": bigquery_project_folder,
+        },
+    )
+
+    assert response.status_code == 200
+    assert_sql_result(
+        response.json(),
+        row_count=3,
+        columns=["id", "name"],
+        expected_data=[
+            {"id": 1, "name": "Alice"},
+            {"id": 2, "name": "Bob"},
+            {"id": 3, "name": "Charlie"},
+        ],
+    )

nao_core/bin/migrations-postgres/0000_user_auth_and_chat_tables.sql

@@ -0,0 +1,98 @@
+CREATE TABLE "account" (
+	"id" text PRIMARY KEY NOT NULL,
+	"account_id" text NOT NULL,
+	"provider_id" text NOT NULL,
+	"user_id" text NOT NULL,
+	"access_token" text,
+	"refresh_token" text,
+	"id_token" text,
+	"access_token_expires_at" timestamp,
+	"refresh_token_expires_at" timestamp,
+	"scope" text,
+	"password" text,
+	"created_at" timestamp DEFAULT now() NOT NULL,
+	"updated_at" timestamp NOT NULL
+);
+--> statement-breakpoint
+CREATE TABLE "chat" (
+	"id" text PRIMARY KEY NOT NULL,
+	"user_id" text NOT NULL,
+	"title" text DEFAULT 'New Conversation' NOT NULL,
+	"created_at" timestamp DEFAULT now() NOT NULL,
+	"updated_at" timestamp DEFAULT now() NOT NULL
+);
+--> statement-breakpoint
+CREATE TABLE "chat_message" (
+	"id" text PRIMARY KEY NOT NULL,
+	"chat_id" text NOT NULL,
+	"role" text NOT NULL,
+	"created_at" timestamp DEFAULT now() NOT NULL
+);
+--> statement-breakpoint
+CREATE TABLE "message_part" (
+	"id" text PRIMARY KEY NOT NULL,
+	"message_id" text NOT NULL,
+	"order" integer NOT NULL,
+	"created_at" timestamp DEFAULT now() NOT NULL,
+	"type" text NOT NULL,
+	"text" text,
+	"reasoning_text" text,
+	"tool_call_id" text,
+	"tool_name" text,
+	"tool_state" text,
+	"tool_error_text" text,
+	"tool_input" jsonb,
+	"tool_output" jsonb,
+	"tool_approval_id" text,
+	"tool_approval_approved" boolean,
+	"tool_approval_reason" text,
+	CONSTRAINT "text_required_if_type_is_text" CHECK (CASE WHEN "message_part"."type" = 'text' THEN "message_part"."text" IS NOT NULL ELSE TRUE END),
+	CONSTRAINT "reasoning_text_required_if_type_is_reasoning" CHECK (CASE WHEN "message_part"."type" = 'reasoning' THEN "message_part"."reasoning_text" IS NOT NULL ELSE TRUE END),
+	CONSTRAINT "tool_call_fields_required" CHECK (CASE WHEN "message_part"."type" LIKE 'tool-%' THEN "message_part"."tool_call_id" IS NOT NULL AND "message_part"."tool_state" IS NOT NULL ELSE TRUE END)
+);
+--> statement-breakpoint
+CREATE TABLE "session" (
+	"id" text PRIMARY KEY NOT NULL,
+	"expires_at" timestamp NOT NULL,
+	"token" text NOT NULL,
+	"created_at" timestamp DEFAULT now() NOT NULL,
+	"updated_at" timestamp NOT NULL,
+	"ip_address" text,
+	"user_agent" text,
+	"user_id" text NOT NULL,
+	CONSTRAINT "session_token_unique" UNIQUE("token")
+);
+--> statement-breakpoint
+CREATE TABLE "user" (
+	"id" text PRIMARY KEY NOT NULL,
+	"name" text NOT NULL,
+	"email" text NOT NULL,
+	"email_verified" boolean DEFAULT false NOT NULL,
+	"image" text,
+	"created_at" timestamp DEFAULT now() NOT NULL,
+	"updated_at" timestamp DEFAULT now() NOT NULL,
+	CONSTRAINT "user_email_unique" UNIQUE("email")
+);
+--> statement-breakpoint
+CREATE TABLE "verification" (
+	"id" text PRIMARY KEY NOT NULL,
+	"identifier" text NOT NULL,
+	"value" text NOT NULL,
+	"expires_at" timestamp NOT NULL,
+	"created_at" timestamp DEFAULT now() NOT NULL,
+	"updated_at" timestamp DEFAULT now() NOT NULL
+);
+--> statement-breakpoint
+ALTER TABLE "account" ADD CONSTRAINT "account_user_id_user_id_fk" FOREIGN KEY ("user_id") REFERENCES "public"."user"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "chat" ADD CONSTRAINT "chat_user_id_user_id_fk" FOREIGN KEY ("user_id") REFERENCES "public"."user"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "chat_message" ADD CONSTRAINT "chat_message_chat_id_chat_id_fk" FOREIGN KEY ("chat_id") REFERENCES "public"."chat"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "message_part" ADD CONSTRAINT "message_part_message_id_chat_message_id_fk" FOREIGN KEY ("message_id") REFERENCES "public"."chat_message"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "session" ADD CONSTRAINT "session_user_id_user_id_fk" FOREIGN KEY ("user_id") REFERENCES "public"."user"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+CREATE INDEX "account_userId_idx" ON "account" USING btree ("user_id");--> statement-breakpoint
+CREATE INDEX "chat_userId_idx" ON "chat" USING btree ("user_id");--> statement-breakpoint
+CREATE INDEX "chat_message_chatId_idx" ON "chat_message" USING btree ("chat_id");--> statement-breakpoint
+CREATE INDEX "chat_message_createdAt_idx" ON "chat_message" USING btree ("created_at");--> statement-breakpoint
+CREATE INDEX "parts_message_id_idx" ON "message_part" USING btree ("message_id");--> statement-breakpoint
+CREATE INDEX "parts_message_id_order_idx" ON "message_part" USING btree ("message_id","order");--> statement-breakpoint
+CREATE INDEX "session_userId_idx" ON "session" USING btree ("user_id");--> statement-breakpoint
+CREATE INDEX "verification_identifier_idx" ON "verification" USING btree ("identifier");

nao_core/bin/migrations-postgres/0001_message_feedback.sql

@@ -0,0 +1,9 @@
+CREATE TABLE "message_feedback" (
+	"message_id" text PRIMARY KEY NOT NULL,
+	"vote" text NOT NULL,
+	"explanation" text,
+	"created_at" timestamp DEFAULT now() NOT NULL,
+	"updated_at" timestamp DEFAULT now() NOT NULL
+);
+--> statement-breakpoint
+ALTER TABLE "message_feedback" ADD CONSTRAINT "message_feedback_message_id_chat_message_id_fk" FOREIGN KEY ("message_id") REFERENCES "public"."chat_message"("id") ON DELETE cascade ON UPDATE no action;

nao_core/bin/migrations-postgres/0002_chat_message_stop_reason_and_error_message.sql

@@ -0,0 +1,2 @@
+ALTER TABLE "chat_message" ADD COLUMN "stop_reason" text;--> statement-breakpoint
+ALTER TABLE "chat_message" ADD COLUMN "error_message" text;

nao_core/bin/migrations-postgres/0003_handle_slack_with_thread.sql

@@ -0,0 +1,2 @@
+ALTER TABLE "chat" ADD COLUMN "slack_thread_id" text;--> statement-breakpoint
+CREATE INDEX "chat_slack_thread_idx" ON "chat" USING btree ("slack_thread_id");

nao_core/bin/migrations-postgres/0004_input_and_output_tokens.sql

@@ -0,0 +1,8 @@
+ALTER TABLE "message_part" ADD COLUMN "input_total_tokens" integer;--> statement-breakpoint
+ALTER TABLE "message_part" ADD COLUMN "input_no_cache_tokens" integer;--> statement-breakpoint
+ALTER TABLE "message_part" ADD COLUMN "input_cache_read_tokens" integer;--> statement-breakpoint
+ALTER TABLE "message_part" ADD COLUMN "input_cache_write_tokens" integer;--> statement-breakpoint
+ALTER TABLE "message_part" ADD COLUMN "output_total_tokens" integer;--> statement-breakpoint
+ALTER TABLE "message_part" ADD COLUMN "output_text_tokens" integer;--> statement-breakpoint
+ALTER TABLE "message_part" ADD COLUMN "output_reasoning_tokens" integer;--> statement-breakpoint
+ALTER TABLE "message_part" ADD COLUMN "total_tokens" integer;

nao_core/bin/migrations-postgres/0005_add_project_tables.sql

@@ -0,0 +1,39 @@
+CREATE TABLE "project" (
+	"id" text PRIMARY KEY NOT NULL,
+	"name" text NOT NULL,
+	"type" text NOT NULL,
+	"path" text,
+	"slack_bot_token" text,
+	"slack_signing_secret" text,
+	"created_at" timestamp DEFAULT now() NOT NULL,
+	"updated_at" timestamp DEFAULT now() NOT NULL,
+	CONSTRAINT "local_project_path_required" CHECK (CASE WHEN "project"."type" = 'local' THEN "project"."path" IS NOT NULL ELSE TRUE END)
+);
+--> statement-breakpoint
+CREATE TABLE "project_llm_config" (
+	"id" text PRIMARY KEY NOT NULL,
+	"project_id" text NOT NULL,
+	"provider" text NOT NULL,
+	"api_key" text NOT NULL,
+	"created_at" timestamp DEFAULT now() NOT NULL,
+	"updated_at" timestamp DEFAULT now() NOT NULL,
+	CONSTRAINT "project_llm_config_unique" UNIQUE("id","project_id","provider")
+);
+--> statement-breakpoint
+CREATE TABLE "project_member" (
+	"project_id" text NOT NULL,
+	"user_id" text NOT NULL,
+	"role" text NOT NULL,
+	"created_at" timestamp DEFAULT now() NOT NULL,
+	CONSTRAINT "project_member_project_id_user_id_pk" PRIMARY KEY("project_id","user_id")
+);
+--> statement-breakpoint
+ALTER TABLE "chat" ADD COLUMN "project_id" text NOT NULL;--> statement-breakpoint
+ALTER TABLE "project_llm_config" ADD CONSTRAINT "project_llm_config_project_id_project_id_fk" FOREIGN KEY ("project_id") REFERENCES "public"."project"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "project_member" ADD CONSTRAINT "project_member_project_id_project_id_fk" FOREIGN KEY ("project_id") REFERENCES "public"."project"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "project_member" ADD CONSTRAINT "project_member_user_id_user_id_fk" FOREIGN KEY ("user_id") REFERENCES "public"."user"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+CREATE INDEX "project_llm_config_projectId_idx" ON "project_llm_config" USING btree ("project_id");--> statement-breakpoint
+CREATE INDEX "project_member_userId_idx" ON "project_member" USING btree ("user_id");--> statement-breakpoint
+DELETE FROM "chat";--> statement-breakpoint
+ALTER TABLE "chat" ADD CONSTRAINT "chat_project_id_project_id_fk" FOREIGN KEY ("project_id") REFERENCES "public"."project"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+CREATE INDEX "chat_projectId_idx" ON "chat" USING btree ("project_id");

nao_core/bin/migrations-postgres/0006_llm_model_ids.sql

@@ -0,0 +1,4 @@
+ALTER TABLE "project_llm_config" DROP CONSTRAINT "project_llm_config_unique";--> statement-breakpoint
+ALTER TABLE "project_llm_config" ADD COLUMN "enabled_models" jsonb DEFAULT '[]'::jsonb NOT NULL;--> statement-breakpoint
+ALTER TABLE "project_llm_config" ADD COLUMN "base_url" text;--> statement-breakpoint
+ALTER TABLE "project_llm_config" ADD CONSTRAINT "project_llm_config_project_provider" UNIQUE("project_id","provider");

nao_core/bin/migrations-postgres/0007_chat_message_llm_info.sql

@@ -0,0 +1,2 @@
+ALTER TABLE "chat_message" ADD COLUMN "llm_provider" text;--> statement-breakpoint
+ALTER TABLE "chat_message" ADD COLUMN "llm_model_id" text;