nao-core 0.0.11__py3-none-any.whl → 0.0.14__py3-none-any.whl

nao_core/__init__.py

@@ -1,2 +1,2 @@
 # nao Core CLI package
-__version__ = "0.0.11"
+__version__ = "0.0.14"

nao_core/bin/fastapi/main.py

@@ -0,0 +1,102 @@
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+import uvicorn
+import os
+import sys
+from pathlib import Path
+
+cli_path = Path(__file__).parent.parent.parent / "cli"
+sys.path.insert(0, str(cli_path))
+
+from nao_core.config import NaoConfig
+
+port = int(os.environ.get("PORT", 8005))
+
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+
+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]
+
+
+@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)
+        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()
+        result = connection.sql(request.sql)
+
+        df = result.to_pandas()
+        data = df.to_dict(orient="records")
+
+        return ExecuteSQLResponse(
+            data=data,
+            row_count=len(data),
+            columns=df.columns.tolist(),
+        )
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+if __name__ == "__main__":
+    uvicorn.run("main:app", host="0.0.0.0", port=port, reload=True)

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

@@ -0,0 +1,85 @@
+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_message" (
+	"id" text PRIMARY KEY NOT NULL,
+	"conversation_id" text NOT NULL,
+	"role" text NOT NULL,
+	"content" text NOT NULL,
+	"created_at" timestamp DEFAULT now() NOT NULL
+);
+--> statement-breakpoint
+CREATE TABLE "conversation" (
+	"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 "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 "tool_call" (
+	"id" text PRIMARY KEY NOT NULL,
+	"message_id" text NOT NULL,
+	"tool_call_id" text NOT NULL,
+	"tool_name" text NOT NULL,
+	"input" jsonb NOT NULL,
+	"output" jsonb,
+	"created_at" timestamp DEFAULT now() NOT NULL
+);
+--> 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_message" ADD CONSTRAINT "chat_message_conversation_id_conversation_id_fk" FOREIGN KEY ("conversation_id") REFERENCES "public"."conversation"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "conversation" ADD CONSTRAINT "conversation_user_id_user_id_fk" FOREIGN KEY ("user_id") REFERENCES "public"."user"("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
+ALTER TABLE "tool_call" ADD CONSTRAINT "tool_call_message_id_chat_message_id_fk" FOREIGN KEY ("message_id") REFERENCES "public"."chat_message"("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_message_conversationId_idx" ON "chat_message" USING btree ("conversation_id");--> statement-breakpoint
+CREATE INDEX "conversation_userId_idx" ON "conversation" USING btree ("user_id");--> statement-breakpoint
+CREATE INDEX "session_userId_idx" ON "session" USING btree ("user_id");--> statement-breakpoint
+CREATE INDEX "tool_call_messageId_idx" ON "tool_call" USING btree ("message_id");--> statement-breakpoint
+CREATE INDEX "verification_identifier_idx" ON "verification" USING btree ("identifier");