@undefineds.co/xpod 0.1.0-local.202602081751

package/dist/api/chatkit/store.js

@@ -0,0 +1,178 @@
+"use strict";
+/**
+ * ChatKit Store Interface
+ *
+ * Abstract interface for persisting threads and items.
+ * Based on https://github.com/openai/chatkit-python/blob/main/chatkit/store.py
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InMemoryStore = void 0;
+const types_1 = require("./types");
+/**
+ * In-Memory Store Implementation
+ *
+ * Simple in-memory store for development and testing.
+ * Data is lost when the server restarts.
+ */
+class InMemoryStore {
+    constructor() {
+        this.threads = new Map();
+        this.items = new Map(); // threadId -> (itemId -> item)
+        this.attachments = new Map();
+    }
+    // Store per-user data using a key from context
+    getUserKey(context) {
+        const userId = context.userId;
+        return typeof userId === 'string' ? userId : 'default';
+    }
+    getThreadKey(threadId, context) {
+        return `${this.getUserKey(context)}:${threadId}`;
+    }
+    // ID Generation
+    generateThreadId(_context) {
+        return (0, types_1.generateId)('thread');
+    }
+    generateItemId(itemType, _thread, _context) {
+        return (0, types_1.generateId)(itemType);
+    }
+    // Thread operations
+    async loadThread(threadId, context) {
+        const key = this.getThreadKey(threadId, context);
+        const thread = this.threads.get(key);
+        if (!thread) {
+            throw new Error(`Thread not found: ${threadId}`);
+        }
+        return thread;
+    }
+    async saveThread(thread, context) {
+        const key = this.getThreadKey(thread.id, context);
+        this.threads.set(key, { ...thread, updated_at: (0, types_1.nowTimestamp)() });
+    }
+    async loadThreads(limit, after, order, context) {
+        const userKey = this.getUserKey(context);
+        const prefix = `${userKey}:`;
+        // Get all threads for this user
+        const userThreads = [];
+        for (const [key, thread] of this.threads) {
+            if (key.startsWith(prefix)) {
+                userThreads.push(thread);
+            }
+        }
+        // Sort by created_at
+        userThreads.sort((a, b) => {
+            return order === 'desc' ? b.created_at - a.created_at : a.created_at - b.created_at;
+        });
+        // Apply pagination
+        let startIndex = 0;
+        if (after) {
+            const afterIndex = userThreads.findIndex((t) => t.id === after);
+            if (afterIndex !== -1) {
+                startIndex = afterIndex + 1;
+            }
+        }
+        const slice = userThreads.slice(startIndex, startIndex + limit);
+        const hasMore = startIndex + limit < userThreads.length;
+        return {
+            data: slice,
+            has_more: hasMore,
+            after: slice.length > 0 ? slice[slice.length - 1].id : undefined,
+        };
+    }
+    async deleteThread(threadId, context) {
+        const key = this.getThreadKey(threadId, context);
+        this.threads.delete(key);
+        this.items.delete(key);
+    }
+    // Thread Item operations
+    async loadThreadItems(threadId, after, limit, order, context) {
+        const key = this.getThreadKey(threadId, context);
+        const threadItems = this.items.get(key);
+        if (!threadItems) {
+            return { data: [], has_more: false };
+        }
+        const itemList = Array.from(threadItems.values());
+        // Sort by created_at
+        itemList.sort((a, b) => {
+            return order === 'desc' ? b.created_at - a.created_at : a.created_at - b.created_at;
+        });
+        // Apply pagination
+        let startIndex = 0;
+        if (after) {
+            const afterIndex = itemList.findIndex((item) => item.id === after);
+            if (afterIndex !== -1) {
+                startIndex = afterIndex + 1;
+            }
+        }
+        const slice = itemList.slice(startIndex, startIndex + limit);
+        const hasMore = startIndex + limit < itemList.length;
+        return {
+            data: slice,
+            has_more: hasMore,
+            after: slice.length > 0 ? slice[slice.length - 1].id : undefined,
+        };
+    }
+    async addThreadItem(threadId, item, context) {
+        const key = this.getThreadKey(threadId, context);
+        let threadItems = this.items.get(key);
+        if (!threadItems) {
+            threadItems = new Map();
+            this.items.set(key, threadItems);
+        }
+        threadItems.set(item.id, item);
+    }
+    async saveItem(threadId, item, context) {
+        await this.addThreadItem(threadId, item, context);
+    }
+    async loadItem(threadId, itemId, context) {
+        const key = this.getThreadKey(threadId, context);
+        const threadItems = this.items.get(key);
+        if (!threadItems) {
+            throw new Error(`Thread not found: ${threadId}`);
+        }
+        const item = threadItems.get(itemId);
+        if (!item) {
+            throw new Error(`Item not found: ${itemId}`);
+        }
+        return item;
+    }
+    async deleteThreadItem(threadId, itemId, context) {
+        const key = this.getThreadKey(threadId, context);
+        const threadItems = this.items.get(key);
+        if (threadItems) {
+            threadItems.delete(itemId);
+        }
+    }
+    // Attachment operations
+    async saveAttachment(attachment, _context) {
+        this.attachments.set(attachment.id, attachment);
+    }
+    async loadAttachment(attachmentId, _context) {
+        const attachment = this.attachments.get(attachmentId);
+        if (!attachment) {
+            throw new Error(`Attachment not found: ${attachmentId}`);
+        }
+        return attachment;
+    }
+    async deleteAttachment(attachmentId, _context) {
+        this.attachments.delete(attachmentId);
+    }
+    // Utility methods for debugging
+    getStats() {
+        let totalItems = 0;
+        for (const threadItems of this.items.values()) {
+            totalItems += threadItems.size;
+        }
+        return {
+            threads: this.threads.size,
+            items: totalItems,
+            attachments: this.attachments.size,
+        };
+    }
+    clear() {
+        this.threads.clear();
+        this.items.clear();
+        this.attachments.clear();
+    }
+}
+exports.InMemoryStore = InMemoryStore;
+//# sourceMappingURL=store.js.map

package/dist/api/chatkit/store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"store.js","sourceRoot":"","sources":["../../../src/api/chatkit/store.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AASH,mCAAmD;AAkCnD;;;;;GAKG;AACH,MAAa,aAAa;IAA1B;QACU,YAAO,GAAgC,IAAI,GAAG,EAAE,CAAC;QACjD,UAAK,GAAyC,IAAI,GAAG,EAAE,CAAC,CAAC,+BAA+B;QACxF,gBAAW,GAA4B,IAAI,GAAG,EAAE,CAAC;IAuL3D,CAAC;IArLC,+CAA+C;IACvC,UAAU,CAAC,OAAiB;QAClC,MAAM,MAAM,GAAI,OAAmC,CAAC,MAAM,CAAC;QAC3D,OAAO,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;IACzD,CAAC;IAEO,YAAY,CAAC,QAAgB,EAAE,OAAiB;QACtD,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,QAAQ,EAAE,CAAC;IACnD,CAAC;IAED,gBAAgB;IAChB,gBAAgB,CAAC,QAAkB;QACjC,OAAO,IAAA,kBAAU,EAAC,QAAQ,CAAC,CAAC;IAC9B,CAAC;IAED,cAAc,CAAC,QAAuB,EAAE,OAAuB,EAAE,QAAkB;QACjF,OAAO,IAAA,kBAAU,EAAC,QAAQ,CAAC,CAAC;IAC9B,CAAC;IAED,oBAAoB;IACpB,KAAK,CAAC,UAAU,CAAC,QAAgB,EAAE,OAAiB;QAClD,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACrC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;QACnD,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,MAAsB,EAAE,OAAiB;QACxD,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAClD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,GAAG,MAAM,EAAE,UAAU,EAAE,IAAA,oBAAY,GAAE,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAa,EAAE,KAAyB,EAAE,KAAa,EAAE,OAAiB;QAC1F,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,GAAG,OAAO,GAAG,CAAC;QAE7B,gCAAgC;QAChC,MAAM,WAAW,GAAqB,EAAE,CAAC;QACzC,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACzC,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC3B,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;QAED,qBAAqB;QACrB,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACxB,OAAO,KAAK,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC;QACtF,CAAC,CAAC,CAAC;QAEH,mBAAmB;QACnB,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,UAAU,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,CAAC;YAChE,IAAI,UAAU,KAAK,CAAC,CAAC,EAAE,CAAC;gBACtB,UAAU,GAAG,UAAU,GAAG,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QAED,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,UAAU,EAAE,UAAU,GAAG,KAAK,CAAC,CAAC;QAChE,MAAM,OAAO,GAAG,UAAU,GAAG,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC;QAExD,OAAO;YACL,IAAI,EAAE,KAAK;YACX,QAAQ,EAAE,OAAO;YACjB,KAAK,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;SACjE,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,QAAgB,EAAE,OAAiB;QACpD,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACjD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACzB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC;IAED,yBAAyB;IACzB,KAAK,CAAC,eAAe,CAAC,QAAgB,EAAE,KAAyB,EAAE,KAAa,EAAE,KAAa,EAAE,OAAiB;QAChH,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACjD,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAExC,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QACvC,CAAC;QAED,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;QAElD,qBAAqB;QACrB,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACrB,OAAO,KAAK,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC;QACtF,CAAC,CAAC,CAAC;QAEH,mBAAmB;QACnB,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,UAAU,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,KAAK,CAAC,CAAC;YACnE,IAAI,UAAU,KAAK,CAAC,CAAC,EAAE,CAAC;gBACtB,UAAU,GAAG,UAAU,GAAG,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QAED,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,UAAU,EAAE,UAAU,GAAG,KAAK,CAAC,CAAC;QAC7D,MAAM,OAAO,GAAG,UAAU,GAAG,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC;QAErD,OAAO;YACL,IAAI,EAAE,KAAK;YACX,QAAQ,EAAE,OAAO;YACjB,KAAK,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;SACjE,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,QAAgB,EAAE,IAAgB,EAAE,OAAiB;QACvE,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACjD,IAAI,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACtC,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,WAAW,GAAG,IAAI,GAAG,EAAE,CAAC;YACxB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;QACnC,CAAC;QACD,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,IAAgB,EAAE,OAAiB;QAClE,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,MAAc,EAAE,OAAiB;QAChE,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACjD,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACxC,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;QACnD,CAAC;QACD,MAAM,IAAI,GAAG,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACrC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,mBAAmB,MAAM,EAAE,CAAC,CAAC;QAC/C,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,QAAgB,EAAE,MAAc,EAAE,OAAiB;QACxE,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACjD,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACxC,IAAI,WAAW,EAAE,CAAC;YAChB,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAED,wBAAwB;IACxB,KAAK,CAAC,cAAc,CAAC,UAAsB,EAAE,QAAkB;QAC7D,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,YAAoB,EAAE,QAAkB;QAC3D,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACtD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,yBAAyB,YAAY,EAAE,CAAC,CAAC;QAC3D,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,YAAoB,EAAE,QAAkB;QAC7D,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;IACxC,CAAC;IAED,gCAAgC;IAChC,QAAQ;QACN,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,KAAK,MAAM,WAAW,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;YAC9C,UAAU,IAAI,WAAW,CAAC,IAAI,CAAC;QACjC,CAAC;QACD,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI;YAC1B,KAAK,EAAE,UAAU;YACjB,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI;SACnC,CAAC;IACJ,CAAC;IAED,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;CACF;AA1LD,sCA0LC","sourcesContent":["/**\n * ChatKit Store Interface\n * \n * Abstract interface for persisting threads and items.\n * Based on https://github.com/openai/chatkit-python/blob/main/chatkit/store.py\n */\n\nimport type {\n  ThreadMetadata,\n  ThreadItem,\n  Attachment,\n  Page,\n  StoreItemType,\n} from './types';\nimport { generateId, nowTimestamp } from './types';\n\n/**\n * Context type for store operations (generic)\n */\nexport type StoreContext = Record<string, unknown>;\n\n/**\n * Abstract Store interface\n */\nexport interface ChatKitStore<TContext = StoreContext> {\n  // ID Generation\n  generateThreadId(context: TContext): string;\n  generateItemId(itemType: StoreItemType, thread: ThreadMetadata, context: TContext): string;\n\n  // Thread operations\n  loadThread(threadId: string, context: TContext): Promise<ThreadMetadata>;\n  saveThread(thread: ThreadMetadata, context: TContext): Promise<void>;\n  loadThreads(limit: number, after: string | undefined, order: string, context: TContext): Promise<Page<ThreadMetadata>>;\n  deleteThread(threadId: string, context: TContext): Promise<void>;\n\n  // Thread Item operations\n  loadThreadItems(threadId: string, after: string | undefined, limit: number, order: string, context: TContext): Promise<Page<ThreadItem>>;\n  addThreadItem(threadId: string, item: ThreadItem, context: TContext): Promise<void>;\n  saveItem(threadId: string, item: ThreadItem, context: TContext): Promise<void>;\n  loadItem(threadId: string, itemId: string, context: TContext): Promise<ThreadItem>;\n  deleteThreadItem(threadId: string, itemId: string, context: TContext): Promise<void>;\n\n  // Attachment operations (optional)\n  saveAttachment?(attachment: Attachment, context: TContext): Promise<void>;\n  loadAttachment?(attachmentId: string, context: TContext): Promise<Attachment>;\n  deleteAttachment?(attachmentId: string, context: TContext): Promise<void>;\n}\n\n/**\n * In-Memory Store Implementation\n * \n * Simple in-memory store for development and testing.\n * Data is lost when the server restarts.\n */\nexport class InMemoryStore<TContext = StoreContext> implements ChatKitStore<TContext> {\n  private threads: Map<string, ThreadMetadata> = new Map();\n  private items: Map<string, Map<string, ThreadItem>> = new Map(); // threadId -> (itemId -> item)\n  private attachments: Map<string, Attachment> = new Map();\n\n  // Store per-user data using a key from context\n  private getUserKey(context: TContext): string {\n    const userId = (context as Record<string, unknown>).userId;\n    return typeof userId === 'string' ? userId : 'default';\n  }\n\n  private getThreadKey(threadId: string, context: TContext): string {\n    return `${this.getUserKey(context)}:${threadId}`;\n  }\n\n  // ID Generation\n  generateThreadId(_context: TContext): string {\n    return generateId('thread');\n  }\n\n  generateItemId(itemType: StoreItemType, _thread: ThreadMetadata, _context: TContext): string {\n    return generateId(itemType);\n  }\n\n  // Thread operations\n  async loadThread(threadId: string, context: TContext): Promise<ThreadMetadata> {\n    const key = this.getThreadKey(threadId, context);\n    const thread = this.threads.get(key);\n    if (!thread) {\n      throw new Error(`Thread not found: ${threadId}`);\n    }\n    return thread;\n  }\n\n  async saveThread(thread: ThreadMetadata, context: TContext): Promise<void> {\n    const key = this.getThreadKey(thread.id, context);\n    this.threads.set(key, { ...thread, updated_at: nowTimestamp() });\n  }\n\n  async loadThreads(limit: number, after: string | undefined, order: string, context: TContext): Promise<Page<ThreadMetadata>> {\n    const userKey = this.getUserKey(context);\n    const prefix = `${userKey}:`;\n    \n    // Get all threads for this user\n    const userThreads: ThreadMetadata[] = [];\n    for (const [key, thread] of this.threads) {\n      if (key.startsWith(prefix)) {\n        userThreads.push(thread);\n      }\n    }\n\n    // Sort by created_at\n    userThreads.sort((a, b) => {\n      return order === 'desc' ? b.created_at - a.created_at : a.created_at - b.created_at;\n    });\n\n    // Apply pagination\n    let startIndex = 0;\n    if (after) {\n      const afterIndex = userThreads.findIndex((t) => t.id === after);\n      if (afterIndex !== -1) {\n        startIndex = afterIndex + 1;\n      }\n    }\n\n    const slice = userThreads.slice(startIndex, startIndex + limit);\n    const hasMore = startIndex + limit < userThreads.length;\n\n    return {\n      data: slice,\n      has_more: hasMore,\n      after: slice.length > 0 ? slice[slice.length - 1].id : undefined,\n    };\n  }\n\n  async deleteThread(threadId: string, context: TContext): Promise<void> {\n    const key = this.getThreadKey(threadId, context);\n    this.threads.delete(key);\n    this.items.delete(key);\n  }\n\n  // Thread Item operations\n  async loadThreadItems(threadId: string, after: string | undefined, limit: number, order: string, context: TContext): Promise<Page<ThreadItem>> {\n    const key = this.getThreadKey(threadId, context);\n    const threadItems = this.items.get(key);\n    \n    if (!threadItems) {\n      return { data: [], has_more: false };\n    }\n\n    const itemList = Array.from(threadItems.values());\n    \n    // Sort by created_at\n    itemList.sort((a, b) => {\n      return order === 'desc' ? b.created_at - a.created_at : a.created_at - b.created_at;\n    });\n\n    // Apply pagination\n    let startIndex = 0;\n    if (after) {\n      const afterIndex = itemList.findIndex((item) => item.id === after);\n      if (afterIndex !== -1) {\n        startIndex = afterIndex + 1;\n      }\n    }\n\n    const slice = itemList.slice(startIndex, startIndex + limit);\n    const hasMore = startIndex + limit < itemList.length;\n\n    return {\n      data: slice,\n      has_more: hasMore,\n      after: slice.length > 0 ? slice[slice.length - 1].id : undefined,\n    };\n  }\n\n  async addThreadItem(threadId: string, item: ThreadItem, context: TContext): Promise<void> {\n    const key = this.getThreadKey(threadId, context);\n    let threadItems = this.items.get(key);\n    if (!threadItems) {\n      threadItems = new Map();\n      this.items.set(key, threadItems);\n    }\n    threadItems.set(item.id, item);\n  }\n\n  async saveItem(threadId: string, item: ThreadItem, context: TContext): Promise<void> {\n    await this.addThreadItem(threadId, item, context);\n  }\n\n  async loadItem(threadId: string, itemId: string, context: TContext): Promise<ThreadItem> {\n    const key = this.getThreadKey(threadId, context);\n    const threadItems = this.items.get(key);\n    if (!threadItems) {\n      throw new Error(`Thread not found: ${threadId}`);\n    }\n    const item = threadItems.get(itemId);\n    if (!item) {\n      throw new Error(`Item not found: ${itemId}`);\n    }\n    return item;\n  }\n\n  async deleteThreadItem(threadId: string, itemId: string, context: TContext): Promise<void> {\n    const key = this.getThreadKey(threadId, context);\n    const threadItems = this.items.get(key);\n    if (threadItems) {\n      threadItems.delete(itemId);\n    }\n  }\n\n  // Attachment operations\n  async saveAttachment(attachment: Attachment, _context: TContext): Promise<void> {\n    this.attachments.set(attachment.id, attachment);\n  }\n\n  async loadAttachment(attachmentId: string, _context: TContext): Promise<Attachment> {\n    const attachment = this.attachments.get(attachmentId);\n    if (!attachment) {\n      throw new Error(`Attachment not found: ${attachmentId}`);\n    }\n    return attachment;\n  }\n\n  async deleteAttachment(attachmentId: string, _context: TContext): Promise<void> {\n    this.attachments.delete(attachmentId);\n  }\n\n  // Utility methods for debugging\n  getStats(): { threads: number; items: number; attachments: number } {\n    let totalItems = 0;\n    for (const threadItems of this.items.values()) {\n      totalItems += threadItems.size;\n    }\n    return {\n      threads: this.threads.size,\n      items: totalItems,\n      attachments: this.attachments.size,\n    };\n  }\n\n  clear(): void {\n    this.threads.clear();\n    this.items.clear();\n    this.attachments.clear();\n  }\n}\n"]}

package/dist/api/chatkit/types.d.ts

@@ -0,0 +1,461 @@
+/**
+ * ChatKit Types
+ *
+ * Type definitions for OpenAI ChatKit protocol.
+ * Based on https://github.com/openai/chatkit-python
+ */
+export interface Page<T> {
+    data: T[];
+    has_more: boolean;
+    after?: string;
+}
+export type ThreadStatusType = 'active' | 'locked' | 'closed';
+export interface ActiveStatus {
+    type: 'active';
+}
+export interface LockedStatus {
+    type: 'locked';
+    reason?: string;
+}
+export interface ClosedStatus {
+    type: 'closed';
+}
+export type ThreadStatus = ActiveStatus | LockedStatus | ClosedStatus;
+export interface ThreadMetadata {
+    id: string;
+    title?: string;
+    status: ThreadStatus;
+    created_at: number;
+    updated_at: number;
+    metadata?: Record<string, unknown>;
+}
+export interface Thread extends ThreadMetadata {
+    items: Page<ThreadItem>;
+}
+export interface ThreadItemBase {
+    id: string;
+    thread_id: string;
+    created_at: number;
+}
+export interface UserMessageTextContent {
+    type: 'input_text';
+    text: string;
+}
+export interface UserMessageTagContent {
+    type: 'input_tag';
+    tag: string;
+    label?: string;
+}
+export type UserMessageContent = UserMessageTextContent | UserMessageTagContent;
+export interface InferenceOptions {
+    model?: string;
+    temperature?: number;
+    max_tokens?: number;
+    tools?: unknown[];
+    tool_choice?: ToolChoice;
+}
+export interface ToolChoice {
+    type: 'function';
+    function: {
+        name: string;
+    };
+}
+export interface UserMessageItem extends ThreadItemBase {
+    type: 'user_message';
+    content: UserMessageContent[];
+    inference_options?: InferenceOptions;
+}
+export interface Annotation {
+    type: string;
+    text: string;
+    start_index?: number;
+    end_index?: number;
+    file_citation?: {
+        file_id: string;
+        quote?: string;
+    };
+    file_path?: {
+        file_id: string;
+    };
+    url_citation?: {
+        url: string;
+        title?: string;
+    };
+}
+export interface AssistantMessageContent {
+    type: 'output_text';
+    text: string;
+    annotations?: Annotation[];
+}
+export interface AssistantMessageItem extends ThreadItemBase {
+    type: 'assistant_message';
+    content: AssistantMessageContent[];
+    status?: 'in_progress' | 'completed' | 'incomplete';
+}
+export interface ClientToolCallItem extends ThreadItemBase {
+    type: 'client_tool_call';
+    name: string;
+    arguments: string;
+    call_id: string;
+    status?: 'pending' | 'completed';
+    output?: string;
+}
+export interface WidgetItem extends ThreadItemBase {
+    type: 'widget';
+    widget_type: string;
+    data: unknown;
+}
+export interface GeneratedImage {
+    id: string;
+    url?: string;
+    b64_json?: string;
+}
+export interface GeneratedImageItem extends ThreadItemBase {
+    type: 'generated_image';
+    image: GeneratedImage;
+    prompt?: string;
+    status?: 'in_progress' | 'completed' | 'failed';
+}
+export interface BaseTask {
+    id: string;
+    status: 'pending' | 'in_progress' | 'completed' | 'failed';
+}
+export interface CustomTask extends BaseTask {
+    type: 'custom';
+    title: string;
+    description?: string;
+}
+export interface SearchTask extends BaseTask {
+    type: 'web_search';
+    query: string;
+}
+export interface ThoughtTask extends BaseTask {
+    type: 'thought';
+    content: string;
+}
+export interface FileTask extends BaseTask {
+    type: 'file';
+    file_id: string;
+    filename?: string;
+}
+export interface ImageTask extends BaseTask {
+    type: 'image';
+    image_id: string;
+}
+export type Task = CustomTask | SearchTask | ThoughtTask | FileTask | ImageTask;
+export interface TaskItem extends ThreadItemBase {
+    type: 'task';
+    task: Task;
+}
+export interface CustomSummary {
+    type: 'custom';
+    title: string;
+    icon?: string;
+}
+export interface DurationSummary {
+    type: 'duration';
+    duration_ms: number;
+}
+export type WorkflowSummary = CustomSummary | DurationSummary;
+export interface Workflow {
+    tasks: Task[];
+    summary?: WorkflowSummary;
+}
+export interface WorkflowItem extends ThreadItemBase {
+    type: 'workflow';
+    workflow: Workflow;
+}
+export interface EndOfTurnItem extends ThreadItemBase {
+    type: 'end_of_turn';
+}
+export interface HiddenContextItem extends ThreadItemBase {
+    type: 'hidden_context_item';
+    context: unknown;
+}
+export interface SDKHiddenContextItem extends ThreadItemBase {
+    type: 'sdk_hidden_context';
+    data: unknown;
+}
+export type ThreadItem = UserMessageItem | AssistantMessageItem | ClientToolCallItem | WidgetItem | GeneratedImageItem | TaskItem | WorkflowItem | EndOfTurnItem | HiddenContextItem | SDKHiddenContextItem;
+export type StoreItemType = ThreadItem['type'];
+export interface AttachmentBase {
+    id: string;
+    name: string;
+    size: number;
+    mime_type: string;
+    created_at: number;
+}
+export interface FileAttachment extends AttachmentBase {
+    type: 'file';
+    url?: string;
+}
+export interface ImageAttachment extends AttachmentBase {
+    type: 'image';
+    url?: string;
+    width?: number;
+    height?: number;
+}
+export type Attachment = FileAttachment | ImageAttachment;
+export interface AttachmentUploadDescriptor {
+    attachment_id: string;
+    upload_url: string;
+    upload_headers?: Record<string, string>;
+}
+export interface BaseReq {
+    metadata?: Record<string, unknown>;
+}
+export interface ThreadGetByIdParams {
+    thread_id: string;
+}
+export interface ThreadsGetByIdReq extends BaseReq {
+    type: 'threads.get_by_id';
+    params: ThreadGetByIdParams;
+}
+export interface ThreadCreateParams {
+    input?: UserMessageInput;
+}
+export interface UserMessageInput {
+    content: UserMessageContent[];
+    inference_options?: InferenceOptions;
+    attachments?: string[];
+}
+export interface ThreadsCreateReq extends BaseReq {
+    type: 'threads.create';
+    params: ThreadCreateParams;
+}
+export interface ThreadListParams {
+    limit?: number;
+    order?: 'asc' | 'desc';
+    after?: string;
+}
+export interface ThreadsListReq extends BaseReq {
+    type: 'threads.list';
+    params: ThreadListParams;
+}
+export interface ThreadAddUserMessageParams {
+    thread_id: string;
+    input: UserMessageInput;
+}
+export interface ThreadsAddUserMessageReq extends BaseReq {
+    type: 'threads.add_user_message';
+    params: ThreadAddUserMessageParams;
+}
+export interface ThreadAddClientToolOutputParams {
+    thread_id: string;
+    item_id: string;
+    output: string;
+}
+export interface ThreadsAddClientToolOutputReq extends BaseReq {
+    type: 'threads.add_client_tool_output';
+    params: ThreadAddClientToolOutputParams;
+}
+export interface ThreadCustomActionParams {
+    thread_id: string;
+    item_id: string;
+    action: string;
+    data?: unknown;
+}
+export interface ThreadsCustomActionReq extends BaseReq {
+    type: 'threads.custom_action';
+    params: ThreadCustomActionParams;
+}
+export interface ThreadRetryAfterItemParams {
+    thread_id: string;
+    item_id: string;
+}
+export interface ThreadsRetryAfterItemReq extends BaseReq {
+    type: 'threads.retry_after_item';
+    params: ThreadRetryAfterItemParams;
+}
+export interface ThreadUpdateParams {
+    thread_id: string;
+    title?: string;
+}
+export interface ThreadsUpdateReq extends BaseReq {
+    type: 'threads.update';
+    params: ThreadUpdateParams;
+}
+export interface ThreadDeleteParams {
+    thread_id: string;
+}
+export interface ThreadsDeleteReq extends BaseReq {
+    type: 'threads.delete';
+    params: ThreadDeleteParams;
+}
+export interface ItemsListParams {
+    thread_id: string;
+    limit?: number;
+    order?: 'asc' | 'desc';
+    after?: string;
+}
+export interface ItemsListReq extends BaseReq {
+    type: 'items.list';
+    params: ItemsListParams;
+}
+export interface ItemFeedbackParams {
+    thread_id: string;
+    item_ids: string[];
+    feedback: FeedbackKind;
+}
+export type FeedbackKind = 'positive' | 'negative';
+export interface ItemsFeedbackReq extends BaseReq {
+    type: 'items.feedback';
+    params: ItemFeedbackParams;
+}
+export interface AttachmentCreateParams {
+    name: string;
+    size: number;
+    mime_type: string;
+}
+export interface AttachmentsCreateReq extends BaseReq {
+    type: 'attachments.create';
+    params: AttachmentCreateParams;
+}
+export interface AttachmentDeleteParams {
+    attachment_id: string;
+}
+export interface AttachmentsDeleteReq extends BaseReq {
+    type: 'attachments.delete';
+    params: AttachmentDeleteParams;
+}
+export type StreamingReq = ThreadsCreateReq | ThreadsAddUserMessageReq | ThreadsAddClientToolOutputReq | ThreadsRetryAfterItemReq | ThreadsCustomActionReq;
+export type NonStreamingReq = ThreadsGetByIdReq | ThreadsListReq | ItemsListReq | ItemsFeedbackReq | AttachmentsCreateReq | AttachmentsDeleteReq | ThreadsUpdateReq | ThreadsDeleteReq;
+export type ChatKitReq = StreamingReq | NonStreamingReq;
+export interface ThreadCreatedEvent {
+    type: 'thread.created';
+    thread: ThreadMetadata;
+}
+export interface ThreadUpdatedEvent {
+    type: 'thread.updated';
+    thread: ThreadMetadata;
+}
+export interface ThreadItemAddedEvent {
+    type: 'thread.item.added';
+    item: ThreadItem;
+}
+export interface ThreadItemUpdatedEvent {
+    type: 'thread.item.updated';
+    item_id: string;
+    update: ThreadItemUpdate;
+}
+export interface ThreadItemDoneEvent {
+    type: 'thread.item.done';
+    item: ThreadItem;
+}
+export interface ThreadItemRemovedEvent {
+    type: 'thread.item.removed';
+    item_id: string;
+}
+export interface ThreadItemReplacedEvent {
+    type: 'thread.item.replaced';
+    item: ThreadItem;
+}
+export interface StreamOptions {
+    allow_cancel?: boolean;
+}
+export interface StreamOptionsEvent {
+    type: 'stream_options';
+    options: StreamOptions;
+}
+export interface ProgressUpdate {
+    progress: number;
+    message?: string;
+}
+export interface ProgressUpdateEvent {
+    type: 'progress_update';
+    update: ProgressUpdate;
+}
+export interface ClientEffect {
+    effect_type: string;
+    data?: unknown;
+}
+export interface ClientEffectEvent {
+    type: 'client_effect';
+    effect: ClientEffect;
+}
+export interface ErrorEvent {
+    type: 'error';
+    error: {
+        code: string;
+        message: string;
+        details?: unknown;
+    };
+}
+export interface NoticeEvent {
+    type: 'notice';
+    notice: {
+        level: 'info' | 'warning' | 'error';
+        message: string;
+    };
+}
+export type ThreadStreamEvent = ThreadCreatedEvent | ThreadUpdatedEvent | ThreadItemAddedEvent | ThreadItemUpdatedEvent | ThreadItemDoneEvent | ThreadItemRemovedEvent | ThreadItemReplacedEvent | StreamOptionsEvent | ProgressUpdateEvent | ClientEffectEvent | ErrorEvent | NoticeEvent;
+export interface AssistantMessageContentPartAdded {
+    type: 'assistant_message.content_part.added';
+    part_index: number;
+    part: AssistantMessageContent;
+}
+export interface AssistantMessageContentPartTextDelta {
+    type: 'assistant_message.content_part.text_delta';
+    part_index: number;
+    delta: string;
+}
+export interface AssistantMessageContentPartAnnotationAdded {
+    type: 'assistant_message.content_part.annotation_added';
+    part_index: number;
+    annotation: Annotation;
+}
+export interface AssistantMessageContentPartDone {
+    type: 'assistant_message.content_part.done';
+    part_index: number;
+}
+export interface WidgetStreamingTextValueDelta {
+    type: 'widget.streaming_text.value_delta';
+    path: string;
+    delta: string;
+}
+export interface WidgetRootUpdated {
+    type: 'widget.root.updated';
+    data: unknown;
+}
+export interface WidgetComponentUpdated {
+    type: 'widget.component.updated';
+    component_id: string;
+    data: unknown;
+}
+export interface WorkflowTaskAdded {
+    type: 'workflow.task.added';
+    task: Task;
+}
+export interface WorkflowTaskUpdated {
+    type: 'workflow.task.updated';
+    task_id: string;
+    task: Partial<Task>;
+}
+export interface GeneratedImageUpdated {
+    type: 'generated_image.updated';
+    image: GeneratedImage;
+}
+export type ThreadItemUpdate = AssistantMessageContentPartAdded | AssistantMessageContentPartTextDelta | AssistantMessageContentPartAnnotationAdded | AssistantMessageContentPartDone | WidgetStreamingTextValueDelta | WidgetRootUpdated | WidgetComponentUpdated | WorkflowTaskAdded | WorkflowTaskUpdated | GeneratedImageUpdated;
+export interface StreamingResult {
+    type: 'streaming';
+    stream: AsyncIterable<ThreadStreamEvent>;
+}
+export interface NonStreamingResult {
+    type: 'non_streaming';
+    data: unknown;
+}
+export type ChatKitResult = StreamingResult | NonStreamingResult;
+export declare function isStreamingReq(req: ChatKitReq): req is StreamingReq;
+export declare function isNonStreamingReq(req: ChatKitReq): req is NonStreamingReq;
+/**
+ * Generate a unique ID with prefix
+ */
+export declare function generateId(prefix: string): string;
+/**
+ * Get current timestamp in seconds
+ */
+export declare function nowTimestamp(): number;
+/**
+ * Extract text content from user message
+ */
+export declare function extractUserMessageText(content: UserMessageContent[]): string;