nao-core 0.0.19__py3-none-any.whl → 0.0.25__py3-none-any.whl

nao_core/__init__.py

@@ -1,2 +1,2 @@
 # nao Core CLI package
-__version__ = "0.0.19"
+__version__ = "0.0.25"

nao_core/bin/.nao-secret

@@ -0,0 +1 @@
+E64sVzzbsFL7_c5XUJE5r2F1tcAIRub5Wco-e1rw51E

nao_core/bin/fastapi/main.py

@@ -1,6 +1,7 @@
+from nao_core.config import NaoConfig
 from fastapi import FastAPI, HTTPException
-from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
+from fastapi.middleware.cors import CORSMiddleware
 import uvicorn
 import os
 import sys
@@ -10,8 +11,6 @@ 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()
@@ -35,23 +34,6 @@ class ExecuteSQLResponse(BaseModel):
     row_count: int
     columns: list[str]
 
-class ContextRequest(BaseModel):
-    nao_project_folder: str
-
-class ContextResponse(BaseModel):
-    context: dict
-
-@app.post("/context", response_model=ContextResponse)
-async def get_context(request: ContextRequest):
-    try:
-        # Load the nao config from the project folder
-        project_path = Path(request.nao_project_folder)
-        config = NaoConfig.try_load(project_path)
-        context = config.get_context()
-        return ContextResponse(context=context)
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
-
 
 @app.post("/execute_sql", response_model=ExecuteSQLResponse)
 async def execute_sql(request: ExecuteSQLRequest):
@@ -102,11 +84,10 @@ async def execute_sql(request: ExecuteSQLRequest):
             )
 
         connection = db_config.connect()
-        result = connection.sql(request.sql)
-
-        df = result.to_pandas()
         
-        # Convert numpy types to native Python types for JSON serialization
+        cursor = connection.raw_sql(request.sql)  # type: ignore[attr-defined]
+        df = cursor.to_dataframe()
+        
         def convert_value(v):
             if isinstance(v, (np.integer,)):
                 return int(v)
@@ -131,6 +112,5 @@ async def execute_sql(request: ExecuteSQLRequest):
     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/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");