@shaxpir/duiduidui-models 1.17.3 → 1.19.0

package/dist/models/Chat.d.ts

@@ -0,0 +1,87 @@
+import { Doc } from '@shaxpir/sharedb/lib/client';
+import { CompactDateTime, MultiTime } from '@shaxpir/shaxpir-common';
+import { ShareSync } from '../repo';
+import { ArrayView } from './ArrayView';
+import { Content, ContentBody, ContentId, ContentMeta } from './Content';
+export interface TextBlock {
+    type: 'text';
+    text: string;
+}
+export interface ImageBlock {
+    type: 'image';
+    media_id: string;
+}
+export interface ToolUseBlock {
+    type: 'tool_use';
+    id: string;
+    name: string;
+    input: any;
+}
+export interface ToolResultBlock {
+    type: 'tool_result';
+    tool_use_id: string;
+    content: string;
+    is_error?: boolean;
+}
+export interface ThinkingBlock {
+    type: 'thinking';
+    thinking: string;
+}
+export type ContentBlock = TextBlock | ImageBlock | ToolUseBlock | ToolResultBlock | ThinkingBlock;
+export interface CacheBreakpoint {
+    ttl: 'ephemeral' | 'standard';
+    created_at: CompactDateTime;
+    expires_at: CompactDateTime;
+}
+export interface ChatUsage {
+    input_tokens?: number;
+    output_tokens?: number;
+    cache_creation_input_tokens?: number;
+    cache_read_input_tokens?: number;
+}
+export interface ChatMessage {
+    role: 'user' | 'assistant';
+    content: ContentBlock[];
+    timestamp: CompactDateTime;
+    system_prompt_hash?: string;
+    model?: string;
+    usage?: ChatUsage;
+    duration_ms?: number;
+    stop_reason?: string;
+    cache_breakpoint?: CacheBreakpoint;
+}
+export interface ChatScratchpad {
+    text?: string;
+    media_ids?: string[];
+    model?: string;
+    extended_thinking?: boolean;
+    [key: string]: any;
+}
+export interface ChatPayload {
+    title?: string;
+    messages: ChatMessage[];
+    scratchpad: ChatScratchpad;
+}
+export interface ChatBody extends ContentBody {
+    meta: ContentMeta;
+    payload: ChatPayload;
+}
+export declare class Chat extends Content {
+    private _messagesView;
+    constructor(doc: Doc, shouldAcquire: boolean, shareSync: ShareSync);
+    static makeChatId(userId: ContentId, chatKey: string): ContentId;
+    static createDefaultPayload(): ChatPayload;
+    static create(userId: ContentId, chatKey: string, payload?: ChatPayload, createdAt?: MultiTime): Chat;
+    get payload(): ChatPayload;
+    get title(): string | undefined;
+    get messages(): ArrayView<ChatMessage>;
+    get scratchpad(): ChatScratchpad;
+    setTitle(title: string): void;
+    setScratchpad(scratchpad: ChatScratchpad): void;
+    updateScratchpadField(key: string, value: any): void;
+    clearScratchpad(): void;
+    appendMessage(message: ChatMessage): void;
+    setBreakpoint(messageIndex: number, ttl: 'ephemeral' | 'standard'): void;
+    removeBreakpoint(messageIndex: number): void;
+    clearExpiredBreakpoints(): void;
+}

package/dist/models/Chat.js

@@ -0,0 +1,122 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Chat = void 0;
+const shaxpir_common_1 = require("@shaxpir/shaxpir-common");
+const repo_1 = require("../repo");
+const ArrayView_1 = require("./ArrayView");
+const Content_1 = require("./Content");
+const ContentKind_1 = require("./ContentKind");
+const Operation_1 = require("./Operation");
+// ---- Chat model ----
+class Chat extends Content_1.Content {
+    constructor(doc, shouldAcquire, shareSync) {
+        super(doc, shouldAcquire, shareSync);
+        this._messagesView = new ArrayView_1.ArrayView(this, ['payload', 'messages']);
+    }
+    // ---- ID generation ----
+    static makeChatId(userId, chatKey) {
+        return shaxpir_common_1.CachingHasher.makeMd5Base62Hash(`${ContentKind_1.ContentKind.CHAT}-${userId}-${chatKey}`);
+    }
+    // ---- Factory ----
+    static createDefaultPayload() {
+        return {
+            messages: [],
+            scratchpad: {},
+        };
+    }
+    static create(userId, chatKey, payload, createdAt) {
+        const now = shaxpir_common_1.ClockService.getClock().now();
+        createdAt ??= now;
+        payload ??= Chat.createDefaultPayload();
+        const chatId = Chat.makeChatId(userId, chatKey);
+        return repo_1.ShareSyncFactory.get().createContent({
+            meta: {
+                ref: chatId,
+                kind: ContentKind_1.ContentKind.CHAT,
+                id: chatId,
+                owner: userId,
+                created_at: createdAt,
+                updated_at: createdAt
+            },
+            payload
+        });
+    }
+    // ---- Getters ----
+    get payload() {
+        return this.doc.data.payload;
+    }
+    get title() {
+        this.checkDisposed('Chat.title');
+        return this.payload.title;
+    }
+    get messages() {
+        this.checkDisposed('Chat.messages');
+        return this._messagesView;
+    }
+    get scratchpad() {
+        this.checkDisposed('Chat.scratchpad');
+        return this.payload.scratchpad;
+    }
+    // ---- Setters ----
+    setTitle(title) {
+        this.checkDisposed('Chat.setTitle');
+        const batch = new Operation_1.BatchOperation(this);
+        batch.setPathValue(['payload', 'title'], title);
+        batch.commit();
+    }
+    setScratchpad(scratchpad) {
+        this.checkDisposed('Chat.setScratchpad');
+        const batch = new Operation_1.BatchOperation(this);
+        batch.setPathValue(['payload', 'scratchpad'], scratchpad);
+        batch.commit();
+    }
+    updateScratchpadField(key, value) {
+        this.checkDisposed('Chat.updateScratchpadField');
+        const batch = new Operation_1.BatchOperation(this);
+        batch.setPathValue(['payload', 'scratchpad', key], value);
+        batch.commit();
+    }
+    clearScratchpad() {
+        this.checkDisposed('Chat.clearScratchpad');
+        const batch = new Operation_1.BatchOperation(this);
+        batch.setPathValue(['payload', 'scratchpad'], {});
+        batch.commit();
+    }
+    // ---- Message operations ----
+    appendMessage(message) {
+        this.checkDisposed('Chat.appendMessage');
+        this._messagesView.push(message);
+    }
+    // ---- Cache breakpoint operations ----
+    setBreakpoint(messageIndex, ttl) {
+        this.checkDisposed('Chat.setBreakpoint');
+        if (messageIndex < 0 || messageIndex >= this._messagesView.length)
+            return;
+        const now = shaxpir_common_1.ClockService.getClock().utc();
+        const durationMinutes = ttl === 'ephemeral' ? 5 : 60;
+        const expiresAt = shaxpir_common_1.Time.plus(now, durationMinutes, 'minutes');
+        const breakpoint = {
+            ttl,
+            created_at: now,
+            expires_at: expiresAt,
+        };
+        this._messagesView.setObjectValueAtIndex(messageIndex, 'cache_breakpoint', breakpoint);
+    }
+    removeBreakpoint(messageIndex) {
+        this.checkDisposed('Chat.removeBreakpoint');
+        if (messageIndex < 0 || messageIndex >= this._messagesView.length)
+            return;
+        this._messagesView.removeObjectPropertyAtIndex(messageIndex, 'cache_breakpoint');
+    }
+    clearExpiredBreakpoints() {
+        this.checkDisposed('Chat.clearExpiredBreakpoints');
+        const now = shaxpir_common_1.ClockService.getClock().utc();
+        for (let i = 0; i < this._messagesView.length; i++) {
+            const message = this._messagesView.get(i);
+            if (message.cache_breakpoint && shaxpir_common_1.Time.isDateTimeBefore(message.cache_breakpoint.expires_at, now)) {
+                this._messagesView.removeObjectPropertyAtIndex(i, 'cache_breakpoint');
+            }
+        }
+    }
+}
+exports.Chat = Chat;

package/dist/models/Content.d.ts

@@ -2,6 +2,7 @@ import { Doc } from '@shaxpir/sharedb/lib/client';
 import { CompactDateTime, MultiTime } from '@shaxpir/shaxpir-common';
 import { ShareSync } from '../repo';
 import { BillingPayload } from './Billing';
+import { ChatPayload } from './Chat';
 import { CollectionPayload } from './Collection';
 import { ContentKind } from './ContentKind';
 import { DevicePayload } from './Device';
@@ -38,7 +39,7 @@ export interface ContentMeta {
     created_at: MultiTime;
     updated_at: MultiTime;
 }
-export type ContentPayload = BillingPayload | CollectionPayload | DevicePayload | ImagePayload | MetricPayload | ProfilePayload | ProgressPayload | SessionPayload | SocialPayload | TermPayload | UserPayload | WorkspacePayload;
+export type ContentPayload = BillingPayload | ChatPayload | CollectionPayload | DevicePayload | ImagePayload | MetricPayload | ProfilePayload | ProgressPayload | SessionPayload | SocialPayload | TermPayload | UserPayload | WorkspacePayload;
 export declare abstract class Content extends Model {
     static ID_LENGTH: number;
     constructor(doc: Doc, shouldAcquire: boolean, shareSync: ShareSync);

package/dist/models/ContentKind.d.ts

@@ -1,5 +1,6 @@
 export declare enum ContentKind {
     BILLING = "billing",
+    CHAT = "chat",
     COLLECTION = "collection",
     DEVICE = "device",
     METRIC = "metric",

package/dist/models/ContentKind.js

@@ -6,6 +6,7 @@ var ContentKind;
     // No models yet for these, but they're definitely coming
     ContentKind["BILLING"] = "billing";
     // These are all the current Content subclasses
+    ContentKind["CHAT"] = "chat";
     ContentKind["COLLECTION"] = "collection";
     ContentKind["DEVICE"] = "device";
     ContentKind["METRIC"] = "metric";

package/dist/models/Manifest.d.ts

@@ -21,6 +21,7 @@ export interface ManifestEntry {
 }
 export interface ManifestPayload {
     billing: any;
+    chat: any;
     collection: any;
     device: any;
     image: any;

package/dist/models/Manifest.js

@@ -27,6 +27,7 @@ class Manifest extends Model_1.Model {
             },
             payload: {
                 billing: {},
+                chat: {},
                 collection: {},
                 device: {},
                 image: {},

package/dist/models/Workspace.d.ts

@@ -14,6 +14,7 @@ export interface WorkspacePayload {
     devices: ContentId[];
     sessions: MultiTime[];
     collections: string[];
+    chats: ContentId[];
     journey: {
         [key: string]: CompactDateTime;
     };
@@ -29,6 +30,7 @@ export declare class Workspace extends Content {
     private _devicesView;
     private _sessionsView;
     private _collectionsView;
+    private _chatsView;
     private _uploadedAvatarsView;
     constructor(doc: Doc, shouldAcquire: boolean, shareSync: ShareSync);
     get payload(): WorkspacePayload;
@@ -42,6 +44,7 @@ export declare class Workspace extends Content {
     static create(userId: ContentId, payload: WorkspacePayload): Workspace;
     get sessions(): ArrayView<MultiTime>;
     get collections(): ArrayView<string>;
+    get chats(): ArrayView<ContentId>;
     get devices(): ArrayView<ContentId>;
     get uploadedAvatars(): ArrayView<ContentId>;
     acquireSessionForUtcTime(utcTime: CompactDateTime): Promise<Session | null>;

package/dist/models/Workspace.js

@@ -14,6 +14,7 @@ class Workspace extends Content_1.Content {
         this._devicesView = new ArrayView_1.ArrayView(this, ['payload', 'devices']);
         this._sessionsView = new ArrayView_1.ArrayView(this, ['payload', 'sessions']);
         this._collectionsView = new ArrayView_1.ArrayView(this, ['payload', 'collections']);
+        this._chatsView = new ArrayView_1.ArrayView(this, ['payload', 'chats']);
         this._uploadedAvatarsView = new ArrayView_1.ArrayView(this, ['payload', 'uploaded_avatars']);
     }
     get payload() {
@@ -43,6 +44,9 @@ class Workspace extends Content_1.Content {
         if (!Array.isArray(payload.collections)) {
             throw new Error('Workspace.create: payload.collections must be an array');
         }
+        if (!Array.isArray(payload.chats)) {
+            throw new Error('Workspace.create: payload.chats must be an array');
+        }
         if (!payload.journey || typeof payload.journey !== 'object' || Array.isArray(payload.journey)) {
             throw new Error('Workspace.create: payload.journey must be an object');
         }
@@ -77,6 +81,10 @@ class Workspace extends Content_1.Content {
         this.checkDisposed("Workspace.collections");
         return this._collectionsView;
     }
+    get chats() {
+        this.checkDisposed("Workspace.chats");
+        return this._chatsView;
+    }
     // NOTE: As the user adds devices, we will always add new devices to the end of the array.
     // So the most-recently-added device will always be at the end.
     get devices() {

package/dist/models/index.d.ts

@@ -1,6 +1,7 @@
 export * from './ArrayView';
 export * from './BayesianScore';
 export * from './Billing';
+export * from './Chat';
 export * from './CognitiveLoad';
 export * from './ChangeModel';
 export * from './Collection';

package/dist/models/index.js

@@ -18,6 +18,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./ArrayView"), exports);
 __exportStar(require("./BayesianScore"), exports);
 __exportStar(require("./Billing"), exports);
+__exportStar(require("./Chat"), exports);
 __exportStar(require("./CognitiveLoad"), exports);
 __exportStar(require("./ChangeModel"), exports);
 __exportStar(require("./Collection"), exports);

package/dist/util/index.d.ts

@@ -3,7 +3,6 @@ export * from './ConditionMatcher';
 export * from './Database';
 export * from './DifficultyRange';
 export * from './Encryption';
-export * from './Logging';
 export * from './PinyinParser';
 export * from './PinyinValidator';
 export * from './SearchPreprocessor';

package/dist/util/index.js

@@ -20,7 +20,6 @@ __exportStar(require("./ConditionMatcher"), exports);
 __exportStar(require("./Database"), exports);
 __exportStar(require("./DifficultyRange"), exports);
 __exportStar(require("./Encryption"), exports);
-__exportStar(require("./Logging"), exports);
 __exportStar(require("./PinyinParser"), exports);
 __exportStar(require("./PinyinValidator"), exports);
 __exportStar(require("./SearchPreprocessor"), exports);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shaxpir/duiduidui-models",
-  "version": "1.17.3",
+  "version": "1.19.0",
   "repository": {
     "type": "git",
     "url": "https://github.com/shaxpir/duiduidui-models"