@m6d/cortex-server 1.8.0 → 2.0.1

package/src/lib/db/migrations/pg/20260812174622_user_id_text/snapshot.json

@@ -0,0 +1,584 @@
+{
+    "version": "8",
+    "dialect": "postgres",
+    "id": "a5037982-d7a6-419a-8d7a-aaacd8bf939b",
+    "prevIds": ["ba8a51ae-9fce-4847-9971-a8d989fd5a4d", "00000000-0000-0000-0000-000000000000"],
+    "ddl": [
+        {
+            "name": "ai",
+            "entityType": "schemas"
+        },
+        {
+            "isRlsEnabled": false,
+            "name": "attachments",
+            "entityType": "tables",
+            "schema": "ai"
+        },
+        {
+            "isRlsEnabled": false,
+            "name": "llm_requests",
+            "entityType": "tables",
+            "schema": "ai"
+        },
+        {
+            "isRlsEnabled": false,
+            "name": "messages",
+            "entityType": "tables",
+            "schema": "ai"
+        },
+        {
+            "isRlsEnabled": false,
+            "name": "threads",
+            "entityType": "tables",
+            "schema": "ai"
+        },
+        {
+            "type": "uuid",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": "gen_random_uuid()",
+            "generated": null,
+            "identity": null,
+            "name": "id",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "type": "uuid",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "thread_id",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "type": "varchar(255)",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "user_id",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "type": "varchar(128)",
+            "typeSchema": null,
+            "notNull": false,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "message_id",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "type": "varchar(2048)",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "filename",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "type": "varchar(256)",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "content_type",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "type": "integer",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "size_bytes",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "type": "varchar(512)",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "storage_key",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "type": "varchar(16)",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "status",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "type": "text",
+            "typeSchema": null,
+            "notNull": false,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "description",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "type": "timestamp with time zone",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "created_at",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "type": "timestamp with time zone",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "updated_at",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "type": "uuid",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": "gen_random_uuid()",
+            "generated": null,
+            "identity": null,
+            "name": "id",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "llm_requests"
+        },
+        {
+            "type": "varchar(128)",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "message_id",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "llm_requests"
+        },
+        {
+            "type": "integer",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "step_number",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "llm_requests"
+        },
+        {
+            "type": "text",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "prompt",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "llm_requests"
+        },
+        {
+            "type": "text",
+            "typeSchema": null,
+            "notNull": false,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "output",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "llm_requests"
+        },
+        {
+            "type": "jsonb",
+            "typeSchema": null,
+            "notNull": false,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "token_usage",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "llm_requests"
+        },
+        {
+            "type": "varchar(128)",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "id",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "messages"
+        },
+        {
+            "type": "uuid",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "thread_id",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "messages"
+        },
+        {
+            "type": "text",
+            "typeSchema": null,
+            "notNull": false,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "text",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "messages"
+        },
+        {
+            "type": "jsonb",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "content",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "messages"
+        },
+        {
+            "type": "integer",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": "0",
+            "generated": null,
+            "identity": null,
+            "name": "ordinal",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "messages"
+        },
+        {
+            "type": "varchar(16)",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "role",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "messages"
+        },
+        {
+            "type": "timestamp with time zone",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "created_at",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "messages"
+        },
+        {
+            "type": "timestamp with time zone",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "updated_at",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "messages"
+        },
+        {
+            "type": "uuid",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": "gen_random_uuid()",
+            "generated": null,
+            "identity": null,
+            "name": "id",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "threads"
+        },
+        {
+            "type": "varchar(255)",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "user_id",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "threads"
+        },
+        {
+            "type": "varchar(128)",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": "'default'",
+            "generated": null,
+            "identity": null,
+            "name": "agent_id",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "threads"
+        },
+        {
+            "type": "varchar(256)",
+            "typeSchema": null,
+            "notNull": false,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "title",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "threads"
+        },
+        {
+            "type": "jsonb",
+            "typeSchema": null,
+            "notNull": false,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "session",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "threads"
+        },
+        {
+            "type": "jsonb",
+            "typeSchema": null,
+            "notNull": false,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "context_meta",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "threads"
+        },
+        {
+            "type": "timestamp with time zone",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "created_at",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "threads"
+        },
+        {
+            "type": "timestamp with time zone",
+            "typeSchema": null,
+            "notNull": true,
+            "dimensions": 0,
+            "default": null,
+            "generated": null,
+            "identity": null,
+            "name": "updated_at",
+            "entityType": "columns",
+            "schema": "ai",
+            "table": "threads"
+        },
+        {
+            "nameExplicit": true,
+            "columns": [
+                {
+                    "value": "thread_id",
+                    "isExpression": false,
+                    "asc": true,
+                    "nullsFirst": false,
+                    "opclass": null
+                }
+            ],
+            "isUnique": false,
+            "where": null,
+            "with": "",
+            "method": "btree",
+            "concurrently": false,
+            "name": "attachments_thread_id_index",
+            "entityType": "indexes",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "nameExplicit": false,
+            "columns": ["thread_id"],
+            "schemaTo": "ai",
+            "tableTo": "threads",
+            "columnsTo": ["id"],
+            "onUpdate": "NO ACTION",
+            "onDelete": "CASCADE",
+            "name": "attachments_thread_id_threads_id_fkey",
+            "entityType": "fks",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "nameExplicit": false,
+            "columns": ["message_id"],
+            "schemaTo": "ai",
+            "tableTo": "messages",
+            "columnsTo": ["id"],
+            "onUpdate": "NO ACTION",
+            "onDelete": "NO ACTION",
+            "name": "attachments_message_id_messages_id_fkey",
+            "entityType": "fks",
+            "schema": "ai",
+            "table": "attachments"
+        },
+        {
+            "nameExplicit": false,
+            "columns": ["message_id"],
+            "schemaTo": "ai",
+            "tableTo": "messages",
+            "columnsTo": ["id"],
+            "onUpdate": "NO ACTION",
+            "onDelete": "CASCADE",
+            "name": "llm_requests_message_id_messages_id_fkey",
+            "entityType": "fks",
+            "schema": "ai",
+            "table": "llm_requests"
+        },
+        {
+            "nameExplicit": false,
+            "columns": ["thread_id"],
+            "schemaTo": "ai",
+            "tableTo": "threads",
+            "columnsTo": ["id"],
+            "onUpdate": "NO ACTION",
+            "onDelete": "CASCADE",
+            "name": "messages_thread_id_threads_id_fkey",
+            "entityType": "fks",
+            "schema": "ai",
+            "table": "messages"
+        },
+        {
+            "columns": ["id"],
+            "nameExplicit": false,
+            "name": "attachments_pkey",
+            "schema": "ai",
+            "table": "attachments",
+            "entityType": "pks"
+        },
+        {
+            "columns": ["id"],
+            "nameExplicit": false,
+            "name": "llm_requests_pkey",
+            "schema": "ai",
+            "table": "llm_requests",
+            "entityType": "pks"
+        },
+        {
+            "columns": ["id"],
+            "nameExplicit": false,
+            "name": "messages_pkey",
+            "schema": "ai",
+            "table": "messages",
+            "entityType": "pks"
+        },
+        {
+            "columns": ["id"],
+            "nameExplicit": false,
+            "name": "threads_pkey",
+            "schema": "ai",
+            "table": "threads",
+            "entityType": "pks"
+        }
+    ],
+    "renames": []
+}

package/src/lib/db/{schema.ts → schema.mssql.ts}

@@ -1,8 +1,9 @@
-import type { ToolUIPart, UIMessage } from "ai";
+// fallow-ignore-file duplicate-export -- mssql twin of schema.pg; table exports intentionally share names so the adapters stay symmetric
+import { ATTACHMENT_STATUSES } from "@cortex/contracts/wire";
 import { sql } from "drizzle-orm";
 import { customType, datetime2, index, int, mssqlSchema, nvarchar } from "drizzle-orm/mssql-core";
-import type { MessageMetadata, TokenUsage } from "../types.ts";
-import type { ThreadContextMeta } from "../ai/context/types.ts";
+import type { ChatMessage, TokenUsage } from "@/types";
+import type { ThreadContextMeta } from "@/ai/context/types";
 
 const uniqueIdentifier = customType<{ data: string }>({
     dataType() {
@@ -26,7 +27,7 @@ export const threads = aiSchema.table("threads", {
     id: uniqueIdentifier()
         .default(sql`NEWID()`)
         .primaryKey(),
-    userId: uniqueIdentifier().notNull(),
+    userId: nvarchar({ length: 255 }).notNull(),
     agentId: nvarchar({ length: 128 }).notNull().default("default"),
     title: nvarchar({ length: 256 }),
     session: nvarchar({ mode: "json", length: "max" }).$type<Record<string, unknown>>(),
@@ -40,9 +41,13 @@ export const messages = aiSchema.table("messages", {
         .references(() => threads.id, { onDelete: "cascade" })
         .notNull(),
     text: nvarchar({ length: "max" }),
-    content: nvarchar({ mode: "json", length: "max" })
-        .notNull()
-        .$type<UIMessage<MessageMetadata>>(),
+    content: nvarchar({ mode: "json", length: "max" }).notNull().$type<ChatMessage>(),
+    /**
+     * Position within the insert batch that wrote the row. A turn commits
+     * several rows at once, so they share a `createdAt`; the transcript orders
+     * on (createdAt, ordinal) to keep them in the order the turn produced them.
+     */
+    ordinal: int().notNull().default(0),
     role: nvarchar({
         enum: ["system", "user", "assistant", "tool"],
         length: 16,
@@ -63,21 +68,27 @@ export const llmRequests = aiSchema.table("llm_requests", {
     tokenUsage: nvarchar({ length: "max", mode: "json" }).$type<TokenUsage>(),
 });
 
-export const capturedFiles = aiSchema.table(
-    "captured_files",
+export const attachments = aiSchema.table(
+    "attachments",
     {
         id: uniqueIdentifier()
             .default(sql`NEWID()`)
             .primaryKey(),
-        name: nvarchar({ length: 2048 }).notNull(),
-        userId: uniqueIdentifier().notNull(),
-        agentGeneratedId: nvarchar({ length: 1024 }).notNull(),
-        messageId: nvarchar({ length: 128 })
-            .references(() => messages.id, {
-                onDelete: "cascade",
-            })
+        threadId: uniqueIdentifier()
+            .references(() => threads.id, { onDelete: "cascade" })
             .notNull(),
-        toolPart: nvarchar({ length: "max", mode: "json" }).notNull().$type<ToolUIPart>(),
+        userId: nvarchar({ length: 255 }).notNull(),
+        messageId: nvarchar({ length: 128 }).references(() => messages.id),
+        filename: nvarchar({ length: 2048 }).notNull(),
+        contentType: nvarchar({ length: 256 }).notNull(),
+        sizeBytes: int().notNull(),
+        storageKey: nvarchar({ length: 512 }).notNull(),
+        status: nvarchar({
+            enum: ATTACHMENT_STATUSES,
+            length: 16,
+        }).notNull(),
+        description: nvarchar({ length: "max" }),
+        ...auditColumns,
     },
-    (table) => [index(table.agentGeneratedId)],
+    (table) => [index("attachments_thread_id_index").on(table.threadId)],
 );