@zaikit/memory-postgres 0.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Lionel Tay
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist/index.d.ts

@@ -0,0 +1,26 @@
+import * as _zaikit_core from '@zaikit/core';
+import { UIMessage } from 'ai';
+
+type PostgresMemoryOptions = {
+    connectionString: string;
+};
+declare function createPostgresMemory({ connectionString, }: PostgresMemoryOptions): {
+    initialize: () => Promise<void>;
+    createThread(id: string, title?: string, ownerId?: string): Promise<_zaikit_core.Thread>;
+    getThread(id: string): Promise<_zaikit_core.Thread | null>;
+    listThreads(opts?: {
+        ownerId?: string;
+    }): Promise<_zaikit_core.Thread[]>;
+    deleteThread(id: string): Promise<void>;
+    updateThread(id: string, updates: {
+        title?: string;
+        ownerId?: string;
+    }): Promise<_zaikit_core.Thread>;
+    getMessages(threadId: string): Promise<UIMessage[]>;
+    addMessage(threadId: string, message: UIMessage): Promise<void>;
+    updateMessage(threadId: string, messageId: string, updates: {
+        parts: UIMessage["parts"];
+    }): Promise<void>;
+};
+
+export { createPostgresMemory };

package/dist/index.js

@@ -0,0 +1,149 @@
+// src/index.ts
+import postgres from "postgres";
+function createPostgresMemory({
+  connectionString
+}) {
+  const sql = postgres(connectionString);
+  async function initialize() {
+    await sql`
+      CREATE TABLE IF NOT EXISTS zaikit_threads (
+        id TEXT PRIMARY KEY,
+        title TEXT,
+        owner_id TEXT,
+        created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+        updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
+      )
+    `;
+    await sql`
+      CREATE INDEX IF NOT EXISTS idx_zaikit_threads_owner_id ON zaikit_threads(owner_id)
+    `;
+    await sql`
+      CREATE TABLE IF NOT EXISTS zaikit_messages (
+        id TEXT PRIMARY KEY,
+        thread_id TEXT NOT NULL REFERENCES zaikit_threads(id) ON DELETE CASCADE,
+        role TEXT NOT NULL,
+        parts JSONB NOT NULL DEFAULT '[]',
+        metadata JSONB,
+        created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
+      )
+    `;
+    await sql`
+      CREATE INDEX IF NOT EXISTS idx_zaikit_messages_thread_id ON zaikit_messages(thread_id, created_at)
+    `;
+  }
+  const memory = {
+    async createThread(id, title, ownerId) {
+      const [row] = await sql`
+        INSERT INTO zaikit_threads (id, title, owner_id)
+        VALUES (${id}, ${title ?? null}, ${ownerId ?? null})
+        RETURNING id, title, owner_id, created_at, updated_at
+      `;
+      return {
+        id: row.id,
+        title: row.title,
+        ownerId: row.owner_id,
+        createdAt: row.created_at,
+        updatedAt: row.updated_at
+      };
+    },
+    async getThread(id) {
+      const [row] = await sql`
+        SELECT id, title, owner_id, created_at, updated_at FROM zaikit_threads WHERE id = ${id}
+      `;
+      if (!row) return null;
+      return {
+        id: row.id,
+        title: row.title,
+        ownerId: row.owner_id,
+        createdAt: row.created_at,
+        updatedAt: row.updated_at
+      };
+    },
+    async listThreads(opts) {
+      const rows = opts?.ownerId ? await sql`
+            SELECT id, title, owner_id, created_at, updated_at FROM zaikit_threads
+            WHERE owner_id = ${opts.ownerId}
+            ORDER BY updated_at DESC
+          ` : await sql`
+            SELECT id, title, owner_id, created_at, updated_at FROM zaikit_threads ORDER BY updated_at DESC
+          `;
+      return rows.map((row) => ({
+        id: row.id,
+        title: row.title,
+        ownerId: row.owner_id,
+        createdAt: row.created_at,
+        updatedAt: row.updated_at
+      }));
+    },
+    async deleteThread(id) {
+      await sql`DELETE FROM zaikit_threads WHERE id = ${id}`;
+    },
+    async updateThread(id, updates) {
+      const [row] = await sql`
+        UPDATE zaikit_threads
+        SET
+          title = COALESCE(${updates.title ?? null}, title),
+          owner_id = COALESCE(${updates.ownerId ?? null}, owner_id),
+          updated_at = NOW()
+        WHERE id = ${id}
+        RETURNING id, title, owner_id, created_at, updated_at
+      `;
+      if (!row) throw new Error(`Thread not found: ${id}`);
+      return {
+        id: row.id,
+        title: row.title,
+        ownerId: row.owner_id,
+        createdAt: row.created_at,
+        updatedAt: row.updated_at
+      };
+    },
+    async getMessages(threadId) {
+      const rows = await sql`
+        SELECT id, role, parts, metadata FROM zaikit_messages
+        WHERE thread_id = ${threadId}
+        ORDER BY created_at
+      `;
+      return rows.map(
+        (row) => ({
+          id: row.id,
+          role: row.role,
+          parts: row.parts,
+          ...row.metadata ? { metadata: row.metadata } : {}
+        })
+      );
+    },
+    async addMessage(threadId, message) {
+      await sql`
+        INSERT INTO zaikit_messages (id, thread_id, role, parts, metadata)
+        VALUES (
+          ${message.id},
+          ${threadId},
+          ${message.role},
+          ${sql.json(message.parts)},
+          ${message.metadata ? sql.json(message.metadata) : null}
+        )
+        ON CONFLICT (id) DO UPDATE SET
+          parts = ${sql.json(message.parts)},
+          metadata = ${message.metadata ? sql.json(message.metadata) : null}
+      `;
+      await sql`
+        UPDATE zaikit_threads SET updated_at = NOW() WHERE id = ${threadId}
+      `;
+    },
+    async updateMessage(threadId, messageId, updates) {
+      await sql`
+        UPDATE zaikit_messages
+        SET parts = ${sql.json(updates.parts)}
+        WHERE id = ${messageId} AND thread_id = ${threadId}
+      `;
+      await sql`
+        UPDATE zaikit_threads SET updated_at = NOW() WHERE id = ${threadId}
+      `;
+    }
+  };
+  return { ...memory, initialize };
+}
+export {
+  createPostgresMemory
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["import type { Memory } from \"@zaikit/core\";\nimport type { UIMessage } from \"ai\";\nimport postgres from \"postgres\";\n\ntype PostgresMemoryOptions = {\n  connectionString: string;\n};\n\nexport function createPostgresMemory({\n  connectionString,\n}: PostgresMemoryOptions) {\n  const sql = postgres(connectionString);\n\n  async function initialize() {\n    await sql`\n      CREATE TABLE IF NOT EXISTS zaikit_threads (\n        id TEXT PRIMARY KEY,\n        title TEXT,\n        owner_id TEXT,\n        created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),\n        updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW()\n      )\n    `;\n    await sql`\n      CREATE INDEX IF NOT EXISTS idx_zaikit_threads_owner_id ON zaikit_threads(owner_id)\n    `;\n    await sql`\n      CREATE TABLE IF NOT EXISTS zaikit_messages (\n        id TEXT PRIMARY KEY,\n        thread_id TEXT NOT NULL REFERENCES zaikit_threads(id) ON DELETE CASCADE,\n        role TEXT NOT NULL,\n        parts JSONB NOT NULL DEFAULT '[]',\n        metadata JSONB,\n        created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()\n      )\n    `;\n    await sql`\n      CREATE INDEX IF NOT EXISTS idx_zaikit_messages_thread_id ON zaikit_messages(thread_id, created_at)\n    `;\n  }\n\n  const memory: Memory = {\n    async createThread(id, title, ownerId) {\n      const [row] = await sql`\n        INSERT INTO zaikit_threads (id, title, owner_id)\n        VALUES (${id}, ${title ?? null}, ${ownerId ?? null})\n        RETURNING id, title, owner_id, created_at, updated_at\n      `;\n      return {\n        id: row.id,\n        title: row.title,\n        ownerId: row.owner_id,\n        createdAt: row.created_at,\n        updatedAt: row.updated_at,\n      };\n    },\n\n    async getThread(id) {\n      const [row] = await sql`\n        SELECT id, title, owner_id, created_at, updated_at FROM zaikit_threads WHERE id = ${id}\n      `;\n      if (!row) return null;\n      return {\n        id: row.id,\n        title: row.title,\n        ownerId: row.owner_id,\n        createdAt: row.created_at,\n        updatedAt: row.updated_at,\n      };\n    },\n\n    async listThreads(opts) {\n      const rows = opts?.ownerId\n        ? await sql`\n            SELECT id, title, owner_id, created_at, updated_at FROM zaikit_threads\n            WHERE owner_id = ${opts.ownerId}\n            ORDER BY updated_at DESC\n          `\n        : await sql`\n            SELECT id, title, owner_id, created_at, updated_at FROM zaikit_threads ORDER BY updated_at DESC\n          `;\n      return rows.map((row) => ({\n        id: row.id,\n        title: row.title,\n        ownerId: row.owner_id,\n        createdAt: row.created_at,\n        updatedAt: row.updated_at,\n      }));\n    },\n\n    async deleteThread(id) {\n      await sql`DELETE FROM zaikit_threads WHERE id = ${id}`;\n    },\n\n    async updateThread(id, updates) {\n      const [row] = await sql`\n        UPDATE zaikit_threads\n        SET\n          title = COALESCE(${updates.title ?? null}, title),\n          owner_id = COALESCE(${updates.ownerId ?? null}, owner_id),\n          updated_at = NOW()\n        WHERE id = ${id}\n        RETURNING id, title, owner_id, created_at, updated_at\n      `;\n      if (!row) throw new Error(`Thread not found: ${id}`);\n      return {\n        id: row.id,\n        title: row.title,\n        ownerId: row.owner_id,\n        createdAt: row.created_at,\n        updatedAt: row.updated_at,\n      };\n    },\n\n    async getMessages(threadId) {\n      const rows = await sql`\n        SELECT id, role, parts, metadata FROM zaikit_messages\n        WHERE thread_id = ${threadId}\n        ORDER BY created_at\n      `;\n      return rows.map(\n        (row): UIMessage => ({\n          id: row.id,\n          role: row.role,\n          parts: row.parts,\n          ...(row.metadata ? { metadata: row.metadata } : {}),\n        }),\n      );\n    },\n\n    async addMessage(threadId, message) {\n      await sql`\n        INSERT INTO zaikit_messages (id, thread_id, role, parts, metadata)\n        VALUES (\n          ${message.id},\n          ${threadId},\n          ${message.role},\n          ${sql.json(message.parts as any)},\n          ${message.metadata ? sql.json(message.metadata as any) : null}\n        )\n        ON CONFLICT (id) DO UPDATE SET\n          parts = ${sql.json(message.parts as any)},\n          metadata = ${message.metadata ? sql.json(message.metadata as any) : null}\n      `;\n      await sql`\n        UPDATE zaikit_threads SET updated_at = NOW() WHERE id = ${threadId}\n      `;\n    },\n\n    async updateMessage(threadId, messageId, updates) {\n      await sql`\n        UPDATE zaikit_messages\n        SET parts = ${sql.json(updates.parts as any)}\n        WHERE id = ${messageId} AND thread_id = ${threadId}\n      `;\n      await sql`\n        UPDATE zaikit_threads SET updated_at = NOW() WHERE id = ${threadId}\n      `;\n    },\n  };\n\n  return { ...memory, initialize };\n}\n"],"mappings":";AAEA,OAAO,cAAc;AAMd,SAAS,qBAAqB;AAAA,EACnC;AACF,GAA0B;AACxB,QAAM,MAAM,SAAS,gBAAgB;AAErC,iBAAe,aAAa;AAC1B,UAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AASN,UAAM;AAAA;AAAA;AAGN,UAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAUN,UAAM;AAAA;AAAA;AAAA,EAGR;AAEA,QAAM,SAAiB;AAAA,IACrB,MAAM,aAAa,IAAI,OAAO,SAAS;AACrC,YAAM,CAAC,GAAG,IAAI,MAAM;AAAA;AAAA,kBAER,EAAE,KAAK,SAAS,IAAI,KAAK,WAAW,IAAI;AAAA;AAAA;AAGpD,aAAO;AAAA,QACL,IAAI,IAAI;AAAA,QACR,OAAO,IAAI;AAAA,QACX,SAAS,IAAI;AAAA,QACb,WAAW,IAAI;AAAA,QACf,WAAW,IAAI;AAAA,MACjB;AAAA,IACF;AAAA,IAEA,MAAM,UAAU,IAAI;AAClB,YAAM,CAAC,GAAG,IAAI,MAAM;AAAA,4FACkE,EAAE;AAAA;AAExF,UAAI,CAAC,IAAK,QAAO;AACjB,aAAO;AAAA,QACL,IAAI,IAAI;AAAA,QACR,OAAO,IAAI;AAAA,QACX,SAAS,IAAI;AAAA,QACb,WAAW,IAAI;AAAA,QACf,WAAW,IAAI;AAAA,MACjB;AAAA,IACF;AAAA,IAEA,MAAM,YAAY,MAAM;AACtB,YAAM,OAAO,MAAM,UACf,MAAM;AAAA;AAAA,+BAEe,KAAK,OAAO;AAAA;AAAA,cAGjC,MAAM;AAAA;AAAA;AAGV,aAAO,KAAK,IAAI,CAAC,SAAS;AAAA,QACxB,IAAI,IAAI;AAAA,QACR,OAAO,IAAI;AAAA,QACX,SAAS,IAAI;AAAA,QACb,WAAW,IAAI;AAAA,QACf,WAAW,IAAI;AAAA,MACjB,EAAE;AAAA,IACJ;AAAA,IAEA,MAAM,aAAa,IAAI;AACrB,YAAM,4CAA4C,EAAE;AAAA,IACtD;AAAA,IAEA,MAAM,aAAa,IAAI,SAAS;AAC9B,YAAM,CAAC,GAAG,IAAI,MAAM;AAAA;AAAA;AAAA,6BAGG,QAAQ,SAAS,IAAI;AAAA,gCAClB,QAAQ,WAAW,IAAI;AAAA;AAAA,qBAElC,EAAE;AAAA;AAAA;AAGjB,UAAI,CAAC,IAAK,OAAM,IAAI,MAAM,qBAAqB,EAAE,EAAE;AACnD,aAAO;AAAA,QACL,IAAI,IAAI;AAAA,QACR,OAAO,IAAI;AAAA,QACX,SAAS,IAAI;AAAA,QACb,WAAW,IAAI;AAAA,QACf,WAAW,IAAI;AAAA,MACjB;AAAA,IACF;AAAA,IAEA,MAAM,YAAY,UAAU;AAC1B,YAAM,OAAO,MAAM;AAAA;AAAA,4BAEG,QAAQ;AAAA;AAAA;AAG9B,aAAO,KAAK;AAAA,QACV,CAAC,SAAoB;AAAA,UACnB,IAAI,IAAI;AAAA,UACR,MAAM,IAAI;AAAA,UACV,OAAO,IAAI;AAAA,UACX,GAAI,IAAI,WAAW,EAAE,UAAU,IAAI,SAAS,IAAI,CAAC;AAAA,QACnD;AAAA,MACF;AAAA,IACF;AAAA,IAEA,MAAM,WAAW,UAAU,SAAS;AAClC,YAAM;AAAA;AAAA;AAAA,YAGA,QAAQ,EAAE;AAAA,YACV,QAAQ;AAAA,YACR,QAAQ,IAAI;AAAA,YACZ,IAAI,KAAK,QAAQ,KAAY,CAAC;AAAA,YAC9B,QAAQ,WAAW,IAAI,KAAK,QAAQ,QAAe,IAAI,IAAI;AAAA;AAAA;AAAA,oBAGnD,IAAI,KAAK,QAAQ,KAAY,CAAC;AAAA,uBAC3B,QAAQ,WAAW,IAAI,KAAK,QAAQ,QAAe,IAAI,IAAI;AAAA;AAE5E,YAAM;AAAA,kEACsD,QAAQ;AAAA;AAAA,IAEtE;AAAA,IAEA,MAAM,cAAc,UAAU,WAAW,SAAS;AAChD,YAAM;AAAA;AAAA,sBAEU,IAAI,KAAK,QAAQ,KAAY,CAAC;AAAA,qBAC/B,SAAS,oBAAoB,QAAQ;AAAA;AAEpD,YAAM;AAAA,kEACsD,QAAQ;AAAA;AAAA,IAEtE;AAAA,EACF;AAEA,SAAO,EAAE,GAAG,QAAQ,WAAW;AACjC;","names":[]}

package/package.json

@@ -0,0 +1,30 @@
+{
+  "name": "@zaikit/memory-postgres",
+  "version": "0.1.0",
+  "license": "MIT",
+  "type": "module",
+  "files": [
+    "dist"
+  ],
+  "exports": {
+    ".": {
+      "import": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    }
+  },
+  "dependencies": {
+    "ai": "^6",
+    "postgres": "^3",
+    "@zaikit/core": "0.1.0"
+  },
+  "devDependencies": {
+    "tsup": "^8",
+    "typescript": "^5",
+    "@zaikit/typescript-config": "0.0.0"
+  },
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch",
+    "check-types": "tsc --noEmit"
+  }
+}