@lichen-ai/lichen-sdk 0.1.0

package/dist/ui/manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../src/ui/manager.ts"],"names":[],"mappings":"OAaO,EAAuB,aAAa,EAAE;OACtC,EAAE,WAAW,EAAE;AAyEtB,MAAM,OAAO,aAAa;IAChB,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAA;IAEhC,QAAQ,CAAqB;IAE7B,SAAS,GAAG,IAAI,GAAG,EAAc,CAAA;IAEjC,KAAK,CAIZ;IAED,YAAY,QAA6B;QACvC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAA;QACxB,IAAI,CAAC,KAAK,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,CAAA;IACnE,CAAC;IAEO,QAAQ,GAAG,CAAC,QAAoC,EAAE,EAAE;QAC1D,IAAI,CAAC,KAAK,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,GAAG,QAAQ,EAAE,CAAA;QAC3C,IAAI,CAAC,eAAe,EAAE,CAAA;IACxB,CAAC,CAAA;IAEO,eAAe,GAAG,GAAG,EAAE;QAC7B,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAA;IAClD,CAAC,CAAA;IAED,SAAS,GAAG,CAAC,QAAoB,EAAgB,EAAE;QACjD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QAC5B,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;IAC9C,CAAC,CAAA;IAED,WAAW,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAA;IAE9B,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAA;IAC7B,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,CAAA;IACvC,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAA;IACzB,CAAC;IAED,eAAe,GAAG,CAChB,MAAoG,EACpG,OAA0B,QAAQ,EAClC,EAAE;QACF,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;YACjC,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAA;YACpE,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAA;YAC/C,IAAI,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,UAAU,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAA;QAC3E,CAAC;aAAM,CAAC;YACN,MAAM,UAAU,GAAG,MAAM,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;YACzD,IAAI,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,UAA4C,EAAE,CAAC,CAAA;QACzE,CAAC;IACH,CAAC,CAAA;IAEO,WAAW,GAAG,CAAC,IAAuB,EAAE,aAAwB,EAAE,EAAE;QAC1E,OAAO,CAAC,MAAsE,EAAE,EAAE;YAChF,MAAM,IAAI,GAAG;gBACX,GAAG,aAAa;gBAChB,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;aAC9C,CAAA;YACD,MAAM,IAAI,GAAG,OAAO,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAA;YACjE,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,IAAI,EAAE,EAAE,IAAI,CAAC,CAAA;QAClD,CAAC,CAAA;IACH,CAAC,CAAA;IAEO,cAAc,GAAG,CAGvB,QAAW,EACX,MAAoG,EACpG,KAAkG,EACK,EAAE;QACzG,OAAO,QAAQ,KAAK,MAAM,IAAI,MAAM,CAAC,UAAU,CAAC,GAAG,QAAQ,GAAG,CAAC,CAAA;IACjE,CAAC,CAAA;IAED,KAAK,GAAG,KAAK,EACX,MAEiH,EACjH,OAcC,EACc,EAAE;QACjB,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS;YAAE,OAAM;QAEhC,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;YACpD,IAAI,CAAC,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAA;YAErC,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;YAE9C,IAAI,WAAoC,CAAA;YACxC,IAAI,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,GAAG,EAAE,CAAC;gBACxC,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;oBACtB,WAAW,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,CAAA;oBACnC,MAAK;gBACP,CAAC;gBAED,MAAM,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;gBAE7E,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,OAAO,CAAC,aAAa,CAAC,CAAA;gBAEhE,IAAI,KAAK,KAAK,UAAU;oBAAE,OAAO,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC,IAAI,CAAC,CAAA;gBACnE,IAAI,KAAK,KAAK,QAAQ;oBAAE,OAAO,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,CAAA;gBAElE,IAAI,IAAI,CAAC,cAAc,CAAC,SAAS,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC;oBAChD,OAAO,CAAC,SAAS,CAAC,aAAa,EAAE,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAA;gBAChE,CAAC;gBAED,IAAI,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC;oBAC/C,OAAO,CAAC,SAAS,CAAC,aAAa,EAAE,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAA;gBAChE,CAAC;gBAED,IAAI,IAAI,CAAC,cAAc,CAAC,aAAa,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC;oBACpD,OAAO,CAAC,SAAS,CAAC,iBAAiB,EAAE,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,CAAC,CAAA;gBAC5D,CAAC;gBAED,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC;oBAC9C,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,CAAC,CAAA;gBACtD,CAAC;gBAED,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC;oBAC9C,OAAO,CAAC,SAAS,CAAC,YAAY,EAAE,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,CAAC,CAAA;gBACvD,CAAC;gBAED,IAAI,KAAK,KAAK,QAAQ,EAAE,CAAC;oBACvB,IAAI,eAAe,IAAI,IAAI,EAAE,CAAC;wBAC5B,IAAI,CAAC,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC,CAAA;oBACxD,CAAC;yBAAM,CAAC;wBACN,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAA;oBAC5B,CAAC;gBACH,CAAC;gBAED,IAAI,IAAI,CAAC,cAAc,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC;oBACjD,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,GAAG,IAAI,CAAA;oBAEnC,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAA;oBACzD,IAAI,CAAC,SAAS,EAAE,CAAC;wBACf,OAAO,CAAC,IAAI,CAAC,uDAAuD,CAAC,CAAA;wBACrE,SAAQ;oBACV,CAAC;oBAED,IAAI,CAAC,eAAe,CAAC,CAAC,YAAY,EAAE,EAAE;wBACpC,MAAM,MAAM,GAAG,EAAE,GAAG,OAAO,CAAC,aAAa,EAAE,GAAG,YAAY,EAAE,CAAA;wBAE5D,8CAA8C;wBAC9C,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,CAAA;wBACpD,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,CAAA;wBAE5E,IAAI,CAAC,KAAK,IAAI,KAAK,IAAI,IAAI;4BAAE,OAAO,MAAM,CAAA;wBAC1C,IAAI,KAAK,CAAC,OAAO,EAAE,KAAK,QAAQ,EAAE,CAAC;4BACjC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAA;wBAC3B,CAAC;6BAAM,CAAC;4BACN,QAAQ,CAAC,KAAK,CAAC,GAAG,aAAa,CAAC,KAAK,CAAC,CAAA;wBACxC,CAAC;wBAED,OAAO,OAAO,CAAC,WAAW,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;oBAC9C,CAAC,CAAC,CAAA;gBACJ,CAAC;YACH,CAAC;YAED,IAAI,WAAW,IAAI,IAAI;gBAAE,MAAM,WAAW,CAAA;YAE1C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,SAAS,EAAE,EAAE,CAAA;YAC1C,IAAI,OAAO,MAAM,KAAK,WAAW;gBAAE,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAA;QACjE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IACE,CAAC,CACC,KAAK,YAAY,KAAK,IAAI,kDAAkD;gBAC5E,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,CAAC,CAC/D,EACD,CAAC;gBACD,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBACpB,IAAI,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;gBACxB,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,CAAA;YAChC,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,QAAQ,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAA;YACnC,IAAI,CAAC,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAA;YACrC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAA;QACtB,CAAC;IACH,CAAC,CAAA;IAED,IAAI,GAAG,KAAK,EACV,aAAwB,EACxB,OAIC,EACc,EAAE;QACjB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAA;QACrB,IAAI,CAAC,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAA;QAErC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,EAAE,CAAC,CAAA;IACvE,CAAC,CAAA;IAED,KAAK,GAAG,GAAG,EAAE;QACX,6BAA6B;QAC7B,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAA;QACrB,IAAI,CAAC,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAA;QAErC,+BAA+B;QAC/B,IAAI,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAA;QAEjD,6BAA6B;QAC7B,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAA;IACvB,CAAC,CAAA;CACF"}

package/dist/ui/manager.mjs

@@ -0,0 +1,158 @@
+import { toMessageDict } from "./messages.mjs";
+import { StreamError } from "./errors.mjs";
+export class StreamManager {
+    abortRef = new AbortController();
+    messages;
+    listeners = new Set();
+    state;
+    constructor(messages) {
+        this.messages = messages;
+        this.state = { isLoading: false, values: null, error: undefined };
+    }
+    setState = (newState) => {
+        this.state = { ...this.state, ...newState };
+        this.notifyListeners();
+    };
+    notifyListeners = () => {
+        this.listeners.forEach((listener) => listener());
+    };
+    subscribe = (listener) => {
+        this.listeners.add(listener);
+        return () => this.listeners.delete(listener);
+    };
+    getSnapshot = () => this.state;
+    get isLoading() {
+        return this.state.isLoading;
+    }
+    get values() {
+        return this.state.values?.[0] ?? null;
+    }
+    get error() {
+        return this.state.error;
+    }
+    setStreamValues = (values, kind = 'stream') => {
+        if (typeof values === 'function') {
+            const [prevValues, prevKind] = this.state.values ?? [null, 'stream'];
+            const nextValues = values(prevValues, prevKind);
+            this.setState({ values: nextValues != null ? [nextValues, kind] : null });
+        }
+        else {
+            const nextValues = values != null ? [values, kind] : null;
+            this.setState({ values: nextValues });
+        }
+    };
+    getMutateFn = (kind, historyValues) => {
+        return (update) => {
+            const prev = {
+                ...historyValues,
+                ...(this.state.values ?? [null, 'stream'])[0]
+            };
+            const next = typeof update === 'function' ? update(prev) : update;
+            this.setStreamValues({ ...prev, ...next }, kind);
+        };
+    };
+    matchEventType = (expected, actual, _data) => {
+        return expected === actual || actual.startsWith(`${expected}|`);
+    };
+    start = async (action, options) => {
+        if (this.state.isLoading)
+            return;
+        try {
+            this.setState({ isLoading: true, error: undefined });
+            this.abortRef = new AbortController();
+            const run = await action(this.abortRef.signal);
+            let streamError;
+            for await (const { event, data } of run) {
+                if (event === 'error') {
+                    streamError = new StreamError(data);
+                    break;
+                }
+                const namespace = event.includes('|') ? event.split('|').slice(1) : undefined;
+                const mutate = this.getMutateFn('stream', options.initialValues);
+                if (event === 'metadata')
+                    options.callbacks.onMetadataEvent?.(data);
+                if (event === 'events')
+                    options.callbacks.onLangChainEvent?.(data);
+                if (this.matchEventType('updates', event, data)) {
+                    options.callbacks.onUpdateEvent?.(data, { namespace, mutate });
+                }
+                if (this.matchEventType('custom', event, data)) {
+                    options.callbacks.onCustomEvent?.(data, { namespace, mutate });
+                }
+                if (this.matchEventType('checkpoints', event, data)) {
+                    options.callbacks.onCheckpointEvent?.(data, { namespace });
+                }
+                if (this.matchEventType('tasks', event, data)) {
+                    options.callbacks.onTaskEvent?.(data, { namespace });
+                }
+                if (this.matchEventType('debug', event, data)) {
+                    options.callbacks.onDebugEvent?.(data, { namespace });
+                }
+                if (event === 'values') {
+                    if ('__interrupt__' in data) {
+                        this.setStreamValues((prev) => ({ ...prev, ...data }));
+                    }
+                    else {
+                        this.setStreamValues(data);
+                    }
+                }
+                if (this.matchEventType('messages', event, data)) {
+                    const [serialized, metadata] = data;
+                    const messageId = this.messages.add(serialized, metadata);
+                    if (!messageId) {
+                        console.warn('Failed to add message to manager, no message ID found');
+                        continue;
+                    }
+                    this.setStreamValues((streamValues) => {
+                        const values = { ...options.initialValues, ...streamValues };
+                        // Assumption: we're concatenating the message
+                        const messages = options.getMessages(values).slice();
+                        const { chunk, index } = this.messages.get(messageId, messages.length) ?? {};
+                        if (!chunk || index == null)
+                            return values;
+                        if (chunk.getType() === 'remove') {
+                            messages.splice(index, 1);
+                        }
+                        else {
+                            messages[index] = toMessageDict(chunk);
+                        }
+                        return options.setMessages(values, messages);
+                    });
+                }
+            }
+            if (streamError != null)
+                throw streamError;
+            const values = await options.onSuccess?.();
+            if (typeof values !== 'undefined')
+                this.setStreamValues(values);
+        }
+        catch (error) {
+            if (!(error instanceof Error && // eslint-disable-line no-instanceof/no-instanceof
+                (error.name === 'AbortError' || error.name === 'TimeoutError'))) {
+                console.error(error);
+                this.setState({ error });
+                await options.onError?.(error);
+            }
+        }
+        finally {
+            this.setState({ isLoading: false });
+            this.abortRef = new AbortController();
+            options.onFinish?.();
+        }
+    };
+    stop = async (historyValues, options) => {
+        this.abortRef.abort();
+        this.abortRef = new AbortController();
+        options.onStop?.({ mutate: this.getMutateFn('stop', historyValues) });
+    };
+    clear = () => {
+        // Cancel any running streams
+        this.abortRef.abort();
+        this.abortRef = new AbortController();
+        // Set the stream state to null
+        this.setState({ error: undefined, values: null });
+        // Clear any pending messages
+        this.messages.clear();
+    };
+}
+//# sourceMappingURL=manager.mjs.map

package/dist/ui/manager.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"manager.mjs","sourceRoot":"","sources":["../../src/ui/manager.ts"],"names":[],"mappings":"OAaO,EAAuB,aAAa,EAAE;OACtC,EAAE,WAAW,EAAE;AAyEtB,MAAM,OAAO,aAAa;IAChB,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAA;IAEhC,QAAQ,CAAqB;IAE7B,SAAS,GAAG,IAAI,GAAG,EAAc,CAAA;IAEjC,KAAK,CAIZ;IAED,YAAY,QAA6B;QACvC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAA;QACxB,IAAI,CAAC,KAAK,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,CAAA;IACnE,CAAC;IAEO,QAAQ,GAAG,CAAC,QAAoC,EAAE,EAAE;QAC1D,IAAI,CAAC,KAAK,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,GAAG,QAAQ,EAAE,CAAA;QAC3C,IAAI,CAAC,eAAe,EAAE,CAAA;IACxB,CAAC,CAAA;IAEO,eAAe,GAAG,GAAG,EAAE;QAC7B,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAA;IAClD,CAAC,CAAA;IAED,SAAS,GAAG,CAAC,QAAoB,EAAgB,EAAE;QACjD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QAC5B,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;IAC9C,CAAC,CAAA;IAED,WAAW,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAA;IAE9B,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAA;IAC7B,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,CAAA;IACvC,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAA;IACzB,CAAC;IAED,eAAe,GAAG,CAChB,MAAoG,EACpG,OAA0B,QAAQ,EAClC,EAAE;QACF,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;YACjC,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAA;YACpE,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAA;YAC/C,IAAI,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,UAAU,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAA;QAC3E,CAAC;aAAM,CAAC;YACN,MAAM,UAAU,GAAG,MAAM,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;YACzD,IAAI,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,UAA4C,EAAE,CAAC,CAAA;QACzE,CAAC;IACH,CAAC,CAAA;IAEO,WAAW,GAAG,CAAC,IAAuB,EAAE,aAAwB,EAAE,EAAE;QAC1E,OAAO,CAAC,MAAsE,EAAE,EAAE;YAChF,MAAM,IAAI,GAAG;gBACX,GAAG,aAAa;gBAChB,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;aAC9C,CAAA;YACD,MAAM,IAAI,GAAG,OAAO,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAA;YACjE,IAAI,CAAC,eAAe,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,IAAI,EAAE,EAAE,IAAI,CAAC,CAAA;QAClD,CAAC,CAAA;IACH,CAAC,CAAA;IAEO,cAAc,GAAG,CAGvB,QAAW,EACX,MAAoG,EACpG,KAAkG,EACK,EAAE;QACzG,OAAO,QAAQ,KAAK,MAAM,IAAI,MAAM,CAAC,UAAU,CAAC,GAAG,QAAQ,GAAG,CAAC,CAAA;IACjE,CAAC,CAAA;IAED,KAAK,GAAG,KAAK,EACX,MAEiH,EACjH,OAcC,EACc,EAAE;QACjB,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS;YAAE,OAAM;QAEhC,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;YACpD,IAAI,CAAC,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAA;YAErC,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;YAE9C,IAAI,WAAoC,CAAA;YACxC,IAAI,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,GAAG,EAAE,CAAC;gBACxC,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;oBACtB,WAAW,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,CAAA;oBACnC,MAAK;gBACP,CAAC;gBAED,MAAM,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;gBAE7E,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,OAAO,CAAC,aAAa,CAAC,CAAA;gBAEhE,IAAI,KAAK,KAAK,UAAU;oBAAE,OAAO,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC,IAAI,CAAC,CAAA;gBACnE,IAAI,KAAK,KAAK,QAAQ;oBAAE,OAAO,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,CAAA;gBAElE,IAAI,IAAI,CAAC,cAAc,CAAC,SAAS,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC;oBAChD,OAAO,CAAC,SAAS,CAAC,aAAa,EAAE,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAA;gBAChE,CAAC;gBAED,IAAI,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC;oBAC/C,OAAO,CAAC,SAAS,CAAC,aAAa,EAAE,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAA;gBAChE,CAAC;gBAED,IAAI,IAAI,CAAC,cAAc,CAAC,aAAa,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC;oBACpD,OAAO,CAAC,SAAS,CAAC,iBAAiB,EAAE,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,CAAC,CAAA;gBAC5D,CAAC;gBAED,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC;oBAC9C,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,CAAC,CAAA;gBACtD,CAAC;gBAED,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC;oBAC9C,OAAO,CAAC,SAAS,CAAC,YAAY,EAAE,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,CAAC,CAAA;gBACvD,CAAC;gBAED,IAAI,KAAK,KAAK,QAAQ,EAAE,CAAC;oBACvB,IAAI,eAAe,IAAI,IAAI,EAAE,CAAC;wBAC5B,IAAI,CAAC,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC,CAAA;oBACxD,CAAC;yBAAM,CAAC;wBACN,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAA;oBAC5B,CAAC;gBACH,CAAC;gBAED,IAAI,IAAI,CAAC,cAAc,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC;oBACjD,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,GAAG,IAAI,CAAA;oBAEnC,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAA;oBACzD,IAAI,CAAC,SAAS,EAAE,CAAC;wBACf,OAAO,CAAC,IAAI,CAAC,uDAAuD,CAAC,CAAA;wBACrE,SAAQ;oBACV,CAAC;oBAED,IAAI,CAAC,eAAe,CAAC,CAAC,YAAY,EAAE,EAAE;wBACpC,MAAM,MAAM,GAAG,EAAE,GAAG,OAAO,CAAC,aAAa,EAAE,GAAG,YAAY,EAAE,CAAA;wBAE5D,8CAA8C;wBAC9C,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,CAAA;wBACpD,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,CAAA;wBAE5E,IAAI,CAAC,KAAK,IAAI,KAAK,IAAI,IAAI;4BAAE,OAAO,MAAM,CAAA;wBAC1C,IAAI,KAAK,CAAC,OAAO,EAAE,KAAK,QAAQ,EAAE,CAAC;4BACjC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAA;wBAC3B,CAAC;6BAAM,CAAC;4BACN,QAAQ,CAAC,KAAK,CAAC,GAAG,aAAa,CAAC,KAAK,CAAC,CAAA;wBACxC,CAAC;wBAED,OAAO,OAAO,CAAC,WAAW,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;oBAC9C,CAAC,CAAC,CAAA;gBACJ,CAAC;YACH,CAAC;YAED,IAAI,WAAW,IAAI,IAAI;gBAAE,MAAM,WAAW,CAAA;YAE1C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,SAAS,EAAE,EAAE,CAAA;YAC1C,IAAI,OAAO,MAAM,KAAK,WAAW;gBAAE,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAA;QACjE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IACE,CAAC,CACC,KAAK,YAAY,KAAK,IAAI,kDAAkD;gBAC5E,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,CAAC,CAC/D,EACD,CAAC;gBACD,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBACpB,IAAI,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;gBACxB,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,CAAA;YAChC,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,QAAQ,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAA;YACnC,IAAI,CAAC,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAA;YACrC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAA;QACtB,CAAC;IACH,CAAC,CAAA;IAED,IAAI,GAAG,KAAK,EACV,aAAwB,EACxB,OAIC,EACc,EAAE;QACjB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAA;QACrB,IAAI,CAAC,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAA;QAErC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,aAAa,CAAC,EAAE,CAAC,CAAA;IACvE,CAAC,CAAA;IAED,KAAK,GAAG,GAAG,EAAE;QACX,6BAA6B;QAC7B,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAA;QACrB,IAAI,CAAC,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAA;QAErC,+BAA+B;QAC/B,IAAI,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAA;QAEjD,6BAA6B;QAC7B,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAA;IACvB,CAAC,CAAA;CACF"}

package/dist/ui/messages.d.ts

@@ -0,0 +1,18 @@
+import { type BaseMessage, type BaseMessageChunk } from '@langchain/core/messages';
+import type { Message } from '../types.messages.js';
+export declare class MessageTupleManager {
+    chunks: Record<string, {
+        chunk?: BaseMessageChunk | BaseMessage;
+        metadata?: Record<string, unknown>;
+        index?: number;
+    }>;
+    constructor();
+    add(serialized: Message, metadata: Record<string, unknown> | undefined): string | null;
+    clear(): void;
+    get(id: string | null | undefined, defaultIndex?: number): {
+        chunk?: BaseMessageChunk | BaseMessage;
+        metadata?: Record<string, unknown>;
+        index?: number;
+    } | null;
+}
+export declare const toMessageDict: (chunk: BaseMessage) => Message;

package/dist/ui/messages.js

@@ -0,0 +1,64 @@
+import { RemoveMessage, convertToChunk, coerceMessageLikeToMessage, isBaseMessageChunk } from '@langchain/core/messages';
+function tryConvertToChunk(message) {
+    try {
+        return convertToChunk(message);
+    }
+    catch {
+        return null;
+    }
+}
+function tryCoerceMessageLikeToMessage(message) {
+    // TODO: this is unnecessary with https://github.com/langchain-ai/langchainjs/pull/8941
+    if (message.type === 'remove' && message.id != null) {
+        return new RemoveMessage({ ...message, id: message.id });
+    }
+    return coerceMessageLikeToMessage(message);
+}
+export class MessageTupleManager {
+    chunks = {};
+    constructor() {
+        this.chunks = {};
+    }
+    add(serialized, metadata) {
+        // TODO: this is sometimes sent from the API
+        // figure out how to prevent this or move this to LC.js
+        if (serialized.type.endsWith('MessageChunk')) {
+            // eslint-disable-next-line no-param-reassign
+            serialized.type = serialized.type.slice(0, -'MessageChunk'.length).toLowerCase();
+        }
+        const message = tryCoerceMessageLikeToMessage(serialized);
+        const chunk = tryConvertToChunk(message);
+        const { id } = chunk ?? message;
+        if (!id) {
+            console.warn('No message ID found for chunk, ignoring in state', serialized);
+            return null;
+        }
+        this.chunks[id] ??= {};
+        this.chunks[id].metadata = metadata ?? this.chunks[id].metadata;
+        if (chunk) {
+            const prev = this.chunks[id].chunk;
+            this.chunks[id].chunk = (isBaseMessageChunk(prev) ? prev : null)?.concat(chunk) ?? chunk;
+        }
+        else {
+            this.chunks[id].chunk = message;
+        }
+        return id;
+    }
+    clear() {
+        this.chunks = {};
+    }
+    get(id, defaultIndex) {
+        if (id == null)
+            return null;
+        if (this.chunks[id] == null)
+            return null;
+        if (defaultIndex != null)
+            this.chunks[id].index ??= defaultIndex;
+        return this.chunks[id];
+    }
+}
+export const toMessageDict = (chunk) => {
+    const { type, data } = chunk.toDict();
+    return { ...data, type };
+};
+//# sourceMappingURL=messages.js.map

package/dist/ui/messages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"messages.js","sourceRoot":"","sources":["../../src/ui/messages.ts"],"names":[],"mappings":"OAAO,EAGL,aAAa,EACb,cAAc,EACd,0BAA0B,EAC1B,kBAAkB,EACnB,MAAM,0BAA0B;AAIjC,SAAS,iBAAiB,CAAC,OAAoB;IAC7C,IAAI,CAAC;QACH,OAAO,cAAc,CAAC,OAAO,CAAC,CAAA;IAChC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAA;IACb,CAAC;AACH,CAAC;AAED,SAAS,6BAA6B,CAAC,OAAgB;IACrD,uFAAuF;IACvF,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,IAAI,OAAO,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;QACpD,OAAO,IAAI,aAAa,CAAC,EAAE,GAAG,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,CAAA;IAC1D,CAAC;IAED,OAAO,0BAA0B,CAAC,OAAO,CAAC,CAAA;AAC5C,CAAC;AAED,MAAM,OAAO,mBAAmB;IAC9B,MAAM,GAOF,EAAE,CAAA;IAEN;QACE,IAAI,CAAC,MAAM,GAAG,EAAE,CAAA;IAClB,CAAC;IAED,GAAG,CAAC,UAAmB,EAAE,QAA6C;QACpE,4CAA4C;QAC5C,uDAAuD;QACvD,IAAI,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;YAC7C,6CAA6C;YAC7C,UAAU,CAAC,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,WAAW,EAAqB,CAAA;QACrG,CAAC;QAED,MAAM,OAAO,GAAG,6BAA6B,CAAC,UAAU,CAAC,CAAA;QACzD,MAAM,KAAK,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAA;QAExC,MAAM,EAAE,EAAE,EAAE,GAAG,KAAK,IAAI,OAAO,CAAA;QAC/B,IAAI,CAAC,EAAE,EAAE,CAAC;YACR,OAAO,CAAC,IAAI,CAAC,kDAAkD,EAAE,UAAU,CAAC,CAAA;YAC5E,OAAO,IAAI,CAAA;QACb,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,CAAA;QACtB,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,GAAG,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAA;QAC/D,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,KAAK,CAAA;YAClC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,KAAK,GAAG,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,KAAK,CAAA;QAC1F,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,KAAK,GAAG,OAAO,CAAA;QACjC,CAAC;QAED,OAAO,EAAE,CAAA;IACX,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,GAAG,EAAE,CAAA;IAClB,CAAC;IAED,GAAG,CAAC,EAA6B,EAAE,YAAqB;QACtD,IAAI,EAAE,IAAI,IAAI;YAAE,OAAO,IAAI,CAAA;QAC3B,IAAI,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,IAAI;YAAE,OAAO,IAAI,CAAA;QACxC,IAAI,YAAY,IAAI,IAAI;YAAE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,KAAK,KAAK,YAAY,CAAA;QAChE,OAAO,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;IACxB,CAAC;CACF;AAED,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,KAAkB,EAAW,EAAE;IAC3D,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC,MAAM,EAAE,CAAA;IACrC,OAAO,EAAE,GAAG,IAAI,EAAE,IAAI,EAAa,CAAA;AACrC,CAAC,CAAA"}

package/dist/ui/messages.mjs

@@ -0,0 +1,64 @@
+import { RemoveMessage, convertToChunk, coerceMessageLikeToMessage, isBaseMessageChunk } from '@langchain/core/messages';
+function tryConvertToChunk(message) {
+    try {
+        return convertToChunk(message);
+    }
+    catch {
+        return null;
+    }
+}
+function tryCoerceMessageLikeToMessage(message) {
+    // TODO: this is unnecessary with https://github.com/langchain-ai/langchainjs/pull/8941
+    if (message.type === 'remove' && message.id != null) {
+        return new RemoveMessage({ ...message, id: message.id });
+    }
+    return coerceMessageLikeToMessage(message);
+}
+export class MessageTupleManager {
+    chunks = {};
+    constructor() {
+        this.chunks = {};
+    }
+    add(serialized, metadata) {
+        // TODO: this is sometimes sent from the API
+        // figure out how to prevent this or move this to LC.js
+        if (serialized.type.endsWith('MessageChunk')) {
+            // eslint-disable-next-line no-param-reassign
+            serialized.type = serialized.type.slice(0, -'MessageChunk'.length).toLowerCase();
+        }
+        const message = tryCoerceMessageLikeToMessage(serialized);
+        const chunk = tryConvertToChunk(message);
+        const { id } = chunk ?? message;
+        if (!id) {
+            console.warn('No message ID found for chunk, ignoring in state', serialized);
+            return null;
+        }
+        this.chunks[id] ??= {};
+        this.chunks[id].metadata = metadata ?? this.chunks[id].metadata;
+        if (chunk) {
+            const prev = this.chunks[id].chunk;
+            this.chunks[id].chunk = (isBaseMessageChunk(prev) ? prev : null)?.concat(chunk) ?? chunk;
+        }
+        else {
+            this.chunks[id].chunk = message;
+        }
+        return id;
+    }
+    clear() {
+        this.chunks = {};
+    }
+    get(id, defaultIndex) {
+        if (id == null)
+            return null;
+        if (this.chunks[id] == null)
+            return null;
+        if (defaultIndex != null)
+            this.chunks[id].index ??= defaultIndex;
+        return this.chunks[id];
+    }
+}
+export const toMessageDict = (chunk) => {
+    const { type, data } = chunk.toDict();
+    return { ...data, type };
+};
+//# sourceMappingURL=messages.mjs.map

package/dist/ui/messages.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"messages.mjs","sourceRoot":"","sources":["../../src/ui/messages.ts"],"names":[],"mappings":"OAAO,EAGL,aAAa,EACb,cAAc,EACd,0BAA0B,EAC1B,kBAAkB,EACnB,MAAM,0BAA0B;AAIjC,SAAS,iBAAiB,CAAC,OAAoB;IAC7C,IAAI,CAAC;QACH,OAAO,cAAc,CAAC,OAAO,CAAC,CAAA;IAChC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAA;IACb,CAAC;AACH,CAAC;AAED,SAAS,6BAA6B,CAAC,OAAgB;IACrD,uFAAuF;IACvF,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,IAAI,OAAO,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;QACpD,OAAO,IAAI,aAAa,CAAC,EAAE,GAAG,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,CAAA;IAC1D,CAAC;IAED,OAAO,0BAA0B,CAAC,OAAO,CAAC,CAAA;AAC5C,CAAC;AAED,MAAM,OAAO,mBAAmB;IAC9B,MAAM,GAOF,EAAE,CAAA;IAEN;QACE,IAAI,CAAC,MAAM,GAAG,EAAE,CAAA;IAClB,CAAC;IAED,GAAG,CAAC,UAAmB,EAAE,QAA6C;QACpE,4CAA4C;QAC5C,uDAAuD;QACvD,IAAI,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;YAC7C,6CAA6C;YAC7C,UAAU,CAAC,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,WAAW,EAAqB,CAAA;QACrG,CAAC;QAED,MAAM,OAAO,GAAG,6BAA6B,CAAC,UAAU,CAAC,CAAA;QACzD,MAAM,KAAK,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAA;QAExC,MAAM,EAAE,EAAE,EAAE,GAAG,KAAK,IAAI,OAAO,CAAA;QAC/B,IAAI,CAAC,EAAE,EAAE,CAAC;YACR,OAAO,CAAC,IAAI,CAAC,kDAAkD,EAAE,UAAU,CAAC,CAAA;YAC5E,OAAO,IAAI,CAAA;QACb,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,CAAA;QACtB,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,GAAG,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAA;QAC/D,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,KAAK,CAAA;YAClC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,KAAK,GAAG,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,KAAK,CAAA;QAC1F,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,KAAK,GAAG,OAAO,CAAA;QACjC,CAAC;QAED,OAAO,EAAE,CAAA;IACX,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,GAAG,EAAE,CAAA;IAClB,CAAC;IAED,GAAG,CAAC,EAA6B,EAAE,YAAqB;QACtD,IAAI,EAAE,IAAI,IAAI;YAAE,OAAO,IAAI,CAAA;QAC3B,IAAI,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,IAAI;YAAE,OAAO,IAAI,CAAA;QACxC,IAAI,YAAY,IAAI,IAAI;YAAE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,KAAK,KAAK,YAAY,CAAA;QAChE,OAAO,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;IACxB,CAAC;CACF;AAED,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,KAAkB,EAAW,EAAE;IAC3D,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC,MAAM,EAAE,CAAA;IACrC,OAAO,EAAE,GAAG,IAAI,EAAE,IAAI,EAAa,CAAA;AACrC,CAAC,CAAA"}

package/dist/ui/utils.d.ts

@@ -0,0 +1,2 @@
+export declare function unique<T>(array: T[]): T[];
+export declare function findLast<T>(array: T[], predicate: (item: T) => boolean): T | undefined;

package/dist/ui/utils.js

@@ -0,0 +1,11 @@
+export function unique(array) {
+    return [...new Set(array)];
+}
+export function findLast(array, predicate) {
+    for (let i = array.length - 1; i >= 0; i -= 1) {
+        if (predicate(array[i]))
+            return array[i];
+    }
+    return undefined;
+}
+//# sourceMappingURL=utils.js.map

package/dist/ui/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/ui/utils.ts"],"names":[],"mappings":"AAAA,MAAM,UAAU,MAAM,CAAI,KAAU;IAClC,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,CAAQ,CAAA;AACnC,CAAC;AAED,MAAM,UAAU,QAAQ,CAAI,KAAU,EAAE,SAA+B;IACrE,KAAK,IAAI,CAAC,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;QAC9C,IAAI,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAAE,OAAO,KAAK,CAAC,CAAC,CAAC,CAAA;IAC1C,CAAC;IACD,OAAO,SAAS,CAAA;AAClB,CAAC"}

package/dist/ui/utils.mjs

@@ -0,0 +1,11 @@
+export function unique(array) {
+    return [...new Set(array)];
+}
+export function findLast(array, predicate) {
+    for (let i = array.length - 1; i >= 0; i -= 1) {
+        if (predicate(array[i]))
+            return array[i];
+    }
+    return undefined;
+}
+//# sourceMappingURL=utils.mjs.map

package/dist/ui/utils.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.mjs","sourceRoot":"","sources":["../../src/ui/utils.ts"],"names":[],"mappings":"AAAA,MAAM,UAAU,MAAM,CAAI,KAAU;IAClC,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,CAAQ,CAAA;AACnC,CAAC;AAED,MAAM,UAAU,QAAQ,CAAI,KAAU,EAAE,SAA+B;IACrE,KAAK,IAAI,CAAC,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;QAC9C,IAAI,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAAE,OAAO,KAAK,CAAC,CAAC,CAAC,CAAA;IAC1C,CAAC;IACD,OAAO,SAAS,CAAA;AAClB,CAAC"}

package/dist/utils/async_caller.d.ts

@@ -0,0 +1,48 @@
+type ResponseCallback = (response?: Response) => Promise<boolean>;
+export interface AsyncCallerParams {
+    /**
+     * The maximum number of concurrent calls that can be made.
+     * Defaults to `Infinity`, which means no limit.
+     */
+    maxConcurrency?: number;
+    /**
+     * The maximum number of retries that can be made for a single call,
+     * with an exponential backoff between each attempt. Defaults to 6.
+     */
+    maxRetries?: number;
+    onFailedResponseHook?: ResponseCallback;
+    /**
+     * Specify a custom fetch implementation.
+     *
+     * By default we expect the `fetch` is available in the global scope.
+     */
+    fetch?: typeof fetch;
+}
+export interface AsyncCallerCallOptions {
+    signal?: AbortSignal;
+}
+/**
+ * A class that can be used to make async calls with concurrency and retry logic.
+ *
+ * This is useful for making calls to any kind of "expensive" external resource,
+ * be it because it's rate-limited, subject to network issues, etc.
+ *
+ * Concurrent calls are limited by the `maxConcurrency` parameter, which defaults
+ * to `Infinity`. This means that by default, all calls will be made in parallel.
+ *
+ * Retries are limited by the `maxRetries` parameter, which defaults to 5. This
+ * means that by default, each call will be retried up to 5 times, with an
+ * exponential backoff between each attempt.
+ */
+export declare class AsyncCaller {
+    protected maxConcurrency: AsyncCallerParams['maxConcurrency'];
+    protected maxRetries: AsyncCallerParams['maxRetries'];
+    private queue;
+    private onFailedResponseHook?;
+    private customFetch?;
+    constructor(params: AsyncCallerParams);
+    call<TArgs extends unknown[], TResult>(callable: (...args: TArgs) => Promise<TResult>, ...args: TArgs): Promise<TResult>;
+    callWithOptions<TArgs extends unknown[], TResult>(options: AsyncCallerCallOptions, callable: (...args: TArgs) => Promise<TResult>, ...args: TArgs): Promise<TResult>;
+    fetch(...args: Parameters<typeof fetch>): ReturnType<typeof fetch>;
+}
+export {};

package/dist/utils/async_caller.js

@@ -0,0 +1,153 @@
+import pRetry from 'p-retry';
+import * as PQueueModule from 'p-queue';
+import { _getFetchImplementation } from "../singletons/fetch.js";
+const STATUS_NO_RETRY = [
+    400, // Bad Request
+    401, // Unauthorized
+    402, // Payment required
+    403, // Forbidden
+    404, // Not Found
+    405, // Method Not Allowed
+    406, // Not Acceptable
+    407, // Proxy Authentication Required
+    408, // Request Timeout
+    409, // Conflict
+    422 // Unprocessable Entity
+];
+function isQueueConstructor(value) {
+    return typeof value === 'function';
+}
+function resolveQueueConstructor() {
+    const moduleValue = PQueueModule;
+    if (isQueueConstructor(moduleValue))
+        return moduleValue;
+    if (moduleValue && typeof moduleValue === 'object' && 'default' in moduleValue) {
+        const defaultExport = moduleValue.default;
+        if (isQueueConstructor(defaultExport))
+            return defaultExport;
+    }
+    throw new Error('p-queue did not expose a queue constructor');
+}
+/**
+ * Do not rely on globalThis.Response, rather just
+ * do duck typing
+ */
+function isResponse(x) {
+    if (x == null || typeof x !== 'object')
+        return false;
+    return 'status' in x && 'statusText' in x && 'text' in x;
+}
+function hasErrorCode(value, code) {
+    return Boolean(value && typeof value === 'object' && 'code' in value && value.code === code);
+}
+/**
+ * Utility error to properly handle failed requests
+ */
+class HTTPError extends Error {
+    status;
+    text;
+    response;
+    constructor(status, message, response) {
+        super(`HTTP ${status}: ${message}`);
+        this.status = status;
+        this.text = message;
+        this.response = response;
+    }
+    static async fromResponse(response, options) {
+        try {
+            return new HTTPError(response.status, await response.text(), options?.includeResponse ? response : undefined);
+        }
+        catch {
+            return new HTTPError(response.status, response.statusText, options?.includeResponse ? response : undefined);
+        }
+    }
+}
+/**
+ * A class that can be used to make async calls with concurrency and retry logic.
+ *
+ * This is useful for making calls to any kind of "expensive" external resource,
+ * be it because it's rate-limited, subject to network issues, etc.
+ *
+ * Concurrent calls are limited by the `maxConcurrency` parameter, which defaults
+ * to `Infinity`. This means that by default, all calls will be made in parallel.
+ *
+ * Retries are limited by the `maxRetries` parameter, which defaults to 5. This
+ * means that by default, each call will be retried up to 5 times, with an
+ * exponential backoff between each attempt.
+ */
+export class AsyncCaller {
+    maxConcurrency;
+    maxRetries;
+    queue;
+    onFailedResponseHook;
+    customFetch;
+    constructor(params) {
+        this.maxConcurrency = params.maxConcurrency ?? Infinity;
+        this.maxRetries = params.maxRetries ?? 4;
+        const Queue = resolveQueueConstructor();
+        this.queue = new Queue({ concurrency: this.maxConcurrency });
+        this.onFailedResponseHook = params?.onFailedResponseHook;
+        this.customFetch = params.fetch;
+    }
+    call(callable, ...args) {
+        const { onFailedResponseHook } = this;
+        return this.queue.add(() => pRetry(() => callable(...args).catch(async (error) => {
+            // eslint-disable-next-line no-instanceof/no-instanceof
+            if (error instanceof Error) {
+                throw error;
+            }
+            else if (isResponse(error)) {
+                throw await HTTPError.fromResponse(error, {
+                    includeResponse: !!onFailedResponseHook
+                });
+            }
+            else {
+                throw new Error(typeof error === 'string' ? error : 'Unknown async caller failure');
+            }
+        }), {
+            async onFailedAttempt(error) {
+                if (error.message.startsWith('Cancel') ||
+                    error.message.startsWith('TimeoutError') ||
+                    error.message.startsWith('AbortError')) {
+                    throw error;
+                }
+                if (hasErrorCode(error, 'ECONNABORTED')) {
+                    throw error;
+                }
+                // eslint-disable-next-line no-instanceof/no-instanceof
+                if (error instanceof HTTPError) {
+                    if (STATUS_NO_RETRY.includes(error.status)) {
+                        throw error;
+                    }
+                    if (onFailedResponseHook && error.response) {
+                        await onFailedResponseHook(error.response);
+                    }
+                }
+            },
+            // If needed we can change some of the defaults here,
+            // but they're quite sensible.
+            retries: this.maxRetries,
+            randomize: true
+        }), { throwOnTimeout: true });
+    }
+    callWithOptions(options, callable, ...args) {
+        // Note this doesn't cancel the underlying request,
+        // when available prefer to use the signal option of the underlying call
+        if (options.signal) {
+            return Promise.race([
+                this.call(callable, ...args),
+                new Promise((_, reject) => {
+                    options.signal?.addEventListener('abort', () => {
+                        reject(new Error('AbortError'));
+                    });
+                })
+            ]);
+        }
+        return this.call(callable, ...args);
+    }
+    fetch(...args) {
+        const fetchFn = this.customFetch ?? _getFetchImplementation();
+        return this.call(() => fetchFn(...args).then((res) => (res.ok ? res : Promise.reject(res))));
+    }
+}
+//# sourceMappingURL=async_caller.js.map

package/dist/utils/async_caller.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"async_caller.js","sourceRoot":"","sources":["../../src/utils/async_caller.ts"],"names":[],"mappings":"OAAO,MAAM,MAAM,SAAS;OACrB,KAAK,YAAY,MAAM,SAAS;OAChC,EAAE,uBAAuB,EAAE;AAElC,MAAM,eAAe,GAAG;IACtB,GAAG,EAAE,cAAc;IACnB,GAAG,EAAE,eAAe;IACpB,GAAG,EAAE,mBAAmB;IACxB,GAAG,EAAE,YAAY;IACjB,GAAG,EAAE,YAAY;IACjB,GAAG,EAAE,qBAAqB;IAC1B,GAAG,EAAE,iBAAiB;IACtB,GAAG,EAAE,gCAAgC;IACrC,GAAG,EAAE,kBAAkB;IACvB,GAAG,EAAE,WAAW;IAChB,GAAG,CAAC,uBAAuB;CAC5B,CAAA;AAYD,SAAS,kBAAkB,CAAC,KAAc;IACxC,OAAO,OAAO,KAAK,KAAK,UAAU,CAAA;AACpC,CAAC;AAED,SAAS,uBAAuB;IAC9B,MAAM,WAAW,GAAY,YAAY,CAAA;IACzC,IAAI,kBAAkB,CAAC,WAAW,CAAC;QAAE,OAAO,WAAW,CAAA;IACvD,IAAI,WAAW,IAAI,OAAO,WAAW,KAAK,QAAQ,IAAI,SAAS,IAAI,WAAW,EAAE,CAAC;QAC/E,MAAM,aAAa,GAAY,WAAW,CAAC,OAAO,CAAA;QAClD,IAAI,kBAAkB,CAAC,aAAa,CAAC;YAAE,OAAO,aAAa,CAAA;IAC7D,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAA;AAC/D,CAAC;AA4BD;;;GAGG;AACH,SAAS,UAAU,CAAC,CAAU;IAC5B,IAAI,CAAC,IAAI,IAAI,IAAI,OAAO,CAAC,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAA;IACpD,OAAO,QAAQ,IAAI,CAAC,IAAI,YAAY,IAAI,CAAC,IAAI,MAAM,IAAI,CAAC,CAAA;AAC1D,CAAC;AAED,SAAS,YAAY,CAAC,KAAc,EAAE,IAAY;IAChD,OAAO,OAAO,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,CAAA;AAC9F,CAAC;AAED;;GAEG;AACH,MAAM,SAAU,SAAQ,KAAK;IAC3B,MAAM,CAAQ;IAEd,IAAI,CAAQ;IAEZ,QAAQ,CAAW;IAEnB,YAAY,MAAc,EAAE,OAAe,EAAE,QAAmB;QAC9D,KAAK,CAAC,QAAQ,MAAM,KAAK,OAAO,EAAE,CAAC,CAAA;QACnC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;QACpB,IAAI,CAAC,IAAI,GAAG,OAAO,CAAA;QACnB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAA;IAC1B,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,QAAkB,EAAE,OAAuC;QACnF,IAAI,CAAC;YACH,OAAO,IAAI,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,QAAQ,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,eAAe,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAA;QAC/G,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,UAAU,EAAE,OAAO,EAAE,eAAe,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAA;QAC7G,CAAC;IACH,CAAC;CACF;AAED;;;;;;;;;;;;GAYG;AACH,MAAM,OAAO,WAAW;IACZ,cAAc,CAAqC;IAEnD,UAAU,CAAiC;IAE7C,KAAK,CAAW;IAEhB,oBAAoB,CAAmB;IAEvC,WAAW,CAAe;IAElC,YAAY,MAAyB;QACnC,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,QAAQ,CAAA;QACvD,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,CAAC,CAAA;QAExC,MAAM,KAAK,GAAG,uBAAuB,EAAE,CAAA;QACvC,IAAI,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAA;QAC5D,IAAI,CAAC,oBAAoB,GAAG,MAAM,EAAE,oBAAoB,CAAA;QACxD,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,KAAK,CAAA;IACjC,CAAC;IAED,IAAI,CACF,QAA8C,EAC9C,GAAG,IAAW;QAEd,MAAM,EAAE,oBAAoB,EAAE,GAAG,IAAI,CAAA;QACrC,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CACnB,GAAG,EAAE,CACH,MAAM,CACJ,GAAG,EAAE,CACH,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,KAAc,EAAE,EAAE;YAC/C,uDAAuD;YACvD,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,MAAM,KAAK,CAAA;YACb,CAAC;iBAAM,IAAI,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC7B,MAAM,MAAM,SAAS,CAAC,YAAY,CAAC,KAAK,EAAE;oBACxC,eAAe,EAAE,CAAC,CAAC,oBAAoB;iBACxC,CAAC,CAAA;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,8BAA8B,CAAC,CAAA;YACrF,CAAC;QACH,CAAC,CAAC,EACJ;YACE,KAAK,CAAC,eAAe,CAAC,KAAK;gBACzB,IACE,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;oBAClC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC;oBACxC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,YAAY,CAAC,EACtC,CAAC;oBACD,MAAM,KAAK,CAAA;gBACb,CAAC;gBACD,IAAI,YAAY,CAAC,KAAK,EAAE,cAAc,CAAC,EAAE,CAAC;oBACxC,MAAM,KAAK,CAAA;gBACb,CAAC;gBAED,uDAAuD;gBACvD,IAAI,KAAK,YAAY,SAAS,EAAE,CAAC;oBAC/B,IAAI,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC3C,MAAM,KAAK,CAAA;oBACb,CAAC;oBACD,IAAI,oBAAoB,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;wBAC3C,MAAM,oBAAoB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;oBAC5C,CAAC;gBACH,CAAC;YACH,CAAC;YACD,qDAAqD;YACrD,8BAA8B;YAC9B,OAAO,EAAE,IAAI,CAAC,UAAU;YACxB,SAAS,EAAE,IAAI;SAChB,CACF,EACH,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,CAAA;IACH,CAAC;IAED,eAAe,CACb,OAA+B,EAC/B,QAA8C,EAC9C,GAAG,IAAW;QAEd,mDAAmD;QACnD,wEAAwE;QACxE,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,OAAO,OAAO,CAAC,IAAI,CAAC;gBAClB,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC;gBAC5B,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;oBAC/B,OAAO,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE;wBAC7C,MAAM,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAA;oBACjC,CAAC,CAAC,CAAA;gBACJ,CAAC,CAAC;aACH,CAAC,CAAA;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,CAAA;IACrC,CAAC;IAED,KAAK,CAAC,GAAG,IAA8B;QACrC,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,IAAK,uBAAuB,EAAmB,CAAA;QAC/E,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;IAC9F,CAAC;CACF"}