@lichen-ai/lichen-sdk 0.1.0

package/dist/ui/branching.test.mjs

@@ -0,0 +1,369 @@
+import { it, expect } from 'vitest';
+import { getBranchSequence } from "./branching.mjs";
+const history = [
+    {
+        values: {
+            messages: [
+                {
+                    content: 'Fork: Hello',
+                    additional_kwargs: {},
+                    response_metadata: {},
+                    id: '33357aea-9c2d-4718-92f4-20038d5a2d29',
+                    type: 'human'
+                },
+                {
+                    content: 'Hey',
+                    additional_kwargs: {},
+                    response_metadata: {},
+                    tool_call_chunks: [],
+                    id: 'run-782a2f90-2e39-4c21-9ee6-0f38585a0ae6',
+                    tool_calls: [],
+                    invalid_tool_calls: [],
+                    type: 'ai'
+                }
+            ]
+        },
+        next: [],
+        tasks: [],
+        metadata: {
+            source: 'loop',
+            step: 2,
+            parents: {},
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc'
+        },
+        created_at: '2025-08-21T12:47:12.640Z',
+        checkpoint: {
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc',
+            checkpoint_id: '1f07e8cf-56a7-6000-8002-5e272b1d5285',
+            checkpoint_ns: '',
+            checkpoint_map: null
+        },
+        parent_checkpoint: {
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc',
+            checkpoint_id: '1f07e8cf-53aa-6d70-8001-2096b0bbf835',
+            checkpoint_ns: '',
+            checkpoint_map: null
+        }
+    },
+    {
+        values: {
+            messages: [
+                {
+                    content: 'Fork: Hello',
+                    additional_kwargs: {},
+                    response_metadata: {},
+                    id: '33357aea-9c2d-4718-92f4-20038d5a2d29',
+                    type: 'human'
+                }
+            ]
+        },
+        next: ['agent'],
+        tasks: [
+            {
+                id: '680ee831-4d10-50f9-956f-e0d937321614',
+                name: 'agent',
+                error: null,
+                interrupts: [],
+                path: ['__pregel_pull', 'agent'],
+                checkpoint: null,
+                state: null,
+                result: {
+                    messages: [
+                        {
+                            content: 'Hey',
+                            additional_kwargs: {},
+                            response_metadata: {},
+                            tool_call_chunks: [],
+                            id: 'run-782a2f90-2e39-4c21-9ee6-0f38585a0ae6',
+                            tool_calls: [],
+                            invalid_tool_calls: [],
+                            type: 'ai'
+                        }
+                    ]
+                }
+            }
+        ],
+        metadata: {
+            source: 'loop',
+            step: 1,
+            parents: {},
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc'
+        },
+        created_at: '2025-08-21T12:47:12.327Z',
+        checkpoint: {
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc',
+            checkpoint_id: '1f07e8cf-53aa-6d70-8001-2096b0bbf835',
+            checkpoint_ns: '',
+            checkpoint_map: null
+        },
+        parent_checkpoint: {
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc',
+            checkpoint_id: '1f07e8cf-53a8-6660-8000-f157c9cf7d66',
+            checkpoint_ns: '',
+            checkpoint_map: null
+        }
+    },
+    {
+        values: { messages: [] },
+        next: ['__start__'],
+        tasks: [
+            {
+                id: '4233f4c0-5ea2-5117-9153-95e5fe59eee3',
+                name: '__start__',
+                error: null,
+                interrupts: [],
+                path: ['__pregel_pull', '__start__'],
+                checkpoint: null,
+                state: null,
+                result: { messages: [{ type: 'human', content: 'Fork: Hello' }] }
+            }
+        ],
+        metadata: {
+            source: 'input',
+            step: 0,
+            parents: {},
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc'
+        },
+        created_at: '2025-08-21T12:47:12.326Z',
+        checkpoint: {
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc',
+            checkpoint_id: '1f07e8cf-53a8-6660-8000-f157c9cf7d66',
+            checkpoint_ns: '',
+            checkpoint_map: null
+        },
+        parent_checkpoint: {
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc',
+            checkpoint_id: '1f07e8cf-4cea-6530-ffff-0995a53299c5',
+            checkpoint_ns: '',
+            checkpoint_map: null
+        }
+    },
+    {
+        values: {
+            messages: [
+                {
+                    content: 'Hello',
+                    additional_kwargs: {},
+                    response_metadata: {},
+                    id: 'ec300e63-1494-4ef6-936c-60d02d7bdce1',
+                    type: 'human'
+                },
+                {
+                    content: 'Hey',
+                    additional_kwargs: {},
+                    response_metadata: {},
+                    tool_call_chunks: [],
+                    id: 'run-593064bd-dd07-4ed3-99c0-a918625f4884',
+                    tool_calls: [],
+                    invalid_tool_calls: [],
+                    type: 'ai'
+                }
+            ]
+        },
+        next: [],
+        tasks: [],
+        metadata: {
+            source: 'loop',
+            step: 1,
+            parents: {},
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc'
+        },
+        created_at: '2025-08-21T12:47:12.285Z',
+        checkpoint: {
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc',
+            checkpoint_id: '1f07e8cf-5344-64d0-8001-8a176e91d12f',
+            checkpoint_ns: '',
+            checkpoint_map: null
+        },
+        parent_checkpoint: {
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc',
+            checkpoint_id: '1f07e8cf-4cf8-6f90-8000-7707fd155a8a',
+            checkpoint_ns: '',
+            checkpoint_map: null
+        }
+    },
+    {
+        values: {
+            messages: [
+                {
+                    content: 'Hello',
+                    additional_kwargs: {},
+                    response_metadata: {},
+                    id: 'ec300e63-1494-4ef6-936c-60d02d7bdce1',
+                    type: 'human'
+                },
+                {
+                    content: 'Hey',
+                    additional_kwargs: {},
+                    response_metadata: {},
+                    tool_call_chunks: [],
+                    id: 'run-b86ef558-eaff-4838-bc03-218f20554a9f',
+                    tool_calls: [],
+                    invalid_tool_calls: [],
+                    type: 'ai'
+                }
+            ]
+        },
+        next: [],
+        tasks: [],
+        metadata: {
+            source: 'loop',
+            step: 1,
+            parents: {},
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc'
+        },
+        created_at: '2025-08-21T12:47:11.936Z',
+        checkpoint: {
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc',
+            checkpoint_id: '1f07e8cf-4ff0-6400-8001-a54aedf52a76',
+            checkpoint_ns: '',
+            checkpoint_map: null
+        },
+        parent_checkpoint: {
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc',
+            checkpoint_id: '1f07e8cf-4cf8-6f90-8000-7707fd155a8a',
+            checkpoint_ns: '',
+            checkpoint_map: null
+        }
+    },
+    {
+        values: {
+            messages: [
+                {
+                    content: 'Hello',
+                    additional_kwargs: {},
+                    response_metadata: {},
+                    id: 'ec300e63-1494-4ef6-936c-60d02d7bdce1',
+                    type: 'human'
+                }
+            ]
+        },
+        next: ['agent'],
+        tasks: [
+            {
+                id: 'e3f503f1-73bc-5db1-8b4d-13f660ad1b51',
+                name: 'agent',
+                error: null,
+                interrupts: [],
+                path: ['__pregel_pull', 'agent'],
+                checkpoint: null,
+                state: null,
+                result: {
+                    messages: [
+                        {
+                            content: 'Hey',
+                            additional_kwargs: {},
+                            response_metadata: {},
+                            tool_call_chunks: [],
+                            id: 'run-b86ef558-eaff-4838-bc03-218f20554a9f',
+                            tool_calls: [],
+                            invalid_tool_calls: [],
+                            type: 'ai'
+                        }
+                    ]
+                }
+            }
+        ],
+        metadata: {
+            source: 'loop',
+            step: 0,
+            parents: {},
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc'
+        },
+        created_at: '2025-08-21T12:47:11.625Z',
+        checkpoint: {
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc',
+            checkpoint_id: '1f07e8cf-4cf8-6f90-8000-7707fd155a8a',
+            checkpoint_ns: '',
+            checkpoint_map: null
+        },
+        parent_checkpoint: {
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc',
+            checkpoint_id: '1f07e8cf-4cea-6530-ffff-0995a53299c5',
+            checkpoint_ns: '',
+            checkpoint_map: null
+        }
+    },
+    {
+        values: { messages: [] },
+        next: ['__start__'],
+        tasks: [
+            {
+                id: 'cddede95-d276-59f1-bc08-8051d6448734',
+                name: '__start__',
+                error: null,
+                interrupts: [],
+                path: ['__pregel_pull', '__start__'],
+                checkpoint: null,
+                state: null,
+                result: { messages: [{ type: 'human', content: 'Hello' }] }
+            }
+        ],
+        metadata: {
+            source: 'input',
+            step: -1,
+            parents: {},
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc'
+        },
+        created_at: '2025-08-21T12:47:11.619Z',
+        checkpoint: {
+            thread_id: 'efa80f4d-725f-4903-b19f-216309f060bc',
+            checkpoint_id: '1f07e8cf-4cea-6530-ffff-0995a53299c5',
+            checkpoint_ns: '',
+            checkpoint_map: null
+        },
+        parent_checkpoint: null
+    }
+];
+const node = (value, paths = []) => ({
+    type: 'node',
+    value: typeof value === 'number' ? history.at(value) : value,
+    path: paths.map((v) => {
+        if (typeof v === 'number') {
+            return history.at(v)?.checkpoint?.checkpoint_id;
+        }
+        return v?.checkpoint?.checkpoint_id;
+    })
+});
+const fork = (...items) => ({ type: 'fork', items });
+const sequence = (...items) => ({ type: 'sequence', items });
+it('full tree', async () => {
+    const { rootSequence, paths } = getBranchSequence(history);
+    expect
+        .soft(paths)
+        .toMatchObject(expect.arrayContaining([[5], [5, 4], [5, 3], [2]].map((p) => p.map((i) => history.at(i)?.checkpoint?.checkpoint_id))));
+    expect
+        .soft(rootSequence)
+        .toMatchObject(sequence(node(6), fork(sequence(node(5, [5]), fork(sequence(node(4, [5, 4])), sequence(node(3, [5, 3])))), sequence(node(2, [2]), node(1, [2]), node(0, [2])))));
+});
+it('partial tree', async () => {
+    expect(getBranchSequence(history.slice(0, 1))).toMatchObject({
+        paths: [],
+        rootSequence: sequence(node(0))
+    });
+    expect(getBranchSequence(history.slice(0, 2))).toMatchObject({
+        paths: [],
+        rootSequence: sequence(node(1), node(0))
+    });
+    expect(getBranchSequence(history.slice(0, 3))).toMatchObject({
+        paths: [],
+        rootSequence: sequence(node(2), node(1), node(0))
+    });
+    expect(getBranchSequence(history.slice(0, 4))).toMatchObject({
+        paths: [],
+        rootSequence: sequence(node(2), node(1), node(0))
+    });
+    expect(getBranchSequence(history.slice(0, 5))).toMatchObject({
+        paths: [],
+        rootSequence: sequence(node(2), node(1), node(0))
+    });
+    expect(getBranchSequence(history.slice(0, 6))).toMatchObject({
+        paths: expect.arrayContaining([[5], [5, 4], [5, 3], [2]].map((p) => p.map((i) => history.at(i)?.checkpoint?.checkpoint_id))),
+        rootSequence: sequence(fork(sequence(node(5, [5]), fork(sequence(node(4, [5, 4])), sequence(node(3, [5, 3])))), sequence(node(2, [2]), node(1, [2]), node(0, [2]))))
+    });
+    expect(getBranchSequence(history.slice(0, 7))).toMatchObject({
+        paths: expect.arrayContaining([[5], [5, 4], [5, 3], [2]].map((p) => p.map((i) => history.at(i)?.checkpoint?.checkpoint_id))),
+        rootSequence: sequence(node(6), fork(sequence(node(5, [5]), fork(sequence(node(4, [5, 4])), sequence(node(3, [5, 3])))), sequence(node(2, [2]), node(1, [2]), node(0, [2]))))
+    });
+});
+//# sourceMappingURL=branching.test.mjs.map

package/dist/ui/branching.test.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"branching.test.mjs","sourceRoot":"","sources":["../../src/ui/branching.test.ts"],"names":[],"mappings":"OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ;OAC5B,EAAE,iBAAiB,EAAE;AAG5B,MAAM,OAAO,GAAG;IACd;QACE,MAAM,EAAE;YACN,QAAQ,EAAE;gBACR;oBACE,OAAO,EAAE,aAAa;oBACtB,iBAAiB,EAAE,EAAE;oBACrB,iBAAiB,EAAE,EAAE;oBACrB,EAAE,EAAE,sCAAsC;oBAC1C,IAAI,EAAE,OAAO;iBACd;gBACD;oBACE,OAAO,EAAE,KAAK;oBACd,iBAAiB,EAAE,EAAE;oBACrB,iBAAiB,EAAE,EAAE;oBACrB,gBAAgB,EAAE,EAAE;oBACpB,EAAE,EAAE,0CAA0C;oBAC9C,UAAU,EAAE,EAAE;oBACd,kBAAkB,EAAE,EAAE;oBACtB,IAAI,EAAE,IAAI;iBACX;aACF;SACF;QACD,IAAI,EAAE,EAAE;QACR,KAAK,EAAE,EAAE;QACT,QAAQ,EAAE;YACR,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,CAAC;YACP,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,sCAAsC;SAClD;QACD,UAAU,EAAE,0BAA0B;QACtC,UAAU,EAAE;YACV,SAAS,EAAE,sCAAsC;YACjD,aAAa,EAAE,sCAAsC;YACrD,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,IAAI;SACrB;QACD,iBAAiB,EAAE;YACjB,SAAS,EAAE,sCAAsC;YACjD,aAAa,EAAE,sCAAsC;YACrD,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,IAAI;SACrB;KACF;IACD;QACE,MAAM,EAAE;YACN,QAAQ,EAAE;gBACR;oBACE,OAAO,EAAE,aAAa;oBACtB,iBAAiB,EAAE,EAAE;oBACrB,iBAAiB,EAAE,EAAE;oBACrB,EAAE,EAAE,sCAAsC;oBAC1C,IAAI,EAAE,OAAO;iBACd;aACF;SACF;QACD,IAAI,EAAE,CAAC,OAAO,CAAC;QACf,KAAK,EAAE;YACL;gBACE,EAAE,EAAE,sCAAsC;gBAC1C,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,IAAI;gBACX,UAAU,EAAE,EAAE;gBACd,IAAI,EAAE,CAAC,eAAe,EAAE,OAAO,CAAC;gBAChC,UAAU,EAAE,IAAI;gBAChB,KAAK,EAAE,IAAI;gBACX,MAAM,EAAE;oBACN,QAAQ,EAAE;wBACR;4BACE,OAAO,EAAE,KAAK;4BACd,iBAAiB,EAAE,EAAE;4BACrB,iBAAiB,EAAE,EAAE;4BACrB,gBAAgB,EAAE,EAAE;4BACpB,EAAE,EAAE,0CAA0C;4BAC9C,UAAU,EAAE,EAAE;4BACd,kBAAkB,EAAE,EAAE;4BACtB,IAAI,EAAE,IAAI;yBACX;qBACF;iBACF;aACF;SACF;QACD,QAAQ,EAAE;YACR,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,CAAC;YACP,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,sCAAsC;SAClD;QACD,UAAU,EAAE,0BAA0B;QACtC,UAAU,EAAE;YACV,SAAS,EAAE,sCAAsC;YACjD,aAAa,EAAE,sCAAsC;YACrD,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,IAAI;SACrB;QACD,iBAAiB,EAAE;YACjB,SAAS,EAAE,sCAAsC;YACjD,aAAa,EAAE,sCAAsC;YACrD,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,IAAI;SACrB;KACF;IACD;QACE,MAAM,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;QACxB,IAAI,EAAE,CAAC,WAAW,CAAC;QACnB,KAAK,EAAE;YACL;gBACE,EAAE,EAAE,sCAAsC;gBAC1C,IAAI,EAAE,WAAW;gBACjB,KAAK,EAAE,IAAI;gBACX,UAAU,EAAE,EAAE;gBACd,IAAI,EAAE,CAAC,eAAe,EAAE,WAAW,CAAC;gBACpC,UAAU,EAAE,IAAI;gBAChB,KAAK,EAAE,IAAI;gBACX,MAAM,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,EAAE;aAClE;SACF;QACD,QAAQ,EAAE;YACR,MAAM,EAAE,OAAO;YACf,IAAI,EAAE,CAAC;YACP,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,sCAAsC;SAClD;QACD,UAAU,EAAE,0BAA0B;QACtC,UAAU,EAAE;YACV,SAAS,EAAE,sCAAsC;YACjD,aAAa,EAAE,sCAAsC;YACrD,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,IAAI;SACrB;QACD,iBAAiB,EAAE;YACjB,SAAS,EAAE,sCAAsC;YACjD,aAAa,EAAE,sCAAsC;YACrD,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,IAAI;SACrB;KACF;IACD;QACE,MAAM,EAAE;YACN,QAAQ,EAAE;gBACR;oBACE,OAAO,EAAE,OAAO;oBAChB,iBAAiB,EAAE,EAAE;oBACrB,iBAAiB,EAAE,EAAE;oBACrB,EAAE,EAAE,sCAAsC;oBAC1C,IAAI,EAAE,OAAO;iBACd;gBACD;oBACE,OAAO,EAAE,KAAK;oBACd,iBAAiB,EAAE,EAAE;oBACrB,iBAAiB,EAAE,EAAE;oBACrB,gBAAgB,EAAE,EAAE;oBACpB,EAAE,EAAE,0CAA0C;oBAC9C,UAAU,EAAE,EAAE;oBACd,kBAAkB,EAAE,EAAE;oBACtB,IAAI,EAAE,IAAI;iBACX;aACF;SACF;QACD,IAAI,EAAE,EAAE;QACR,KAAK,EAAE,EAAE;QACT,QAAQ,EAAE;YACR,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,CAAC;YACP,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,sCAAsC;SAClD;QACD,UAAU,EAAE,0BAA0B;QACtC,UAAU,EAAE;YACV,SAAS,EAAE,sCAAsC;YACjD,aAAa,EAAE,sCAAsC;YACrD,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,IAAI;SACrB;QACD,iBAAiB,EAAE;YACjB,SAAS,EAAE,sCAAsC;YACjD,aAAa,EAAE,sCAAsC;YACrD,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,IAAI;SACrB;KACF;IACD;QACE,MAAM,EAAE;YACN,QAAQ,EAAE;gBACR;oBACE,OAAO,EAAE,OAAO;oBAChB,iBAAiB,EAAE,EAAE;oBACrB,iBAAiB,EAAE,EAAE;oBACrB,EAAE,EAAE,sCAAsC;oBAC1C,IAAI,EAAE,OAAO;iBACd;gBACD;oBACE,OAAO,EAAE,KAAK;oBACd,iBAAiB,EAAE,EAAE;oBACrB,iBAAiB,EAAE,EAAE;oBACrB,gBAAgB,EAAE,EAAE;oBACpB,EAAE,EAAE,0CAA0C;oBAC9C,UAAU,EAAE,EAAE;oBACd,kBAAkB,EAAE,EAAE;oBACtB,IAAI,EAAE,IAAI;iBACX;aACF;SACF;QACD,IAAI,EAAE,EAAE;QACR,KAAK,EAAE,EAAE;QACT,QAAQ,EAAE;YACR,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,CAAC;YACP,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,sCAAsC;SAClD;QACD,UAAU,EAAE,0BAA0B;QACtC,UAAU,EAAE;YACV,SAAS,EAAE,sCAAsC;YACjD,aAAa,EAAE,sCAAsC;YACrD,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,IAAI;SACrB;QACD,iBAAiB,EAAE;YACjB,SAAS,EAAE,sCAAsC;YACjD,aAAa,EAAE,sCAAsC;YACrD,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,IAAI;SACrB;KACF;IACD;QACE,MAAM,EAAE;YACN,QAAQ,EAAE;gBACR;oBACE,OAAO,EAAE,OAAO;oBAChB,iBAAiB,EAAE,EAAE;oBACrB,iBAAiB,EAAE,EAAE;oBACrB,EAAE,EAAE,sCAAsC;oBAC1C,IAAI,EAAE,OAAO;iBACd;aACF;SACF;QACD,IAAI,EAAE,CAAC,OAAO,CAAC;QACf,KAAK,EAAE;YACL;gBACE,EAAE,EAAE,sCAAsC;gBAC1C,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,IAAI;gBACX,UAAU,EAAE,EAAE;gBACd,IAAI,EAAE,CAAC,eAAe,EAAE,OAAO,CAAC;gBAChC,UAAU,EAAE,IAAI;gBAChB,KAAK,EAAE,IAAI;gBACX,MAAM,EAAE;oBACN,QAAQ,EAAE;wBACR;4BACE,OAAO,EAAE,KAAK;4BACd,iBAAiB,EAAE,EAAE;4BACrB,iBAAiB,EAAE,EAAE;4BACrB,gBAAgB,EAAE,EAAE;4BACpB,EAAE,EAAE,0CAA0C;4BAC9C,UAAU,EAAE,EAAE;4BACd,kBAAkB,EAAE,EAAE;4BACtB,IAAI,EAAE,IAAI;yBACX;qBACF;iBACF;aACF;SACF;QACD,QAAQ,EAAE;YACR,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,CAAC;YACP,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,sCAAsC;SAClD;QACD,UAAU,EAAE,0BAA0B;QACtC,UAAU,EAAE;YACV,SAAS,EAAE,sCAAsC;YACjD,aAAa,EAAE,sCAAsC;YACrD,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,IAAI;SACrB;QACD,iBAAiB,EAAE;YACjB,SAAS,EAAE,sCAAsC;YACjD,aAAa,EAAE,sCAAsC;YACrD,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,IAAI;SACrB;KACF;IACD;QACE,MAAM,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;QACxB,IAAI,EAAE,CAAC,WAAW,CAAC;QACnB,KAAK,EAAE;YACL;gBACE,EAAE,EAAE,sCAAsC;gBAC1C,IAAI,EAAE,WAAW;gBACjB,KAAK,EAAE,IAAI;gBACX,UAAU,EAAE,EAAE;gBACd,IAAI,EAAE,CAAC,eAAe,EAAE,WAAW,CAAC;gBACpC,UAAU,EAAE,IAAI;gBAChB,KAAK,EAAE,IAAI;gBACX,MAAM,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE;aAC5D;SACF;QACD,QAAQ,EAAE;YACR,MAAM,EAAE,OAAO;YACf,IAAI,EAAE,CAAC,CAAC;YACR,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,sCAAsC;SAClD;QACD,UAAU,EAAE,0BAA0B;QACtC,UAAU,EAAE;YACV,SAAS,EAAE,sCAAsC;YACjD,aAAa,EAAE,sCAAsC;YACrD,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,IAAI;SACrB;QACD,iBAAiB,EAAE,IAAI;KACxB;CACF,CAAA;AAED,MAAM,IAAI,GAAG,CAAC,KAAuC,EAAE,QAA8C,EAAE,EAAE,EAAE,CAAC,CAAC;IAC3G,IAAI,EAAE,MAAM;IACZ,KAAK,EAAE,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK;IAC5D,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QACpB,IAAI,OAAO,CAAC,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,aAAa,CAAA;QACjD,CAAC;QACD,OAAO,CAAC,EAAE,UAAU,EAAE,aAAa,CAAA;IACrC,CAAC,CAAC;CACH,CAAC,CAAA;AAEF,MAAM,IAAI,GAAG,CAAC,GAAG,KAAgB,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAA;AAC/D,MAAM,QAAQ,GAAG,CAAC,GAAG,KAAgB,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAA;AAEvE,EAAE,CAAC,WAAW,EAAE,KAAK,IAAI,EAAE;IACzB,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAA;IAE1D,MAAM;SACH,IAAI,CAAC,KAAK,CAAC;SACX,aAAa,CACZ,MAAM,CAAC,eAAe,CACpB,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,aAAa,CAAC,CAAC,CAC9F,CACF,CAAA;IAEH,MAAM;SACH,IAAI,CAAC,YAAY,CAAC;SAClB,aAAa,CACZ,QAAQ,CACN,IAAI,CAAC,CAAC,CAAC,EACP,IAAI,CACF,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAClF,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CACnD,CACF,CACF,CAAA;AACL,CAAC,CAAC,CAAA;AAEF,EAAE,CAAC,cAAc,EAAE,KAAK,IAAI,EAAE;IAC5B,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;QAC3D,KAAK,EAAE,EAAE;QACT,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;KAChC,CAAC,CAAA;IAEF,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;QAC3D,KAAK,EAAE,EAAE;QACT,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;KACzC,CAAC,CAAA;IAEF,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;QAC3D,KAAK,EAAE,EAAE;QACT,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;KAClD,CAAC,CAAA;IAEF,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;QAC3D,KAAK,EAAE,EAAE;QACT,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;KAClD,CAAC,CAAA;IAEF,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;QAC3D,KAAK,EAAE,EAAE;QACT,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;KAClD,CAAC,CAAA;IAEF,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;QAC3D,KAAK,EAAE,MAAM,CAAC,eAAe,CAC3B,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,aAAa,CAAC,CAAC,CAC9F;QACD,YAAY,EAAE,QAAQ,CACpB,IAAI,CACF,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAClF,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CACnD,CACF;KACF,CAAC,CAAA;IAEF,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;QAC3D,KAAK,EAAE,MAAM,CAAC,eAAe,CAC3B,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,aAAa,CAAC,CAAC,CAC9F;QACD,YAAY,EAAE,QAAQ,CACpB,IAAI,CAAC,CAAC,CAAC,EACP,IAAI,CACF,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAClF,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CACnD,CACF;KACF,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/ui/errors.d.ts

@@ -0,0 +1,12 @@
+export declare class StreamError extends Error {
+    constructor(data: {
+        error?: string;
+        name?: string;
+        message: string;
+    });
+    static isStructuredError(error: unknown): error is {
+        error?: string;
+        name?: string;
+        message: string;
+    };
+}

package/dist/ui/errors.js

@@ -0,0 +1,10 @@
+export class StreamError extends Error {
+    constructor(data) {
+        super(data.message);
+        this.name = data.name ?? data.error ?? 'StreamError';
+    }
+    static isStructuredError(error) {
+        return typeof error === 'object' && error != null && 'message' in error;
+    }
+}
+//# sourceMappingURL=errors.js.map

package/dist/ui/errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../src/ui/errors.ts"],"names":[],"mappings":"AAAA,MAAM,OAAO,WAAY,SAAQ,KAAK;IACpC,YAAY,IAAwD;QAClE,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QACnB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,KAAK,IAAI,aAAa,CAAA;IACtD,CAAC;IAED,MAAM,CAAC,iBAAiB,CAAC,KAAc;QAKrC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,IAAI,IAAI,IAAI,SAAS,IAAI,KAAK,CAAA;IACzE,CAAC;CACF"}

package/dist/ui/errors.mjs

@@ -0,0 +1,10 @@
+export class StreamError extends Error {
+    constructor(data) {
+        super(data.message);
+        this.name = data.name ?? data.error ?? 'StreamError';
+    }
+    static isStructuredError(error) {
+        return typeof error === 'object' && error != null && 'message' in error;
+    }
+}
+//# sourceMappingURL=errors.mjs.map

package/dist/ui/errors.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.mjs","sourceRoot":"","sources":["../../src/ui/errors.ts"],"names":[],"mappings":"AAAA,MAAM,OAAO,WAAY,SAAQ,KAAK;IACpC,YAAY,IAAwD;QAClE,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QACnB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,KAAK,IAAI,aAAa,CAAA;IACtD,CAAC;IAED,MAAM,CAAC,iBAAiB,CAAC,KAAc;QAKrC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,IAAI,IAAI,IAAI,SAAS,IAAI,KAAK,CAAA;IACzE,CAAC;CACF"}

package/dist/ui/manager.d.ts

@@ -0,0 +1,87 @@
+import type { CheckpointsStreamEvent, CustomStreamEvent, DebugStreamEvent, ErrorStreamEvent, EventsStreamEvent, FeedbackStreamEvent, MessagesTupleStreamEvent, MetadataStreamEvent, TasksStreamEvent, UpdatesStreamEvent, ValuesStreamEvent } from '../types.stream.js';
+import { MessageTupleManager } from './messages.js';
+import type { Message } from '../types.messages.js';
+type BagTemplate = {
+    ConfigurableType?: Record<string, unknown>;
+    InterruptType?: unknown;
+    CustomEventType?: unknown;
+    UpdateType?: unknown;
+};
+type GetUpdateType<Bag extends BagTemplate, StateType extends Record<string, unknown>> = Bag extends {
+    UpdateType: unknown;
+} ? Bag['UpdateType'] : Partial<StateType>;
+type GetCustomEventType<Bag extends BagTemplate> = Bag extends {
+    CustomEventType: unknown;
+} ? Bag['CustomEventType'] : unknown;
+type EventStreamMap<StateType, UpdateType, CustomType> = {
+    values: ValuesStreamEvent<StateType>;
+    updates: UpdatesStreamEvent<UpdateType>;
+    custom: CustomStreamEvent<CustomType>;
+    debug: DebugStreamEvent;
+    messages: MessagesTupleStreamEvent;
+    events: EventsStreamEvent;
+    metadata: MetadataStreamEvent;
+    checkpoints: CheckpointsStreamEvent<StateType>;
+    tasks: TasksStreamEvent<StateType, UpdateType>;
+    error: ErrorStreamEvent;
+    feedback: FeedbackStreamEvent;
+};
+export type EventStreamEvent<StateType, UpdateType, CustomType> = EventStreamMap<StateType, UpdateType, CustomType>[keyof EventStreamMap<StateType, UpdateType, CustomType>];
+interface StreamManagerEventCallbacks<StateType extends Record<string, unknown>, Bag extends BagTemplate = BagTemplate> {
+    onUpdateEvent?: (data: UpdatesStreamEvent<GetUpdateType<Bag, StateType>>['data'], options: {
+        namespace: string[] | undefined;
+        mutate: (update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)) => void;
+    }) => void;
+    onCustomEvent?: (data: GetCustomEventType<Bag>, options: {
+        namespace: string[] | undefined;
+        mutate: (update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)) => void;
+    }) => void;
+    onMetadataEvent?: (data: MetadataStreamEvent['data']) => void;
+    onLangChainEvent?: (data: EventsStreamEvent['data']) => void;
+    onDebugEvent?: (data: DebugStreamEvent['data'], options: {
+        namespace: string[] | undefined;
+    }) => void;
+    onCheckpointEvent?: (data: CheckpointsStreamEvent<StateType>['data'], options: {
+        namespace: string[] | undefined;
+    }) => void;
+    onTaskEvent?: (data: TasksStreamEvent<StateType, GetUpdateType<Bag, StateType>>['data'], options: {
+        namespace: string[] | undefined;
+    }) => void;
+}
+export declare class StreamManager<StateType extends Record<string, unknown>, Bag extends BagTemplate = BagTemplate> {
+    private abortRef;
+    private messages;
+    private listeners;
+    private state;
+    constructor(messages: MessageTupleManager);
+    private setState;
+    private notifyListeners;
+    subscribe: (listener: () => void) => (() => void);
+    getSnapshot: () => {
+        isLoading: boolean;
+        values: [values: StateType, kind: "stream" | "stop"] | null;
+        error: unknown;
+    };
+    get isLoading(): boolean;
+    get values(): StateType | null;
+    get error(): unknown;
+    setStreamValues: (values: (StateType | null) | ((prev: StateType | null, kind: "stream" | "stop") => StateType | null), kind?: "stream" | "stop") => void;
+    private getMutateFn;
+    private matchEventType;
+    start: (action: (signal: AbortSignal) => Promise<AsyncGenerator<EventStreamEvent<StateType, GetUpdateType<Bag, StateType>, GetCustomEventType<Bag>>>>, options: {
+        getMessages: (values: StateType) => Message[];
+        setMessages: (current: StateType, messages: Message[]) => StateType;
+        initialValues: StateType;
+        callbacks: StreamManagerEventCallbacks<StateType, Bag>;
+        onSuccess: () => StateType | null | undefined | void | Promise<StateType | null | undefined | void>;
+        onError: (error: unknown) => void | Promise<void>;
+        onFinish?: () => void;
+    }) => Promise<void>;
+    stop: (historyValues: StateType, options: {
+        onStop?: (options: {
+            mutate: (update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)) => void;
+        }) => void;
+    }) => Promise<void>;
+    clear: () => void;
+}
+export {};

package/dist/ui/manager.js

@@ -0,0 +1,158 @@
+import { toMessageDict } from "./messages.js";
+import { StreamError } from "./errors.js";
+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.js.map