@dremio/js-sdk 0.59.0 → 0.61.0

package/dist/enterprise/ai/conversations/createConversationMachine.js

@@ -23,8 +23,17 @@ import { createExchangeRun } from "./methods/createExchangeRun.js";
 import { stopExchangeRun } from "./methods/stopExchangeRun.js";
 import { streamRunEvents } from "./methods/streamRunEvents.js";
 import { fromTextEventStream } from "../../../common/fromTextEventStream.js";
-import { map } from "rxjs";
+import { map, timeout } from "rxjs";
 import { chatEventCodec } from "../chat/chatEventSchema.js";
+const RETRIEVE_STATE_TIMEOUT_MS = 10_000;
+const STREAM_IDLE_TIMEOUT_MS = 60_000;
+const STREAM_RETRY_DELAY_MS = 2500;
+export const MAX_STREAM_RETRIES = 5;
+const STREAM_STATE_RESET = {
+    receivedEndOfStream: false,
+    streamRetryCount: 0,
+    userRequestedStop: false,
+};
 export const createConversationMachine = (config) => setup({
     actors: {
         createConversation: fromPromise(({ input }) => createConversation(config)(input).then((result) => {
@@ -46,9 +55,13 @@ export const createConversationMachine = (config) => setup({
             return result.value;
         })),
         retrieveState: fromPromise(async ({ input, signal }) => {
+            const timeoutSignal = AbortSignal.any([
+                signal,
+                AbortSignal.timeout(RETRIEVE_STATE_TIMEOUT_MS),
+            ]);
             const [conversationResult, historyResult] = await Promise.all([
-                retrieveConversation(config)(input.conversationId, { signal }),
-                retrieveConversationHistory(config)(input.conversationId, { signal }),
+                retrieveConversation(config)(input.conversationId, { signal: timeoutSignal }),
+                retrieveConversationHistory(config)(input.conversationId, { signal: timeoutSignal }),
             ]);
             if (conversationResult.isErr()) {
                 throw conversationResult.error;
@@ -73,7 +86,10 @@ export const createConversationMachine = (config) => setup({
             }
             return result.value;
         })),
-        streamRun: fromObservable(({ input }) => fromTextEventStream(({ signal }) => streamRunEvents(config)({ conversationId: input.conversationId, runId: input.runId }, { signal })).pipe(map((event) => z.decode(chatEventCodec, JSON.parse(event.data))))),
+        streamRun: fromObservable(({ input }) => fromTextEventStream(({ signal }) => streamRunEvents(config)({ conversationId: input.conversationId, runId: input.runId }, { signal })).pipe(map((event) => z.decode(chatEventCodec, JSON.parse(event.data))), timeout({ each: STREAM_IDLE_TIMEOUT_MS }))),
+    },
+    delays: {
+        STREAM_RETRY_DELAY: STREAM_RETRY_DELAY_MS,
     },
     types: {
         context: {},
@@ -87,6 +103,7 @@ export const createConversationMachine = (config) => setup({
         conversationSnapshot: undefined,
         currentRunId: undefined,
         messageAttempt: undefined,
+        ...STREAM_STATE_RESET,
     }),
     id: "agentConversation",
     initial: "uninitialized",
@@ -114,6 +131,7 @@ export const createConversationMachine = (config) => setup({
                                 ],
                                 currentRunId: ({ event }) => event.output.currentRunId,
                                 messageAttempt: undefined,
+                                ...STREAM_STATE_RESET,
                             }),
                             emit(({ event }) => ({
                                 id: event.output.id,
@@ -177,6 +195,7 @@ export const createConversationMachine = (config) => setup({
                         actions: assign({
                             conversationSnapshot: ({ context, event }) => AgentConversation.reduceChatEvents(context.conversationSnapshot, event.output.history.data),
                             currentRunId: ({ event }) => event.output.conversation.currentRunId,
+                            receivedEndOfStream: false,
                         }),
                         guard: ({ event }) => !!event.output.conversation.currentRunId,
                         target: "streaming",
@@ -185,6 +204,7 @@ export const createConversationMachine = (config) => setup({
                         actions: assign({
                             conversationSnapshot: ({ context, event }) => AgentConversation.reduceChatEvents(context.conversationSnapshot, event.output.history.data),
                             currentRunId: ({ event }) => event.output.conversation.currentRunId,
+                            ...STREAM_STATE_RESET,
                         }),
                         target: "idle",
                     },
@@ -193,12 +213,18 @@ export const createConversationMachine = (config) => setup({
                     {
                         guard: ({ event }) => {
                             const e = event.error;
+                            if (e?.name === "TimeoutError")
+                                return false;
                             return (e?.name === "AbortError" ||
                                 e?.message?.includes("aborted") ||
                                 e?.code === "ABORT_ERR");
                         },
                         target: "uninitialized",
                     },
+                    {
+                        guard: ({ context }) => context.streamRetryCount > 0,
+                        target: "stream_interrupted",
+                    },
                     {
                         actions: [
                             assign({
@@ -215,6 +241,30 @@ export const createConversationMachine = (config) => setup({
                 src: "retrieveState",
             },
         },
+        stream_interrupted: {
+            after: {
+                STREAM_RETRY_DELAY: [
+                    {
+                        actions: assign({
+                            streamRetryCount: ({ context }) => context.streamRetryCount + 1,
+                        }),
+                        guard: ({ context }) => context.streamRetryCount < MAX_STREAM_RETRIES,
+                        target: "retrieving_history",
+                    },
+                    {
+                        target: "stream_recovery_failed",
+                    },
+                ],
+            },
+        },
+        stream_recovery_failed: {
+            on: {
+                REFRESH_HISTORY: {
+                    actions: assign({ lastError: undefined, streamRetryCount: 0 }),
+                    target: "retrieving_history",
+                },
+            },
+        },
         streaming: {
             description: "Manages monitoring and canceling the current run",
             initial: "monitoring_replies",
@@ -228,20 +278,37 @@ export const createConversationMachine = (config) => setup({
                         runId: context.currentRunId,
                     };
                 },
-                onDone: {
-                    actions: assign({ currentRunId: null }),
-                    target: "idle",
-                },
-                onError: {
-                    actions: assign({ currentRunId: null }),
-                    target: "idle",
-                },
+                onDone: [
+                    {
+                        actions: assign({ currentRunId: null, userRequestedStop: false }),
+                        guard: ({ context }) => !!context.receivedEndOfStream || !!context.userRequestedStop,
+                        target: "idle",
+                    },
+                    {
+                        actions: assign({ currentRunId: null }),
+                        target: "stream_interrupted",
+                    },
+                ],
+                onError: [
+                    {
+                        actions: assign({ currentRunId: null, userRequestedStop: false }),
+                        guard: ({ context }) => !!context.receivedEndOfStream || !!context.userRequestedStop,
+                        target: "idle",
+                    },
+                    {
+                        actions: assign({ currentRunId: null }),
+                        target: "stream_interrupted",
+                    },
+                ],
                 onSnapshot: {
                     actions: enqueueActions(({ context, enqueue, event }) => {
                         const chatEvent = event.snapshot.context;
                         if (!chatEvent) {
                             return;
                         }
+                        if (chatEvent.content.chunkType === "endOfStream" && !context.receivedEndOfStream) {
+                            enqueue.assign({ receivedEndOfStream: true });
+                        }
                         const nextSnapshot = AgentConversation.reduceChatEvents(context.conversationSnapshot, [chatEvent]);
                         enqueue.assign({
                             conversationSnapshot: nextSnapshot,
@@ -259,6 +326,7 @@ export const createConversationMachine = (config) => setup({
                     description: "Attached to the run event stream",
                     on: {
                         STOP_RUN: {
+                            actions: assign({ userRequestedStop: true }),
                             target: "stopping",
                         },
                     },
@@ -271,10 +339,11 @@ export const createConversationMachine = (config) => setup({
                             runId: context.currentRunId,
                         }),
                         onDone: {
-                            actions: assign({ currentRunId: undefined }),
+                            actions: assign({ currentRunId: undefined, userRequestedStop: false }),
                             target: "#agentConversation.idle",
                         },
                         onError: {
+                            actions: assign({ userRequestedStop: false }),
                             target: "#agentConversation.retrieving_history",
                         },
                         src: "stopRun",
@@ -317,6 +386,7 @@ export const createConversationMachine = (config) => setup({
                         ],
                         currentRunId: ({ event }) => event.output.currentRunId,
                         messageAttempt: undefined,
+                        ...STREAM_STATE_RESET,
                     }),
                     target: "streaming",
                 },

package/dist/enterprise/ai/conversations/createConversationMachine.js.map

@@ -1 +1 @@
-{"version":3,"file":"createConversationMachine.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/conversations/createConversationMachine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAE1F,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAG3D,OAAO,EAAE,kBAAkB,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAC;AAC9F,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AACzE,OAAO,EAAE,2BAA2B,EAAE,MAAM,0CAA0C,CAAC;AACvF,OAAO,EAAE,iBAAiB,EAAE,MAAM,gCAAgC,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,mBAAmB,EAAE,MAAM,wCAAwC,CAAC;AAC7E,OAAO,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AAC3B,OAAO,EAAE,cAAc,EAAkB,MAAM,4BAA4B,CAAC;AAkC5E,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,MAAqB,EAAE,EAAE,CACjE,KAAK,CAAC;IACJ,MAAM,EAAE;QACN,kBAAkB,EAAE,WAAW,CAC7B,CAAC,EAAE,KAAK,EAAsD,EAAE,EAAE,CAChE,kBAAkB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YAChD,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACL;QACD,oBAAoB,EAAE,WAAW,CAC/B,CAAC,EAAE,KAAK,EAAE,MAAM,EAA8D,EAAE,EAAE,CAChF,oBAAoB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YAC7E,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACL;QACD,eAAe,EAAE,WAAW,CAC1B,CAAC,EAAE,KAAK,EAAE,MAAM,EAA8D,EAAE,EAAE,CAChF,2BAA2B,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YACpF,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACL;QACD,aAAa,EAAE,WAAW,CACxB,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAA8D,EAAE,EAAE;YACtF,MAAM,CAAC,kBAAkB,EAAE,aAAa,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC5D,oBAAoB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC;gBAC9D,2BAA2B,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC;aACtE,CAAC,CAAC;YAEH,IAAI,kBAAkB,CAAC,KAAK,EAAE,EAAE,CAAC;gBAC/B,MAAM,kBAAkB,CAAC,KAAK,CAAC;YACjC,CAAC;YAED,IAAI,aAAa,CAAC,KAAK,EAAE,EAAE,CAAC;gBAC1B,MAAM,aAAa,CAAC,KAAK,CAAC;YAC5B,CAAC;YAED,OAAO;gBACL,YAAY,EAAE,kBAAkB,CAAC,KAAK;gBACtC,OAAO,EAAE,aAAa,CAAC,KAAK;aAC7B,CAAC;QACJ,CAAC,CACF;QACD,QAAQ,EAAE,WAAW,CACnB,CAAC,EAAE,KAAK,EAAmE,EAAE,EAAE,CAC7E,iBAAiB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAC9E,CAAC,MAAM,EAAE,EAAE;YACT,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CACF,CACJ;QACD,OAAO,EAAE,WAAW,CAAC,CAAC,EAAE,KAAK,EAAwD,EAAE,EAAE,CACvF,eAAe,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YACzE,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACH;QACD,SAAS,EAAE,cAAc,CAAC,CAAC,EAAE,KAAK,EAAwD,EAAE,EAAE,CAC5F,mBAAmB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CACjC,eAAe,CAAC,MAAM,CAAC,CACrB,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,EAC5D,EAAE,MAAM,EAAE,CACX,CACF,CAAC,IAAI,CACJ,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,CAAC,CAAC,MAAM,CAAC,cAAc,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAmC,CAAC,CACnF,CACF,CACF;KACF;IACD,KAAK,EAAE;QACL,OAAO,EAAE,EAAgC;QACzC,OAAO,EAAE,EAAgC;QACzC,MAAM,EAAE,EAA+B;QACvC,KAAK,EAAE,EAA8B;KACtC;CACF,CAAC,CAAC,aAAa,CAAC;IACf,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACvB,cAAc,EAAE,KAAK,CAAC,cAAc;QACpC,oBAAoB,EAAE,SAAS;QAC/B,YAAY,EAAE,SAAS;QACvB,cAAc,EAAE,SAAS;KAC1B,CAAC;IACF,EAAE,EAAE,mBAAmB;IACvB,OAAO,EAAE,eAAe;IACxB,MAAM,EAAE;QACN,qBAAqB,EAAE;YACrB,WAAW,EAAE,oCAAoC;YACjD,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;oBACvB,OAAO,EAAE,OAAO,CAAC,cAAe;iBACjC,CAAC;gBACF,MAAM,EAAE;oBACN;wBACE,OAAO,EAAE;4BACP,MAAM,CAAC;gCACL,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE;gCAC9C,qDAAqD;gCACrD,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;oCAC5C,GAAG,CAAC,OAAO,CAAC,oBAAoB,IAAI,EAAE,CAAC;oCACvC;wCACE,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,YAAa;wCAC9B,QAAQ,EAAE,IAAI,GAAG,EAAE;wCACnB,oBAAoB,EAAE,OAAO,CAAC,cAAc;wCAC5C,SAAS,EAAE,IAAI,GAAG,EAAE;qCACU;iCACjC;gCACD,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;gCACtD,cAAc,EAAE,SAAS;6BAC1B,CAAC;4BACF,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;gCACnB,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,EAAE;gCACnB,IAAI,EAAE,qBAAqB;6BAC5B,CAAC,CAAC;yBACJ;wBACD,sFAAsF;wBACtF,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;wBACjD,MAAM,EAAE,WAAW;qBACpB;oBACD;;uBAEG;oBACH;wBACE,OAAO,EAAE,MAAM,CAAC;4BACd,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE;4BAC9C,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;4BACtD,cAAc,EAAE,SAAS;yBAC1B,CAAC;wBACF,MAAM,EAAE,oBAAoB;qBAC7B;iBACF;gBACD,OAAO,EAAE;oBACP,OAAO,EAAE,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;wBAC5B,KAAK,EAAE,KAAK,CAAC,KAAkB;wBAC/B,IAAI,EAAE,UAAU;qBACjB,CAAC,CAAC;oBACH,MAAM,EAAE,eAAe;iBACxB;gBACD,GAAG,EAAE,oBAAoB;aAC1B;SACF;QACD,IAAI,EAAE;YACJ,WAAW,EAAE,uDAAuD;YACpE,EAAE,EAAE;gBACF,eAAe,EAAE;oBACf,MAAM,EAAE,oBAAoB;iBAC7B;gBACD,mBAAmB,EAAE;oBACnB,OAAO,EAAE,MAAM,CAAC;wBACd,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO;qBAC7C,CAAC;oBACF,MAAM,EAAE,oBAAoB;iBAC7B;aACF;SACF;QACD,uBAAuB,EAAE;YACvB,EAAE,EAAE;gBACF,eAAe,EAAE;oBACf,OAAO,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;oBACzC,MAAM,EAAE,oBAAoB;iBAC7B;aACF;SACF;QACD,kBAAkB,EAAE;YAClB,WAAW,EAAE,gDAAgD;YAC7D,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,cAAc,EAAE,OAAO,CAAC,cAAe,EAAE,CAAC;gBACrE,MAAM,EAAE;oBACN;wBACE,OAAO,EAAE,MAAM,CAAC;4BACd,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAC3C,iBAAiB,CAAC,gBAAgB,CAChC,OAAO,CAAC,oBAAoB,EAC5B,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAC1B;4BACH,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,YAAY;yBACpE,CAAC;wBACF,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,YAAY;wBAC9D,MAAM,EAAE,WAAW;qBACpB;oBACD;wBACE,OAAO,EAAE,MAAM,CAAC;4BACd,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAC3C,iBAAiB,CAAC,gBAAgB,CAChC,OAAO,CAAC,oBAAoB,EAC5B,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAC1B;4BACH,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,YAAY;yBACpE,CAAC;wBACF,MAAM,EAAE,MAAM;qBACf;iBACF;gBACD,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE;4BACnB,MAAM,CAAC,GAAG,KAAK,CAAC,KAAkC,CAAC;4BACnD,OAAO,CACL,CAAC,EAAE,IAAI,KAAK,YAAY;gCACxB,CAAC,EAAE,OAAO,EAAE,QAAQ,CAAC,SAAS,CAAC;gCAC/B,CAAC,EAAE,IAAI,KAAK,WAAW,CACxB,CAAC;wBACJ,CAAC;wBACD,MAAM,EAAE,eAAe;qBACxB;oBACD;wBACE,OAAO,EAAE;4BACP,MAAM,CAAC;gCACL,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK;6BACtC,CAAC;4BACF,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;gCACnB,KAAK,EAAE,KAAK,CAAC,KAAkB;gCAC/B,IAAI,EAAE,UAAU;6BACjB,CAAC,CAAC;yBACJ;wBACD,MAAM,EAAE,yBAAyB;qBAClC;iBACF;gBACD,GAAG,EAAE,eAAe;aACrB;SACF;QACD,SAAS,EAAE;YACT,WAAW,EAAE,kDAAkD;YAC/D,OAAO,EAAE,oBAAoB;YAC7B,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;oBACrB,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;wBAC1B,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;oBACzD,CAAC;oBACD,OAAO;wBACL,cAAc,EAAE,OAAO,CAAC,cAAe;wBACvC,KAAK,EAAE,OAAO,CAAC,YAAY;qBAC5B,CAAC;gBACJ,CAAC;gBACD,MAAM,EAAE;oBACN,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;oBACvC,MAAM,EAAE,MAAM;iBACf;gBACD,OAAO,EAAE;oBACP,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;oBACvC,MAAM,EAAE,MAAM;iBACf;gBACD,UAAU,EAAE;oBACV,OAAO,EAAE,cAAc,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;wBACtD,MAAM,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;wBACzC,IAAI,CAAC,SAAS,EAAE,CAAC;4BACf,OAAO;wBACT,CAAC;wBAED,MAAM,YAAY,GAAG,iBAAiB,CAAC,gBAAgB,CACrD,OAAO,CAAC,oBAAoB,EAC5B,CAAC,SAAS,CAAC,CACZ,CAAC;wBAEF,OAAO,CAAC,MAAM,CAAC;4BACb,oBAAoB,EAAE,YAAY;yBACnC,CAAC,CAAC;wBACH,OAAO,CAAC,IAAI,CAAC;4BACX,SAAS;4BACT,IAAI,EAAE,mBAAmB;yBAC1B,CAAC,CAAC;oBACL,CAAC,CAAC;iBACH;gBACD,GAAG,EAAE,WAAW;aACjB;YACD,MAAM,EAAE;gBACN,kBAAkB,EAAE;oBAClB,WAAW,EAAE,kCAAkC;oBAC/C,EAAE,EAAE;wBACF,QAAQ,EAAE;4BACR,MAAM,EAAE,UAAU;yBACnB;qBACF;iBACF;gBACD,QAAQ,EAAE;oBACR,WAAW,EAAE,0CAA0C;oBACvD,MAAM,EAAE;wBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;4BACvB,cAAc,EAAE,OAAO,CAAC,cAAe;4BACvC,KAAK,EAAE,OAAO,CAAC,YAAa;yBAC7B,CAAC;wBACF,MAAM,EAAE;4BACN,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC;4BAC5C,MAAM,EAAE,yBAAyB;yBAClC;wBACD,OAAO,EAAE;4BACP,MAAM,EAAE,uCAAuC;yBAChD;wBACD,GAAG,EAAE,SAAS;qBACf;iBACF;aACF;SACF;QACD,kBAAkB,EAAE;YAClB,WAAW,EAAE,gDAAgD;YAC7D,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;oBAC5B,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;wBACnB,KAAK,eAAe;4BAClB,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;gCAC5B,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;4BAC3D,CAAC;4BACD,OAAO;gCACL,cAAc,EAAE,OAAO,CAAC,cAAe;gCACvC,OAAO,EAAE,OAAO,CAAC,cAAc;6BAChC,CAAC;wBACJ,KAAK,qBAAqB;4BACxB,OAAO;gCACL,cAAc,EAAE,OAAO,CAAC,cAAe;gCACvC,OAAO,EAAE,KAAK,CAAC,OAAO;6BACvB,CAAC;oBACN,CAAC;oBACD,MAAM,IAAI,KAAK,CAAC,yDAAyD,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;gBACzF,CAAC;gBACD,MAAM,EAAE;oBACN,OAAO,EAAE,MAAM,CAAC;wBACd,qDAAqD;wBACrD,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;4BAC5C,GAAG,CAAC,OAAO,CAAC,oBAAoB,IAAI,EAAE,CAAC;4BACvC;gCACE,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,YAAY;gCAC7B,QAAQ,EAAE,IAAI,GAAG,EAAE;gCACnB,oBAAoB,EAAE,OAAO,CAAC,cAAc,IAAI,SAAS;gCACzD,SAAS,EAAE,IAAI,GAAG,EAAE;6BACU;yBACjC;wBACD,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;wBACtD,cAAc,EAAE,SAAS;qBAC1B,CAAC;oBACF,MAAM,EAAE,WAAW;iBACpB;gBACD,OAAO,EAAE;oBACP,OAAO,EAAE;wBACP,MAAM,CAAC;4BACL,YAAY,EAAE,GAAG,EAAE,CAAC,IAAI;4BACxB,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK;yBACtC,CAAC;wBACF,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;4BACnB,KAAK,EAAE,KAAK,CAAC,KAAkB;4BAC/B,IAAI,EAAE,UAAU;yBACjB,CAAC,CAAC;qBACJ;oBACD,MAAM,EAAE,2BAA2B;iBACpC;gBACD,GAAG,EAAE,UAAU;aAChB;SACF;QACD,yBAAyB,EAAE;YACzB,WAAW,EAAE,0BAA0B;YACvC,EAAE,EAAE;gBACF,eAAe,EAAE;oBACf,OAAO,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;oBACzC,MAAM,EAAE,oBAAoB;iBAC7B;gBACD,aAAa,EAAE;oBACb,OAAO,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;oBACzC,MAAM,EAAE,oBAAoB;iBAC7B;gBACD,mBAAmB,EAAE;oBACnB,OAAO,EAAE,MAAM,CAAC;wBACd,SAAS,EAAE,SAAS;wBACpB,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO;qBAC7C,CAAC;oBACF,MAAM,EAAE,oBAAoB;iBAC7B;aACF;SACF;QACD,aAAa,EAAE;YACb,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc;gBAChD,MAAM,EAAE,oBAAoB;aAC7B;YACD,WAAW,EACT,gFAAgF;YAClF,EAAE,EAAE;gBACF,mBAAmB,EAAE;oBACnB,OAAO,EAAE,MAAM,CAAC;wBACd,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO;qBAC7C,CAAC;oBACF,MAAM,EAAE,uBAAuB;iBAChC;aACF;SACF;KACF;CACF,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { setup, fromObservable, fromPromise, assign, enqueueActions, emit } from \"xstate\";\nimport type { UserChatMessage } from \"../chat/UserChatMessage.ts\";\nimport { AgentConversation } from \"./AgentConversation.ts\";\nimport type { ConversationExchange } from \"./reduceChatEvents.ts\";\nimport type { SonarV4Config } from \"../../../common/Config.ts\";\nimport { createConversation, createConversationCodec } from \"./methods/createConversation.ts\";\nimport * as z from \"zod/mini\";\nimport { retrieveConversation } from \"./methods/retrieveConversation.ts\";\nimport { retrieveConversationHistory } from \"./methods/retrieveConversationHistory.ts\";\nimport { createExchangeRun } from \"./methods/createExchangeRun.ts\";\nimport { stopExchangeRun } from \"./methods/stopExchangeRun.ts\";\nimport { streamRunEvents } from \"./methods/streamRunEvents.ts\";\nimport { fromTextEventStream } from \"../../../common/fromTextEventStream.ts\";\nimport { map } from \"rxjs\";\nimport { chatEventCodec, type ChatEvent } from \"../chat/chatEventSchema.ts\";\nimport type { HttpError } from \"../../../common/HttpError.ts\";\n\nexport type ConversationMachineInput = {\n  conversationId: string | null;\n};\n\nexport type ConversationMachineContext = {\n  conversationId: string | null;\n  conversationSnapshot?: ConversationExchange[];\n  currentRunId?: string | null;\n  lastError?: unknown;\n  messageAttempt?: UserChatMessage;\n};\nexport type ConversationMachineEvents =\n  | { type: \"STOP_RUN\" }\n  | { type: \"SUBMIT_USER_MESSAGE\"; message: UserChatMessage }\n  | { type: \"RETRY_MESSAGE\" }\n  | { type: \"REFRESH_HISTORY\" };\n\nexport type ConversationMachineEmitted =\n  | {\n      type: \"apiError\";\n      error: HttpError;\n    }\n  | {\n      type: \"chatEventReceived\";\n      chatEvent: ChatEvent;\n    }\n  | {\n      type: \"conversationCreated\";\n      id: string;\n    };\n\nexport const createConversationMachine = (config: SonarV4Config) =>\n  setup({\n    actors: {\n      createConversation: fromPromise(\n        ({ input }: { input: z.input<typeof createConversationCodec> }) =>\n          createConversation(config)(input).then((result) => {\n            if (result.isErr()) {\n              throw result.error;\n            }\n            return result.value;\n          }),\n      ),\n      retrieveConversation: fromPromise(\n        ({ input, signal }: { input: { conversationId: string }; signal: AbortSignal }) =>\n          retrieveConversation(config)(input.conversationId, { signal }).then((result) => {\n            if (result.isErr()) {\n              throw result.error;\n            }\n            return result.value;\n          }),\n      ),\n      retrieveHistory: fromPromise(\n        ({ input, signal }: { input: { conversationId: string }; signal: AbortSignal }) =>\n          retrieveConversationHistory(config)(input.conversationId, { signal }).then((result) => {\n            if (result.isErr()) {\n              throw result.error;\n            }\n            return result.value;\n          }),\n      ),\n      retrieveState: fromPromise(\n        async ({ input, signal }: { input: { conversationId: string }; signal: AbortSignal }) => {\n          const [conversationResult, historyResult] = await Promise.all([\n            retrieveConversation(config)(input.conversationId, { signal }),\n            retrieveConversationHistory(config)(input.conversationId, { signal }),\n          ]);\n\n          if (conversationResult.isErr()) {\n            throw conversationResult.error;\n          }\n\n          if (historyResult.isErr()) {\n            throw historyResult.error;\n          }\n\n          return {\n            conversation: conversationResult.value,\n            history: historyResult.value,\n          };\n        },\n      ),\n      startRun: fromPromise(\n        ({ input }: { input: { conversationId: string; message: UserChatMessage } }) =>\n          createExchangeRun(config)(input.conversationId, { message: input.message }).then(\n            (result) => {\n              if (result.isErr()) {\n                throw result.error;\n              }\n              return result.value;\n            },\n          ),\n      ),\n      stopRun: fromPromise(({ input }: { input: { conversationId: string; runId: string } }) =>\n        stopExchangeRun(config)(input.conversationId, input.runId).then((result) => {\n          if (result.isErr()) {\n            throw result.error;\n          }\n          return result.value;\n        }),\n      ),\n      streamRun: fromObservable(({ input }: { input: { conversationId: string; runId: string } }) =>\n        fromTextEventStream(({ signal }) =>\n          streamRunEvents(config)(\n            { conversationId: input.conversationId, runId: input.runId },\n            { signal },\n          ),\n        ).pipe(\n          map((event) =>\n            z.decode(chatEventCodec, JSON.parse(event.data) as z.input<typeof chatEventCodec>),\n          ),\n        ),\n      ),\n    },\n    types: {\n      context: {} as ConversationMachineContext,\n      emitted: {} as ConversationMachineEmitted,\n      events: {} as ConversationMachineEvents,\n      input: {} as ConversationMachineInput,\n    },\n  }).createMachine({\n    context: ({ input }) => ({\n      conversationId: input.conversationId,\n      conversationSnapshot: undefined,\n      currentRunId: undefined,\n      messageAttempt: undefined,\n    }),\n    id: \"agentConversation\",\n    initial: \"uninitialized\",\n    states: {\n      creating_conversation: {\n        description: \"Creating a new conversation object\",\n        invoke: {\n          input: ({ context }) => ({\n            message: context.messageAttempt!,\n          }),\n          onDone: [\n            {\n              actions: [\n                assign({\n                  conversationId: ({ event }) => event.output.id,\n                  // Add the message attempt into the snapshot manually\n                  conversationSnapshot: ({ context, event }) => [\n                    ...(context.conversationSnapshot ?? []),\n                    {\n                      id: event.output.currentRunId!,\n                      messages: new Map(),\n                      submittedUserMessage: context.messageAttempt,\n                      toolCalls: new Map(),\n                    } satisfies ConversationExchange,\n                  ],\n                  currentRunId: ({ event }) => event.output.currentRunId,\n                  messageAttempt: undefined,\n                }),\n                emit(({ event }) => ({\n                  id: event.output.id,\n                  type: \"conversationCreated\",\n                })),\n              ],\n              // Don't transition to streaming in a situation where we didn't receive a currentRunId\n              guard: ({ event }) => !!event.output.currentRunId,\n              target: \"streaming\",\n            },\n            /**\n             * Fall back to the retrieving_history state if we didn't receive a currentRunId\n             */\n            {\n              actions: assign({\n                conversationId: ({ event }) => event.output.id,\n                currentRunId: ({ event }) => event.output.currentRunId,\n                messageAttempt: undefined,\n              }),\n              target: \"retrieving_history\",\n            },\n          ],\n          onError: {\n            actions: emit(({ event }) => ({\n              error: event.error as HttpError,\n              type: \"apiError\",\n            })),\n            target: \"uninitialized\",\n          },\n          src: \"createConversation\",\n        },\n      },\n      idle: {\n        description: \"Waiting for user input to start the next reply stream\",\n        on: {\n          REFRESH_HISTORY: {\n            target: \"retrieving_history\",\n          },\n          SUBMIT_USER_MESSAGE: {\n            actions: assign({\n              messageAttempt: ({ event }) => event.message,\n            }),\n            target: \"submitting_message\",\n          },\n        },\n      },\n      retrieve_history_failed: {\n        on: {\n          REFRESH_HISTORY: {\n            actions: assign({ lastError: undefined }),\n            target: \"retrieving_history\",\n          },\n        },\n      },\n      retrieving_history: {\n        description: \"Retrieving latest copy of conversation history\",\n        invoke: {\n          input: ({ context }) => ({ conversationId: context.conversationId! }),\n          onDone: [\n            {\n              actions: assign({\n                conversationSnapshot: ({ context, event }) =>\n                  AgentConversation.reduceChatEvents(\n                    context.conversationSnapshot,\n                    event.output.history.data,\n                  ),\n                currentRunId: ({ event }) => event.output.conversation.currentRunId,\n              }),\n              guard: ({ event }) => !!event.output.conversation.currentRunId,\n              target: \"streaming\",\n            },\n            {\n              actions: assign({\n                conversationSnapshot: ({ context, event }) =>\n                  AgentConversation.reduceChatEvents(\n                    context.conversationSnapshot,\n                    event.output.history.data,\n                  ),\n                currentRunId: ({ event }) => event.output.conversation.currentRunId,\n              }),\n              target: \"idle\",\n            },\n          ],\n          onError: [\n            {\n              guard: ({ event }) => {\n                const e = event.error as Error & { code?: string };\n                return (\n                  e?.name === \"AbortError\" ||\n                  e?.message?.includes(\"aborted\") ||\n                  e?.code === \"ABORT_ERR\"\n                );\n              },\n              target: \"uninitialized\",\n            },\n            {\n              actions: [\n                assign({\n                  lastError: ({ event }) => event.error,\n                }),\n                emit(({ event }) => ({\n                  error: event.error as HttpError,\n                  type: \"apiError\",\n                })),\n              ],\n              target: \"retrieve_history_failed\",\n            },\n          ],\n          src: \"retrieveState\",\n        },\n      },\n      streaming: {\n        description: \"Manages monitoring and canceling the current run\",\n        initial: \"monitoring_replies\",\n        invoke: {\n          input: ({ context }) => {\n            if (!context.currentRunId) {\n              throw new Error(\"Attempted to stream without a runId\");\n            }\n            return {\n              conversationId: context.conversationId!,\n              runId: context.currentRunId,\n            };\n          },\n          onDone: {\n            actions: assign({ currentRunId: null }),\n            target: \"idle\",\n          },\n          onError: {\n            actions: assign({ currentRunId: null }),\n            target: \"idle\",\n          },\n          onSnapshot: {\n            actions: enqueueActions(({ context, enqueue, event }) => {\n              const chatEvent = event.snapshot.context;\n              if (!chatEvent) {\n                return;\n              }\n\n              const nextSnapshot = AgentConversation.reduceChatEvents(\n                context.conversationSnapshot,\n                [chatEvent],\n              );\n\n              enqueue.assign({\n                conversationSnapshot: nextSnapshot,\n              });\n              enqueue.emit({\n                chatEvent,\n                type: \"chatEventReceived\",\n              });\n            }),\n          },\n          src: \"streamRun\",\n        },\n        states: {\n          monitoring_replies: {\n            description: \"Attached to the run event stream\",\n            on: {\n              STOP_RUN: {\n                target: \"stopping\",\n              },\n            },\n          },\n          stopping: {\n            description: \"Background cancelation of the run stream\",\n            invoke: {\n              input: ({ context }) => ({\n                conversationId: context.conversationId!,\n                runId: context.currentRunId!,\n              }),\n              onDone: {\n                actions: assign({ currentRunId: undefined }),\n                target: \"#agentConversation.idle\",\n              },\n              onError: {\n                target: \"#agentConversation.retrieving_history\",\n              },\n              src: \"stopRun\",\n            },\n          },\n        },\n      },\n      submitting_message: {\n        description: \"Submitting message to start a new reply stream\",\n        invoke: {\n          input: ({ context, event }) => {\n            switch (event.type) {\n              case \"RETRY_MESSAGE\":\n                if (!context.messageAttempt) {\n                  throw new Error(\"Tried to resubmit nonexistent message\");\n                }\n                return {\n                  conversationId: context.conversationId!,\n                  message: context.messageAttempt,\n                };\n              case \"SUBMIT_USER_MESSAGE\":\n                return {\n                  conversationId: context.conversationId!,\n                  message: event.message,\n                };\n            }\n            throw new Error(`Unexpected event type received in submitting_message: ${event.type}`);\n          },\n          onDone: {\n            actions: assign({\n              // Add the message attempt into the snapshot manually\n              conversationSnapshot: ({ context, event }) => [\n                ...(context.conversationSnapshot ?? []),\n                {\n                  id: event.output.currentRunId,\n                  messages: new Map(),\n                  submittedUserMessage: context.messageAttempt || undefined,\n                  toolCalls: new Map(),\n                } satisfies ConversationExchange,\n              ],\n              currentRunId: ({ event }) => event.output.currentRunId,\n              messageAttempt: undefined,\n            }),\n            target: \"streaming\",\n          },\n          onError: {\n            actions: [\n              assign({\n                currentRunId: () => null,\n                lastError: ({ event }) => event.error,\n              }),\n              emit(({ event }) => ({\n                error: event.error as HttpError,\n                type: \"apiError\",\n              })),\n            ],\n            target: \"submitting_message_failed\",\n          },\n          src: \"startRun\",\n        },\n      },\n      submitting_message_failed: {\n        description: \"Failed to submit message\",\n        on: {\n          REFRESH_HISTORY: {\n            actions: assign({ lastError: undefined }),\n            target: \"retrieving_history\",\n          },\n          RETRY_MESSAGE: {\n            actions: assign({ lastError: undefined }),\n            target: \"submitting_message\",\n          },\n          SUBMIT_USER_MESSAGE: {\n            actions: assign({\n              lastError: undefined,\n              messageAttempt: ({ event }) => event.message,\n            }),\n            target: \"submitting_message\",\n          },\n        },\n      },\n      uninitialized: {\n        always: {\n          guard: ({ context }) => !!context.conversationId,\n          target: \"retrieving_history\",\n        },\n        description:\n          \"Entry state that the machine remains in when no conversation has been provided\",\n        on: {\n          SUBMIT_USER_MESSAGE: {\n            actions: assign({\n              messageAttempt: ({ event }) => event.message,\n            }),\n            target: \"creating_conversation\",\n          },\n        },\n      },\n    },\n  });\n\nexport type AgentConversationMachine = ReturnType<typeof createConversationMachine>;\n"]}
+{"version":3,"file":"createConversationMachine.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/conversations/createConversationMachine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAE1F,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAG3D,OAAO,EAAE,kBAAkB,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAC;AAC9F,OAAO,KAAK,CAAC,MAAM,UAAU,CAAC;AAC9B,OAAO,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AACzE,OAAO,EAAE,2BAA2B,EAAE,MAAM,0CAA0C,CAAC;AACvF,OAAO,EAAE,iBAAiB,EAAE,MAAM,gCAAgC,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,mBAAmB,EAAE,MAAM,wCAAwC,CAAC;AAC7E,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACpC,OAAO,EAAE,cAAc,EAAkB,MAAM,4BAA4B,CAAC;AAqC5E,MAAM,yBAAyB,GAAG,MAAM,CAAC;AACzC,MAAM,sBAAsB,GAAG,MAAM,CAAC;AACtC,MAAM,qBAAqB,GAAG,IAAI,CAAC;AACnC,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC;AACpC,MAAM,kBAAkB,GAAG;IACzB,mBAAmB,EAAE,KAAK;IAC1B,gBAAgB,EAAE,CAAC;IACnB,iBAAiB,EAAE,KAAK;CACzB,CAAC;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,MAAqB,EAAE,EAAE,CACjE,KAAK,CAAC;IACJ,MAAM,EAAE;QACN,kBAAkB,EAAE,WAAW,CAC7B,CAAC,EAAE,KAAK,EAAsD,EAAE,EAAE,CAChE,kBAAkB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YAChD,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACL;QACD,oBAAoB,EAAE,WAAW,CAC/B,CAAC,EAAE,KAAK,EAAE,MAAM,EAA8D,EAAE,EAAE,CAChF,oBAAoB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YAC7E,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACL;QACD,eAAe,EAAE,WAAW,CAC1B,CAAC,EAAE,KAAK,EAAE,MAAM,EAA8D,EAAE,EAAE,CAChF,2BAA2B,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YACpF,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACL;QACD,aAAa,EAAE,WAAW,CACxB,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAA8D,EAAE,EAAE;YACtF,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,CAAC;gBACpC,MAAM;gBACN,WAAW,CAAC,OAAO,CAAC,yBAAyB,CAAC;aAC/C,CAAC,CAAC;YACH,MAAM,CAAC,kBAAkB,EAAE,aAAa,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC5D,oBAAoB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;gBAC7E,2BAA2B,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;aACrF,CAAC,CAAC;YAEH,IAAI,kBAAkB,CAAC,KAAK,EAAE,EAAE,CAAC;gBAC/B,MAAM,kBAAkB,CAAC,KAAK,CAAC;YACjC,CAAC;YAED,IAAI,aAAa,CAAC,KAAK,EAAE,EAAE,CAAC;gBAC1B,MAAM,aAAa,CAAC,KAAK,CAAC;YAC5B,CAAC;YAED,OAAO;gBACL,YAAY,EAAE,kBAAkB,CAAC,KAAK;gBACtC,OAAO,EAAE,aAAa,CAAC,KAAK;aAC7B,CAAC;QACJ,CAAC,CACF;QACD,QAAQ,EAAE,WAAW,CACnB,CAAC,EAAE,KAAK,EAAmE,EAAE,EAAE,CAC7E,iBAAiB,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAC9E,CAAC,MAAM,EAAE,EAAE;YACT,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CACF,CACJ;QACD,OAAO,EAAE,WAAW,CAAC,CAAC,EAAE,KAAK,EAAwD,EAAE,EAAE,CACvF,eAAe,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YACzE,IAAI,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC;gBACnB,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC,CAAC,CACH;QACD,SAAS,EAAE,cAAc,CAAC,CAAC,EAAE,KAAK,EAAwD,EAAE,EAAE,CAC5F,mBAAmB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CACjC,eAAe,CAAC,MAAM,CAAC,CACrB,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,EAC5D,EAAE,MAAM,EAAE,CACX,CACF,CAAC,IAAI,CACJ,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACZ,CAAC,CAAC,MAAM,CAAC,cAAc,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAmC,CAAC,CACnF,EACD,OAAO,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC,CAC1C,CACF;KACF;IACD,MAAM,EAAE;QACN,kBAAkB,EAAE,qBAAqB;KAC1C;IACD,KAAK,EAAE;QACL,OAAO,EAAE,EAAgC;QACzC,OAAO,EAAE,EAAgC;QACzC,MAAM,EAAE,EAA+B;QACvC,KAAK,EAAE,EAA8B;KACtC;CACF,CAAC,CAAC,aAAa,CAAC;IACf,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACvB,cAAc,EAAE,KAAK,CAAC,cAAc;QACpC,oBAAoB,EAAE,SAAS;QAC/B,YAAY,EAAE,SAAS;QACvB,cAAc,EAAE,SAAS;QACzB,GAAG,kBAAkB;KACtB,CAAC;IACF,EAAE,EAAE,mBAAmB;IACvB,OAAO,EAAE,eAAe;IACxB,MAAM,EAAE;QACN,qBAAqB,EAAE;YACrB,WAAW,EAAE,oCAAoC;YACjD,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;oBACvB,OAAO,EAAE,OAAO,CAAC,cAAe;iBACjC,CAAC;gBACF,MAAM,EAAE;oBACN;wBACE,OAAO,EAAE;4BACP,MAAM,CAAC;gCACL,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE;gCAC9C,qDAAqD;gCACrD,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;oCAC5C,GAAG,CAAC,OAAO,CAAC,oBAAoB,IAAI,EAAE,CAAC;oCACvC;wCACE,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,YAAa;wCAC9B,QAAQ,EAAE,IAAI,GAAG,EAAE;wCACnB,oBAAoB,EAAE,OAAO,CAAC,cAAc;wCAC5C,SAAS,EAAE,IAAI,GAAG,EAAE;qCACU;iCACjC;gCACD,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;gCACtD,cAAc,EAAE,SAAS;gCACzB,GAAG,kBAAkB;6BACtB,CAAC;4BACF,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;gCACnB,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,EAAE;gCACnB,IAAI,EAAE,qBAAqB;6BAC5B,CAAC,CAAC;yBACJ;wBACD,sFAAsF;wBACtF,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;wBACjD,MAAM,EAAE,WAAW;qBACpB;oBACD;;uBAEG;oBACH;wBACE,OAAO,EAAE,MAAM,CAAC;4BACd,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE;4BAC9C,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;4BACtD,cAAc,EAAE,SAAS;yBAC1B,CAAC;wBACF,MAAM,EAAE,oBAAoB;qBAC7B;iBACF;gBACD,OAAO,EAAE;oBACP,OAAO,EAAE,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;wBAC5B,KAAK,EAAE,KAAK,CAAC,KAAkB;wBAC/B,IAAI,EAAE,UAAU;qBACjB,CAAC,CAAC;oBACH,MAAM,EAAE,eAAe;iBACxB;gBACD,GAAG,EAAE,oBAAoB;aAC1B;SACF;QACD,IAAI,EAAE;YACJ,WAAW,EAAE,uDAAuD;YACpE,EAAE,EAAE;gBACF,eAAe,EAAE;oBACf,MAAM,EAAE,oBAAoB;iBAC7B;gBACD,mBAAmB,EAAE;oBACnB,OAAO,EAAE,MAAM,CAAC;wBACd,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO;qBAC7C,CAAC;oBACF,MAAM,EAAE,oBAAoB;iBAC7B;aACF;SACF;QACD,uBAAuB,EAAE;YACvB,EAAE,EAAE;gBACF,eAAe,EAAE;oBACf,OAAO,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;oBACzC,MAAM,EAAE,oBAAoB;iBAC7B;aACF;SACF;QACD,kBAAkB,EAAE;YAClB,WAAW,EAAE,gDAAgD;YAC7D,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,cAAc,EAAE,OAAO,CAAC,cAAe,EAAE,CAAC;gBACrE,MAAM,EAAE;oBACN;wBACE,OAAO,EAAE,MAAM,CAAC;4BACd,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAC3C,iBAAiB,CAAC,gBAAgB,CAChC,OAAO,CAAC,oBAAoB,EAC5B,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAC1B;4BACH,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,YAAY;4BACnE,mBAAmB,EAAE,KAAK;yBAC3B,CAAC;wBACF,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,YAAY;wBAC9D,MAAM,EAAE,WAAW;qBACpB;oBACD;wBACE,OAAO,EAAE,MAAM,CAAC;4BACd,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAC3C,iBAAiB,CAAC,gBAAgB,CAChC,OAAO,CAAC,oBAAoB,EAC5B,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAC1B;4BACH,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,YAAY;4BACnE,GAAG,kBAAkB;yBACtB,CAAC;wBACF,MAAM,EAAE,MAAM;qBACf;iBACF;gBACD,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE;4BACnB,MAAM,CAAC,GAAG,KAAK,CAAC,KAAkC,CAAC;4BACnD,IAAI,CAAC,EAAE,IAAI,KAAK,cAAc;gCAAE,OAAO,KAAK,CAAC;4BAC7C,OAAO,CACL,CAAC,EAAE,IAAI,KAAK,YAAY;gCACxB,CAAC,EAAE,OAAO,EAAE,QAAQ,CAAC,SAAS,CAAC;gCAC/B,CAAC,EAAE,IAAI,KAAK,WAAW,CACxB,CAAC;wBACJ,CAAC;wBACD,MAAM,EAAE,eAAe;qBACxB;oBACD;wBACE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,gBAAgB,GAAG,CAAC;wBACpD,MAAM,EAAE,oBAAoB;qBAC7B;oBACD;wBACE,OAAO,EAAE;4BACP,MAAM,CAAC;gCACL,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK;6BACtC,CAAC;4BACF,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;gCACnB,KAAK,EAAE,KAAK,CAAC,KAAkB;gCAC/B,IAAI,EAAE,UAAU;6BACjB,CAAC,CAAC;yBACJ;wBACD,MAAM,EAAE,yBAAyB;qBAClC;iBACF;gBACD,GAAG,EAAE,eAAe;aACrB;SACF;QACD,kBAAkB,EAAE;YAClB,KAAK,EAAE;gBACL,kBAAkB,EAAE;oBAClB;wBACE,OAAO,EAAE,MAAM,CAAC;4BACd,gBAAgB,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,gBAAgB,GAAG,CAAC;yBAChE,CAAC;wBACF,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,gBAAgB,GAAG,kBAAkB;wBACrE,MAAM,EAAE,oBAAoB;qBAC7B;oBACD;wBACE,MAAM,EAAE,wBAAwB;qBACjC;iBACF;aACF;SACF;QACD,sBAAsB,EAAE;YACtB,EAAE,EAAE;gBACF,eAAe,EAAE;oBACf,OAAO,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,gBAAgB,EAAE,CAAC,EAAE,CAAC;oBAC9D,MAAM,EAAE,oBAAoB;iBAC7B;aACF;SACF;QACD,SAAS,EAAE;YACT,WAAW,EAAE,kDAAkD;YAC/D,OAAO,EAAE,oBAAoB;YAC7B,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;oBACrB,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;wBAC1B,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;oBACzD,CAAC;oBACD,OAAO;wBACL,cAAc,EAAE,OAAO,CAAC,cAAe;wBACvC,KAAK,EAAE,OAAO,CAAC,YAAY;qBAC5B,CAAC;gBACJ,CAAC;gBACD,MAAM,EAAE;oBACN;wBACE,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;wBACjE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,mBAAmB,IAAI,CAAC,CAAC,OAAO,CAAC,iBAAiB;wBACpF,MAAM,EAAE,MAAM;qBACf;oBACD;wBACE,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;wBACvC,MAAM,EAAE,oBAAoB;qBAC7B;iBACF;gBACD,OAAO,EAAE;oBACP;wBACE,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;wBACjE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,mBAAmB,IAAI,CAAC,CAAC,OAAO,CAAC,iBAAiB;wBACpF,MAAM,EAAE,MAAM;qBACf;oBACD;wBACE,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;wBACvC,MAAM,EAAE,oBAAoB;qBAC7B;iBACF;gBACD,UAAU,EAAE;oBACV,OAAO,EAAE,cAAc,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;wBACtD,MAAM,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;wBACzC,IAAI,CAAC,SAAS,EAAE,CAAC;4BACf,OAAO;wBACT,CAAC;wBAED,IAAI,SAAS,CAAC,OAAO,CAAC,SAAS,KAAK,aAAa,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC;4BAClF,OAAO,CAAC,MAAM,CAAC,EAAE,mBAAmB,EAAE,IAAI,EAAE,CAAC,CAAC;wBAChD,CAAC;wBAED,MAAM,YAAY,GAAG,iBAAiB,CAAC,gBAAgB,CACrD,OAAO,CAAC,oBAAoB,EAC5B,CAAC,SAAS,CAAC,CACZ,CAAC;wBAEF,OAAO,CAAC,MAAM,CAAC;4BACb,oBAAoB,EAAE,YAAY;yBACnC,CAAC,CAAC;wBACH,OAAO,CAAC,IAAI,CAAC;4BACX,SAAS;4BACT,IAAI,EAAE,mBAAmB;yBAC1B,CAAC,CAAC;oBACL,CAAC,CAAC;iBACH;gBACD,GAAG,EAAE,WAAW;aACjB;YACD,MAAM,EAAE;gBACN,kBAAkB,EAAE;oBAClB,WAAW,EAAE,kCAAkC;oBAC/C,EAAE,EAAE;wBACF,QAAQ,EAAE;4BACR,OAAO,EAAE,MAAM,CAAC,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC;4BAC5C,MAAM,EAAE,UAAU;yBACnB;qBACF;iBACF;gBACD,QAAQ,EAAE;oBACR,WAAW,EAAE,0CAA0C;oBACvD,MAAM,EAAE;wBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;4BACvB,cAAc,EAAE,OAAO,CAAC,cAAe;4BACvC,KAAK,EAAE,OAAO,CAAC,YAAa;yBAC7B,CAAC;wBACF,MAAM,EAAE;4BACN,OAAO,EAAE,MAAM,CAAC,EAAE,YAAY,EAAE,SAAS,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;4BACtE,MAAM,EAAE,yBAAyB;yBAClC;wBACD,OAAO,EAAE;4BACP,OAAO,EAAE,MAAM,CAAC,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;4BAC7C,MAAM,EAAE,uCAAuC;yBAChD;wBACD,GAAG,EAAE,SAAS;qBACf;iBACF;aACF;SACF;QACD,kBAAkB,EAAE;YAClB,WAAW,EAAE,gDAAgD;YAC7D,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;oBAC5B,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;wBACnB,KAAK,eAAe;4BAClB,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;gCAC5B,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;4BAC3D,CAAC;4BACD,OAAO;gCACL,cAAc,EAAE,OAAO,CAAC,cAAe;gCACvC,OAAO,EAAE,OAAO,CAAC,cAAc;6BAChC,CAAC;wBACJ,KAAK,qBAAqB;4BACxB,OAAO;gCACL,cAAc,EAAE,OAAO,CAAC,cAAe;gCACvC,OAAO,EAAE,KAAK,CAAC,OAAO;6BACvB,CAAC;oBACN,CAAC;oBACD,MAAM,IAAI,KAAK,CAAC,yDAAyD,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;gBACzF,CAAC;gBACD,MAAM,EAAE;oBACN,OAAO,EAAE,MAAM,CAAC;wBACd,qDAAqD;wBACrD,oBAAoB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;4BAC5C,GAAG,CAAC,OAAO,CAAC,oBAAoB,IAAI,EAAE,CAAC;4BACvC;gCACE,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,YAAY;gCAC7B,QAAQ,EAAE,IAAI,GAAG,EAAE;gCACnB,oBAAoB,EAAE,OAAO,CAAC,cAAc,IAAI,SAAS;gCACzD,SAAS,EAAE,IAAI,GAAG,EAAE;6BACU;yBACjC;wBACD,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY;wBACtD,cAAc,EAAE,SAAS;wBACzB,GAAG,kBAAkB;qBACtB,CAAC;oBACF,MAAM,EAAE,WAAW;iBACpB;gBACD,OAAO,EAAE;oBACP,OAAO,EAAE;wBACP,MAAM,CAAC;4BACL,YAAY,EAAE,GAAG,EAAE,CAAC,IAAI;4BACxB,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK;yBACtC,CAAC;wBACF,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;4BACnB,KAAK,EAAE,KAAK,CAAC,KAAkB;4BAC/B,IAAI,EAAE,UAAU;yBACjB,CAAC,CAAC;qBACJ;oBACD,MAAM,EAAE,2BAA2B;iBACpC;gBACD,GAAG,EAAE,UAAU;aAChB;SACF;QACD,yBAAyB,EAAE;YACzB,WAAW,EAAE,0BAA0B;YACvC,EAAE,EAAE;gBACF,eAAe,EAAE;oBACf,OAAO,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;oBACzC,MAAM,EAAE,oBAAoB;iBAC7B;gBACD,aAAa,EAAE;oBACb,OAAO,EAAE,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;oBACzC,MAAM,EAAE,oBAAoB;iBAC7B;gBACD,mBAAmB,EAAE;oBACnB,OAAO,EAAE,MAAM,CAAC;wBACd,SAAS,EAAE,SAAS;wBACpB,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO;qBAC7C,CAAC;oBACF,MAAM,EAAE,oBAAoB;iBAC7B;aACF;SACF;QACD,aAAa,EAAE;YACb,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc;gBAChD,MAAM,EAAE,oBAAoB;aAC7B;YACD,WAAW,EACT,gFAAgF;YAClF,EAAE,EAAE;gBACF,mBAAmB,EAAE;oBACnB,OAAO,EAAE,MAAM,CAAC;wBACd,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO;qBAC7C,CAAC;oBACF,MAAM,EAAE,uBAAuB;iBAChC;aACF;SACF;KACF;CACF,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { setup, fromObservable, fromPromise, assign, enqueueActions, emit } from \"xstate\";\nimport type { UserChatMessage } from \"../chat/UserChatMessage.ts\";\nimport { AgentConversation } from \"./AgentConversation.ts\";\nimport type { ConversationExchange } from \"./reduceChatEvents.ts\";\nimport type { SonarV4Config } from \"../../../common/Config.ts\";\nimport { createConversation, createConversationCodec } from \"./methods/createConversation.ts\";\nimport * as z from \"zod/mini\";\nimport { retrieveConversation } from \"./methods/retrieveConversation.ts\";\nimport { retrieveConversationHistory } from \"./methods/retrieveConversationHistory.ts\";\nimport { createExchangeRun } from \"./methods/createExchangeRun.ts\";\nimport { stopExchangeRun } from \"./methods/stopExchangeRun.ts\";\nimport { streamRunEvents } from \"./methods/streamRunEvents.ts\";\nimport { fromTextEventStream } from \"../../../common/fromTextEventStream.ts\";\nimport { map, timeout } from \"rxjs\";\nimport { chatEventCodec, type ChatEvent } from \"../chat/chatEventSchema.ts\";\nimport type { HttpError } from \"../../../common/HttpError.ts\";\n\nexport type ConversationMachineInput = {\n  conversationId: string | null;\n};\n\nexport type ConversationMachineContext = {\n  conversationId: string | null;\n  conversationSnapshot?: ConversationExchange[];\n  currentRunId?: string | null;\n  lastError?: unknown;\n  messageAttempt?: UserChatMessage;\n  receivedEndOfStream: boolean;\n  streamRetryCount: number;\n  userRequestedStop: boolean;\n};\nexport type ConversationMachineEvents =\n  | { type: \"STOP_RUN\" }\n  | { type: \"SUBMIT_USER_MESSAGE\"; message: UserChatMessage }\n  | { type: \"RETRY_MESSAGE\" }\n  | { type: \"REFRESH_HISTORY\" };\n\nexport type ConversationMachineEmitted =\n  | {\n      type: \"apiError\";\n      error: HttpError;\n    }\n  | {\n      type: \"chatEventReceived\";\n      chatEvent: ChatEvent;\n    }\n  | {\n      type: \"conversationCreated\";\n      id: string;\n    };\n\nconst RETRIEVE_STATE_TIMEOUT_MS = 10_000;\nconst STREAM_IDLE_TIMEOUT_MS = 60_000;\nconst STREAM_RETRY_DELAY_MS = 2500;\nexport const MAX_STREAM_RETRIES = 5;\nconst STREAM_STATE_RESET = {\n  receivedEndOfStream: false,\n  streamRetryCount: 0,\n  userRequestedStop: false,\n};\n\nexport const createConversationMachine = (config: SonarV4Config) =>\n  setup({\n    actors: {\n      createConversation: fromPromise(\n        ({ input }: { input: z.input<typeof createConversationCodec> }) =>\n          createConversation(config)(input).then((result) => {\n            if (result.isErr()) {\n              throw result.error;\n            }\n            return result.value;\n          }),\n      ),\n      retrieveConversation: fromPromise(\n        ({ input, signal }: { input: { conversationId: string }; signal: AbortSignal }) =>\n          retrieveConversation(config)(input.conversationId, { signal }).then((result) => {\n            if (result.isErr()) {\n              throw result.error;\n            }\n            return result.value;\n          }),\n      ),\n      retrieveHistory: fromPromise(\n        ({ input, signal }: { input: { conversationId: string }; signal: AbortSignal }) =>\n          retrieveConversationHistory(config)(input.conversationId, { signal }).then((result) => {\n            if (result.isErr()) {\n              throw result.error;\n            }\n            return result.value;\n          }),\n      ),\n      retrieveState: fromPromise(\n        async ({ input, signal }: { input: { conversationId: string }; signal: AbortSignal }) => {\n          const timeoutSignal = AbortSignal.any([\n            signal,\n            AbortSignal.timeout(RETRIEVE_STATE_TIMEOUT_MS),\n          ]);\n          const [conversationResult, historyResult] = await Promise.all([\n            retrieveConversation(config)(input.conversationId, { signal: timeoutSignal }),\n            retrieveConversationHistory(config)(input.conversationId, { signal: timeoutSignal }),\n          ]);\n\n          if (conversationResult.isErr()) {\n            throw conversationResult.error;\n          }\n\n          if (historyResult.isErr()) {\n            throw historyResult.error;\n          }\n\n          return {\n            conversation: conversationResult.value,\n            history: historyResult.value,\n          };\n        },\n      ),\n      startRun: fromPromise(\n        ({ input }: { input: { conversationId: string; message: UserChatMessage } }) =>\n          createExchangeRun(config)(input.conversationId, { message: input.message }).then(\n            (result) => {\n              if (result.isErr()) {\n                throw result.error;\n              }\n              return result.value;\n            },\n          ),\n      ),\n      stopRun: fromPromise(({ input }: { input: { conversationId: string; runId: string } }) =>\n        stopExchangeRun(config)(input.conversationId, input.runId).then((result) => {\n          if (result.isErr()) {\n            throw result.error;\n          }\n          return result.value;\n        }),\n      ),\n      streamRun: fromObservable(({ input }: { input: { conversationId: string; runId: string } }) =>\n        fromTextEventStream(({ signal }) =>\n          streamRunEvents(config)(\n            { conversationId: input.conversationId, runId: input.runId },\n            { signal },\n          ),\n        ).pipe(\n          map((event) =>\n            z.decode(chatEventCodec, JSON.parse(event.data) as z.input<typeof chatEventCodec>),\n          ),\n          timeout({ each: STREAM_IDLE_TIMEOUT_MS }),\n        ),\n      ),\n    },\n    delays: {\n      STREAM_RETRY_DELAY: STREAM_RETRY_DELAY_MS,\n    },\n    types: {\n      context: {} as ConversationMachineContext,\n      emitted: {} as ConversationMachineEmitted,\n      events: {} as ConversationMachineEvents,\n      input: {} as ConversationMachineInput,\n    },\n  }).createMachine({\n    context: ({ input }) => ({\n      conversationId: input.conversationId,\n      conversationSnapshot: undefined,\n      currentRunId: undefined,\n      messageAttempt: undefined,\n      ...STREAM_STATE_RESET,\n    }),\n    id: \"agentConversation\",\n    initial: \"uninitialized\",\n    states: {\n      creating_conversation: {\n        description: \"Creating a new conversation object\",\n        invoke: {\n          input: ({ context }) => ({\n            message: context.messageAttempt!,\n          }),\n          onDone: [\n            {\n              actions: [\n                assign({\n                  conversationId: ({ event }) => event.output.id,\n                  // Add the message attempt into the snapshot manually\n                  conversationSnapshot: ({ context, event }) => [\n                    ...(context.conversationSnapshot ?? []),\n                    {\n                      id: event.output.currentRunId!,\n                      messages: new Map(),\n                      submittedUserMessage: context.messageAttempt,\n                      toolCalls: new Map(),\n                    } satisfies ConversationExchange,\n                  ],\n                  currentRunId: ({ event }) => event.output.currentRunId,\n                  messageAttempt: undefined,\n                  ...STREAM_STATE_RESET,\n                }),\n                emit(({ event }) => ({\n                  id: event.output.id,\n                  type: \"conversationCreated\",\n                })),\n              ],\n              // Don't transition to streaming in a situation where we didn't receive a currentRunId\n              guard: ({ event }) => !!event.output.currentRunId,\n              target: \"streaming\",\n            },\n            /**\n             * Fall back to the retrieving_history state if we didn't receive a currentRunId\n             */\n            {\n              actions: assign({\n                conversationId: ({ event }) => event.output.id,\n                currentRunId: ({ event }) => event.output.currentRunId,\n                messageAttempt: undefined,\n              }),\n              target: \"retrieving_history\",\n            },\n          ],\n          onError: {\n            actions: emit(({ event }) => ({\n              error: event.error as HttpError,\n              type: \"apiError\",\n            })),\n            target: \"uninitialized\",\n          },\n          src: \"createConversation\",\n        },\n      },\n      idle: {\n        description: \"Waiting for user input to start the next reply stream\",\n        on: {\n          REFRESH_HISTORY: {\n            target: \"retrieving_history\",\n          },\n          SUBMIT_USER_MESSAGE: {\n            actions: assign({\n              messageAttempt: ({ event }) => event.message,\n            }),\n            target: \"submitting_message\",\n          },\n        },\n      },\n      retrieve_history_failed: {\n        on: {\n          REFRESH_HISTORY: {\n            actions: assign({ lastError: undefined }),\n            target: \"retrieving_history\",\n          },\n        },\n      },\n      retrieving_history: {\n        description: \"Retrieving latest copy of conversation history\",\n        invoke: {\n          input: ({ context }) => ({ conversationId: context.conversationId! }),\n          onDone: [\n            {\n              actions: assign({\n                conversationSnapshot: ({ context, event }) =>\n                  AgentConversation.reduceChatEvents(\n                    context.conversationSnapshot,\n                    event.output.history.data,\n                  ),\n                currentRunId: ({ event }) => event.output.conversation.currentRunId,\n                receivedEndOfStream: false,\n              }),\n              guard: ({ event }) => !!event.output.conversation.currentRunId,\n              target: \"streaming\",\n            },\n            {\n              actions: assign({\n                conversationSnapshot: ({ context, event }) =>\n                  AgentConversation.reduceChatEvents(\n                    context.conversationSnapshot,\n                    event.output.history.data,\n                  ),\n                currentRunId: ({ event }) => event.output.conversation.currentRunId,\n                ...STREAM_STATE_RESET,\n              }),\n              target: \"idle\",\n            },\n          ],\n          onError: [\n            {\n              guard: ({ event }) => {\n                const e = event.error as Error & { code?: string };\n                if (e?.name === \"TimeoutError\") return false;\n                return (\n                  e?.name === \"AbortError\" ||\n                  e?.message?.includes(\"aborted\") ||\n                  e?.code === \"ABORT_ERR\"\n                );\n              },\n              target: \"uninitialized\",\n            },\n            {\n              guard: ({ context }) => context.streamRetryCount > 0,\n              target: \"stream_interrupted\",\n            },\n            {\n              actions: [\n                assign({\n                  lastError: ({ event }) => event.error,\n                }),\n                emit(({ event }) => ({\n                  error: event.error as HttpError,\n                  type: \"apiError\",\n                })),\n              ],\n              target: \"retrieve_history_failed\",\n            },\n          ],\n          src: \"retrieveState\",\n        },\n      },\n      stream_interrupted: {\n        after: {\n          STREAM_RETRY_DELAY: [\n            {\n              actions: assign({\n                streamRetryCount: ({ context }) => context.streamRetryCount + 1,\n              }),\n              guard: ({ context }) => context.streamRetryCount < MAX_STREAM_RETRIES,\n              target: \"retrieving_history\",\n            },\n            {\n              target: \"stream_recovery_failed\",\n            },\n          ],\n        },\n      },\n      stream_recovery_failed: {\n        on: {\n          REFRESH_HISTORY: {\n            actions: assign({ lastError: undefined, streamRetryCount: 0 }),\n            target: \"retrieving_history\",\n          },\n        },\n      },\n      streaming: {\n        description: \"Manages monitoring and canceling the current run\",\n        initial: \"monitoring_replies\",\n        invoke: {\n          input: ({ context }) => {\n            if (!context.currentRunId) {\n              throw new Error(\"Attempted to stream without a runId\");\n            }\n            return {\n              conversationId: context.conversationId!,\n              runId: context.currentRunId,\n            };\n          },\n          onDone: [\n            {\n              actions: assign({ currentRunId: null, userRequestedStop: false }),\n              guard: ({ context }) => !!context.receivedEndOfStream || !!context.userRequestedStop,\n              target: \"idle\",\n            },\n            {\n              actions: assign({ currentRunId: null }),\n              target: \"stream_interrupted\",\n            },\n          ],\n          onError: [\n            {\n              actions: assign({ currentRunId: null, userRequestedStop: false }),\n              guard: ({ context }) => !!context.receivedEndOfStream || !!context.userRequestedStop,\n              target: \"idle\",\n            },\n            {\n              actions: assign({ currentRunId: null }),\n              target: \"stream_interrupted\",\n            },\n          ],\n          onSnapshot: {\n            actions: enqueueActions(({ context, enqueue, event }) => {\n              const chatEvent = event.snapshot.context;\n              if (!chatEvent) {\n                return;\n              }\n\n              if (chatEvent.content.chunkType === \"endOfStream\" && !context.receivedEndOfStream) {\n                enqueue.assign({ receivedEndOfStream: true });\n              }\n\n              const nextSnapshot = AgentConversation.reduceChatEvents(\n                context.conversationSnapshot,\n                [chatEvent],\n              );\n\n              enqueue.assign({\n                conversationSnapshot: nextSnapshot,\n              });\n              enqueue.emit({\n                chatEvent,\n                type: \"chatEventReceived\",\n              });\n            }),\n          },\n          src: \"streamRun\",\n        },\n        states: {\n          monitoring_replies: {\n            description: \"Attached to the run event stream\",\n            on: {\n              STOP_RUN: {\n                actions: assign({ userRequestedStop: true }),\n                target: \"stopping\",\n              },\n            },\n          },\n          stopping: {\n            description: \"Background cancelation of the run stream\",\n            invoke: {\n              input: ({ context }) => ({\n                conversationId: context.conversationId!,\n                runId: context.currentRunId!,\n              }),\n              onDone: {\n                actions: assign({ currentRunId: undefined, userRequestedStop: false }),\n                target: \"#agentConversation.idle\",\n              },\n              onError: {\n                actions: assign({ userRequestedStop: false }),\n                target: \"#agentConversation.retrieving_history\",\n              },\n              src: \"stopRun\",\n            },\n          },\n        },\n      },\n      submitting_message: {\n        description: \"Submitting message to start a new reply stream\",\n        invoke: {\n          input: ({ context, event }) => {\n            switch (event.type) {\n              case \"RETRY_MESSAGE\":\n                if (!context.messageAttempt) {\n                  throw new Error(\"Tried to resubmit nonexistent message\");\n                }\n                return {\n                  conversationId: context.conversationId!,\n                  message: context.messageAttempt,\n                };\n              case \"SUBMIT_USER_MESSAGE\":\n                return {\n                  conversationId: context.conversationId!,\n                  message: event.message,\n                };\n            }\n            throw new Error(`Unexpected event type received in submitting_message: ${event.type}`);\n          },\n          onDone: {\n            actions: assign({\n              // Add the message attempt into the snapshot manually\n              conversationSnapshot: ({ context, event }) => [\n                ...(context.conversationSnapshot ?? []),\n                {\n                  id: event.output.currentRunId,\n                  messages: new Map(),\n                  submittedUserMessage: context.messageAttempt || undefined,\n                  toolCalls: new Map(),\n                } satisfies ConversationExchange,\n              ],\n              currentRunId: ({ event }) => event.output.currentRunId,\n              messageAttempt: undefined,\n              ...STREAM_STATE_RESET,\n            }),\n            target: \"streaming\",\n          },\n          onError: {\n            actions: [\n              assign({\n                currentRunId: () => null,\n                lastError: ({ event }) => event.error,\n              }),\n              emit(({ event }) => ({\n                error: event.error as HttpError,\n                type: \"apiError\",\n              })),\n            ],\n            target: \"submitting_message_failed\",\n          },\n          src: \"startRun\",\n        },\n      },\n      submitting_message_failed: {\n        description: \"Failed to submit message\",\n        on: {\n          REFRESH_HISTORY: {\n            actions: assign({ lastError: undefined }),\n            target: \"retrieving_history\",\n          },\n          RETRY_MESSAGE: {\n            actions: assign({ lastError: undefined }),\n            target: \"submitting_message\",\n          },\n          SUBMIT_USER_MESSAGE: {\n            actions: assign({\n              lastError: undefined,\n              messageAttempt: ({ event }) => event.message,\n            }),\n            target: \"submitting_message\",\n          },\n        },\n      },\n      uninitialized: {\n        always: {\n          guard: ({ context }) => !!context.conversationId,\n          target: \"retrieving_history\",\n        },\n        description:\n          \"Entry state that the machine remains in when no conversation has been provided\",\n        on: {\n          SUBMIT_USER_MESSAGE: {\n            actions: assign({\n              messageAttempt: ({ event }) => event.message,\n            }),\n            target: \"creating_conversation\",\n          },\n        },\n      },\n    },\n  });\n\nexport type AgentConversationMachine = ReturnType<typeof createConversationMachine>;\n"]}

package/dist/enterprise/ai/conversations/methods/retrieveConversationHistory.d.ts

@@ -32,12 +32,11 @@ export declare const retrieveConversationHistory: (config: SonarV4Config) => (id
         };
         role: "agent";
     } | {
+        createdAt: import("temporal-polyfill").Temporal.Instant;
         conversationId: string;
         modelName: string;
         modelProviderId: string;
         runId: string;
-        createdAt: import("temporal-polyfill").Temporal.Instant;
-        id: string;
         content: {
             chunkType: "endOfStream";
         };
@@ -94,6 +93,20 @@ export declare const retrieveConversationHistory: (config: SonarV4Config) => (id
             text: string;
         };
         role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            callId: string;
+            chunkType: "sandboxStdout";
+            isFinal: boolean;
+            text: string;
+        };
+        role: "agent";
     } | {
         conversationId: string;
         modelName: string;

package/dist/enterprise/ai/conversations/reduceChatEvents.d.ts

@@ -6,7 +6,7 @@ export type AgentToolCall = {
     request: ChatEventWithChunkType<"toolRequest"> | undefined;
     result: ChatEventWithChunkType<"toolResponse"> | undefined;
 };
-export type ConversationExchangeMessage = ChatEventWithChunkType<"error"> | ChatEventWithChunkType<"model"> | ChatEventWithChunkType<"sandboxProgress"> | ChatEventWithChunkType<"userMessage">;
+export type ConversationExchangeMessage = ChatEventWithChunkType<"error"> | ChatEventWithChunkType<"model"> | ChatEventWithChunkType<"sandboxProgress"> | ChatEventWithChunkType<"sandboxStdout"> | ChatEventWithChunkType<"userMessage">;
 export type ConversationExchange = {
     id: string;
     messages: Map<string, ConversationExchangeMessage>;

package/dist/enterprise/ai/conversations/reduceChatEvents.js

@@ -40,8 +40,11 @@ function applyChatEventToConversationExchange(conversationExchange, chatEvent) {
         case "error":
         case "model":
         case "sandboxProgress":
+        case "sandboxStdout":
         case "userMessage": {
-            conversationExchange.messages.set(chatEvent.id, chatEvent);
+            // TypeScript doesn't narrow outer union via nested content.chunkType; cast is sound here
+            const event = chatEvent;
+            conversationExchange.messages.set(event.id, event);
             break;
         }
         case "toolRequest":

package/dist/enterprise/ai/conversations/reduceChatEvents.js.map

@@ -1 +1 @@
-{"version":3,"file":"reduceChatEvents.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/conversations/reduceChatEvents.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAc,MAAM,UAAU,CAAC;AA0B9C,SAAS,4BAA4B,CACnC,KAAmC,EACnC,SAAoB;IAEpB,IAAI,oBAAoB,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,KAAK,SAAS,CAAC,KAAK,CAAC,CAAC;IAEzF,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC1B,oBAAoB,GAAG;YACrB,EAAE,EAAE,SAAS,CAAC,KAAK;YACnB,QAAQ,EAAE,IAAI,GAAG,EAAE;YACnB,SAAS,EAAE,IAAI,GAAG,EAAE;SACrB,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACjC,oBAAoB,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,oCAAoC,CAAC,oBAAqB,EAAE,SAAS,CAAC,CAAC;AACzE,CAAC;AAED,SAAS,oCAAoC,CAC3C,oBAAiD,EACjD,SAAoB;IAEpB,QAAQ,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;QACpC,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,KAAK,MAAM,CAAC,EAAE,QAAQ,CAAC,IAAI,oBAAoB,CAAC,SAAS,EAAE,CAAC;gBAC1D,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;oBACrB,QAAQ,CAAC,KAAK,GAAG,UAAU,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,MAAM;QACR,CAAC;QACD,KAAK,OAAO,CAAC;QACb,KAAK,OAAO,CAAC;QACb,KAAK,iBAAiB,CAAC;QACvB,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,oBAAoB,CAAC,QAAQ,CAAC,GAAG,CAC/B,SAAS,CAAC,EAAE,EACZ,SAA+C,CAChD,CAAC;YACF,MAAM;QACR,CAAC;QACD,KAAK,aAAa,CAAC;QACnB,KAAK,cAAc,CAAC,CAAC,CAAC;YACpB,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC;YAExC,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBAChD,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,EAAE;oBACzC,EAAE,EAAE,MAAM;oBACV,OAAO,EAAE,SAAS;oBAClB,MAAM,EAAE,SAAS;oBACjB,KAAK,EAAE,SAAS;iBACjB,CAAC,CAAC;YACL,CAAC;YAED,MAAM,QAAQ,GAAG,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC;YAE7D,QAAQ,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;gBACpC,KAAK,aAAa;oBAChB,QAAQ,CAAC,OAAO,GAAG,SAAoE,CAAC;oBACxF,MAAM;gBACR,KAAK,cAAc;oBACjB,QAAQ,CAAC,MAAM,GAAG,SAAqE,CAAC;oBACxF,IAAI,cAAc,IAAI,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;wBACrD,QAAQ,CAAC,KAAK,GAAG,OAAO,CAAC;oBAC3B,CAAC;yBAAM,CAAC;wBACN,QAAQ,CAAC,KAAK,GAAG,SAAS,CAAC;oBAC7B,CAAC;oBACD,MAAM;YACV,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,QAA+B,EAAE,EAAE,UAAuB;IACzF,OAAO,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;QAC7B,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,4BAA4B,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;QAChE,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { create, type Draft } from \"mutative\";\nimport type { ChatEvent, ChatEventWithChunkType } from \"../chat/chatEventSchema.ts\";\nimport type { UserChatMessage } from \"../chat/UserChatMessage.ts\";\n\nexport type AgentToolCall = {\n  id: string;\n  state: \"canceled\" | \"error\" | \"pending\" | \"success\";\n  request: ChatEventWithChunkType<\"toolRequest\"> | undefined;\n  result: ChatEventWithChunkType<\"toolResponse\"> | undefined;\n};\n\nexport type ConversationExchangeMessage =\n  | ChatEventWithChunkType<\"error\">\n  | ChatEventWithChunkType<\"model\">\n  | ChatEventWithChunkType<\"sandboxProgress\">\n  | ChatEventWithChunkType<\"userMessage\">;\n\nexport type ConversationExchange = {\n  id: string;\n  messages: Map<string, ConversationExchangeMessage>;\n  submittedUserMessage?: UserChatMessage;\n  toolCalls: Map<string, AgentToolCall>;\n};\n\nexport type ChatEventReducerState = ConversationExchange[];\n\nfunction applyChatEventToConversation(\n  draft: Draft<ChatEventReducerState>,\n  chatEvent: ChatEvent,\n): void {\n  let conversationExchange = draft.findLast((exchange) => exchange.id === chatEvent.runId);\n\n  if (!conversationExchange) {\n    conversationExchange = {\n      id: chatEvent.runId,\n      messages: new Map(),\n      toolCalls: new Map(),\n    };\n    draft.push(conversationExchange);\n    conversationExchange = draft[draft.length - 1];\n  }\n\n  applyChatEventToConversationExchange(conversationExchange!, chatEvent);\n}\n\nfunction applyChatEventToConversationExchange(\n  conversationExchange: Draft<ConversationExchange>,\n  chatEvent: ChatEvent,\n): void {\n  switch (chatEvent.content.chunkType) {\n    case \"endOfStream\": {\n      for (const [, toolCall] of conversationExchange.toolCalls) {\n        if (!toolCall.result) {\n          toolCall.state = \"canceled\";\n        }\n      }\n      break;\n    }\n    case \"error\":\n    case \"model\":\n    case \"sandboxProgress\":\n    case \"userMessage\": {\n      conversationExchange.messages.set(\n        chatEvent.id,\n        chatEvent as Draft<ConversationExchangeMessage>,\n      );\n      break;\n    }\n    case \"toolRequest\":\n    case \"toolResponse\": {\n      const callId = chatEvent.content.callId;\n\n      if (!conversationExchange.toolCalls.has(callId)) {\n        conversationExchange.toolCalls.set(callId, {\n          id: callId,\n          request: undefined,\n          result: undefined,\n          state: \"pending\",\n        });\n      }\n\n      const toolCall = conversationExchange.toolCalls.get(callId)!;\n\n      switch (chatEvent.content.chunkType) {\n        case \"toolRequest\":\n          toolCall.request = chatEvent as unknown as Draft<ChatEventWithChunkType<\"toolRequest\">>;\n          break;\n        case \"toolResponse\":\n          toolCall.result = chatEvent as unknown as Draft<ChatEventWithChunkType<\"toolResponse\">>;\n          if (\"errorMessage\" in toolCall.result.content.result) {\n            toolCall.state = \"error\";\n          } else {\n            toolCall.state = \"success\";\n          }\n          break;\n      }\n    }\n  }\n}\n\nexport function reduceChatEvents(state: ChatEventReducerState = [], chatEvents: ChatEvent[]) {\n  return create(state, (draft) => {\n    for (const chatEvent of chatEvents) {\n      applyChatEventToConversation(draft, Object.freeze(chatEvent));\n    }\n  });\n}\n"]}
+{"version":3,"file":"reduceChatEvents.js","sourceRoot":"","sources":["../../../../src/enterprise/ai/conversations/reduceChatEvents.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,MAAM,EAAc,MAAM,UAAU,CAAC;AA2B9C,SAAS,4BAA4B,CACnC,KAAmC,EACnC,SAAoB;IAEpB,IAAI,oBAAoB,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,KAAK,SAAS,CAAC,KAAK,CAAC,CAAC;IAEzF,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC1B,oBAAoB,GAAG;YACrB,EAAE,EAAE,SAAS,CAAC,KAAK;YACnB,QAAQ,EAAE,IAAI,GAAG,EAAE;YACnB,SAAS,EAAE,IAAI,GAAG,EAAE;SACrB,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACjC,oBAAoB,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,oCAAoC,CAAC,oBAAqB,EAAE,SAAS,CAAC,CAAC;AACzE,CAAC;AAED,SAAS,oCAAoC,CAC3C,oBAAiD,EACjD,SAAoB;IAEpB,QAAQ,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;QACpC,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,KAAK,MAAM,CAAC,EAAE,QAAQ,CAAC,IAAI,oBAAoB,CAAC,SAAS,EAAE,CAAC;gBAC1D,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;oBACrB,QAAQ,CAAC,KAAK,GAAG,UAAU,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,MAAM;QACR,CAAC;QACD,KAAK,OAAO,CAAC;QACb,KAAK,OAAO,CAAC;QACb,KAAK,iBAAiB,CAAC;QACvB,KAAK,eAAe,CAAC;QACrB,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,yFAAyF;YACzF,MAAM,KAAK,GAAG,SAA+C,CAAC;YAC9D,oBAAoB,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAA2C,CAAC,CAAC;YACzF,MAAM;QACR,CAAC;QACD,KAAK,aAAa,CAAC;QACnB,KAAK,cAAc,CAAC,CAAC,CAAC;YACpB,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC;YAExC,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBAChD,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,EAAE;oBACzC,EAAE,EAAE,MAAM;oBACV,OAAO,EAAE,SAAS;oBAClB,MAAM,EAAE,SAAS;oBACjB,KAAK,EAAE,SAAS;iBACjB,CAAC,CAAC;YACL,CAAC;YAED,MAAM,QAAQ,GAAG,oBAAoB,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC;YAE7D,QAAQ,SAAS,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;gBACpC,KAAK,aAAa;oBAChB,QAAQ,CAAC,OAAO,GAAG,SAAoE,CAAC;oBACxF,MAAM;gBACR,KAAK,cAAc;oBACjB,QAAQ,CAAC,MAAM,GAAG,SAAqE,CAAC;oBACxF,IAAI,cAAc,IAAI,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;wBACrD,QAAQ,CAAC,KAAK,GAAG,OAAO,CAAC;oBAC3B,CAAC;yBAAM,CAAC;wBACN,QAAQ,CAAC,KAAK,GAAG,SAAS,CAAC;oBAC7B,CAAC;oBACD,MAAM;YACV,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,QAA+B,EAAE,EAAE,UAAuB;IACzF,OAAO,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;QAC7B,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,4BAA4B,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;QAChE,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { create, type Draft } from \"mutative\";\nimport type { ChatEvent, ChatEventWithChunkType } from \"../chat/chatEventSchema.ts\";\nimport type { UserChatMessage } from \"../chat/UserChatMessage.ts\";\n\nexport type AgentToolCall = {\n  id: string;\n  state: \"canceled\" | \"error\" | \"pending\" | \"success\";\n  request: ChatEventWithChunkType<\"toolRequest\"> | undefined;\n  result: ChatEventWithChunkType<\"toolResponse\"> | undefined;\n};\n\nexport type ConversationExchangeMessage =\n  | ChatEventWithChunkType<\"error\">\n  | ChatEventWithChunkType<\"model\">\n  | ChatEventWithChunkType<\"sandboxProgress\">\n  | ChatEventWithChunkType<\"sandboxStdout\">\n  | ChatEventWithChunkType<\"userMessage\">;\n\nexport type ConversationExchange = {\n  id: string;\n  messages: Map<string, ConversationExchangeMessage>;\n  submittedUserMessage?: UserChatMessage;\n  toolCalls: Map<string, AgentToolCall>;\n};\n\nexport type ChatEventReducerState = ConversationExchange[];\n\nfunction applyChatEventToConversation(\n  draft: Draft<ChatEventReducerState>,\n  chatEvent: ChatEvent,\n): void {\n  let conversationExchange = draft.findLast((exchange) => exchange.id === chatEvent.runId);\n\n  if (!conversationExchange) {\n    conversationExchange = {\n      id: chatEvent.runId,\n      messages: new Map(),\n      toolCalls: new Map(),\n    };\n    draft.push(conversationExchange);\n    conversationExchange = draft[draft.length - 1];\n  }\n\n  applyChatEventToConversationExchange(conversationExchange!, chatEvent);\n}\n\nfunction applyChatEventToConversationExchange(\n  conversationExchange: Draft<ConversationExchange>,\n  chatEvent: ChatEvent,\n): void {\n  switch (chatEvent.content.chunkType) {\n    case \"endOfStream\": {\n      for (const [, toolCall] of conversationExchange.toolCalls) {\n        if (!toolCall.result) {\n          toolCall.state = \"canceled\";\n        }\n      }\n      break;\n    }\n    case \"error\":\n    case \"model\":\n    case \"sandboxProgress\":\n    case \"sandboxStdout\":\n    case \"userMessage\": {\n      // TypeScript doesn't narrow outer union via nested content.chunkType; cast is sound here\n      const event = chatEvent as Extract<ChatEvent, { id: string }>;\n      conversationExchange.messages.set(event.id, event as Draft<ConversationExchangeMessage>);\n      break;\n    }\n    case \"toolRequest\":\n    case \"toolResponse\": {\n      const callId = chatEvent.content.callId;\n\n      if (!conversationExchange.toolCalls.has(callId)) {\n        conversationExchange.toolCalls.set(callId, {\n          id: callId,\n          request: undefined,\n          result: undefined,\n          state: \"pending\",\n        });\n      }\n\n      const toolCall = conversationExchange.toolCalls.get(callId)!;\n\n      switch (chatEvent.content.chunkType) {\n        case \"toolRequest\":\n          toolCall.request = chatEvent as unknown as Draft<ChatEventWithChunkType<\"toolRequest\">>;\n          break;\n        case \"toolResponse\":\n          toolCall.result = chatEvent as unknown as Draft<ChatEventWithChunkType<\"toolResponse\">>;\n          if (\"errorMessage\" in toolCall.result.content.result) {\n            toolCall.state = \"error\";\n          } else {\n            toolCall.state = \"success\";\n          }\n          break;\n      }\n    }\n  }\n}\n\nexport function reduceChatEvents(state: ChatEventReducerState = [], chatEvents: ChatEvent[]) {\n  return create(state, (draft) => {\n    for (const chatEvent of chatEvents) {\n      applyChatEventToConversation(draft, Object.freeze(chatEvent));\n    }\n  });\n}\n"]}