nao-core 0.0.16__tar.gz → 0.0.18__tar.gz

{nao_core-0.0.16 → nao_core-0.0.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nao-core
-Version: 0.0.16
+Version: 0.0.18
 Summary: nao Core is your analytics context builder with the best chat interface.
 Project-URL: Homepage, https://getnao.io
 Project-URL: Repository, https://github.com/naolabs/chat

{nao_core-0.0.16 → nao_core-0.0.18}/nao_core/__init__.py

@@ -1,2 +1,2 @@
 # nao Core CLI package
-__version__ = "0.0.16"
+__version__ = "0.0.18"

nao_core-0.0.18/nao_core/bin/chats/588bbb8a-dcbc-42dc-b0d8-b6b9ed13e2fb.json

@@ -0,0 +1,45 @@
+{
+  "id": "588bbb8a-dcbc-42dc-b0d8-b6b9ed13e2fb",
+  "title": "hello",
+  "createdAt": 1768466612813,
+  "updatedAt": 1768466612813,
+  "messages": [
+    {
+      "parts": [
+        {
+          "type": "text",
+          "text": "hello"
+        }
+      ],
+      "id": "0Tpxv0d7sz7KWBN7",
+      "role": "user"
+    },
+    {
+      "id": "dGKqwzEtjD2wtI5C",
+      "role": "assistant",
+      "parts": [
+        {
+          "type": "data-newChat",
+          "data": {
+            "id": "588bbb8a-dcbc-42dc-b0d8-b6b9ed13e2fb",
+            "title": "hello",
+            "createdAt": 1768466612820,
+            "updatedAt": 1768466612820,
+            "messages": [
+              {
+                "parts": [
+                  {
+                    "type": "text",
+                    "text": "hello"
+                  }
+                ],
+                "id": "0Tpxv0d7sz7KWBN7",
+                "role": "user"
+              }
+            ]
+          }
+        }
+      ]
+    }
+  ]
+}

{nao_core-0.0.16 → nao_core-0.0.18}/nao_core/bin/fastapi/main.py

@@ -4,6 +4,7 @@ from pydantic import BaseModel
 import uvicorn
 import os
 import sys
+import numpy as np
 from pathlib import Path
 
 cli_path = Path(__file__).parent.parent.parent / "cli"
@@ -87,12 +88,28 @@ async def execute_sql(request: ExecuteSQLRequest):
         result = connection.sql(request.sql)
 
         df = result.to_pandas()
-        data = df.to_dict(orient="records")
+        
+        # Convert numpy types to native Python types for JSON serialization
+        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=df.columns.tolist(),
+            columns=[str(c) for c in df.columns.tolist()],
         )
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))

nao_core-0.0.16/nao_core/bin/migrations-postgres/0000_supreme_cable.sql → nao_core-0.0.18/nao_core/bin/migrations-postgres/0000_user_auth_and_chat_tables.sql

@@ -14,20 +14,41 @@ CREATE TABLE "account" (
 	"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,
-	"conversation_id" text NOT NULL,
+	"chat_id" text NOT NULL,
 	"role" text NOT NULL,
-	"content" text NOT NULL,
 	"created_at" timestamp DEFAULT now() NOT NULL
 );
 --> statement-breakpoint
-CREATE TABLE "conversation" (
+CREATE TABLE "message_part" (
 	"id" text PRIMARY KEY NOT NULL,
-	"user_id" text NOT NULL,
-	"title" text DEFAULT 'New Conversation' NOT NULL,
+	"message_id" text NOT NULL,
+	"order" integer NOT NULL,
 	"created_at" timestamp DEFAULT now() NOT NULL,
-	"updated_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" (
@@ -42,16 +63,6 @@ CREATE TABLE "session" (
 	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,
@@ -73,13 +84,15 @@ CREATE TABLE "verification" (
 );
 --> 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 "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
-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 "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 "tool_call_messageId_idx" ON "tool_call" USING btree ("message_id");--> statement-breakpoint
 CREATE INDEX "verification_identifier_idx" ON "verification" USING btree ("identifier");