@dremio/js-sdk 0.60.0 → 0.62.0

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

@@ -108,6 +108,21 @@ export declare class AgentConversation implements AgentConversationProperties {
             result: Record<string, unknown>;
         };
         role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            chunkType: "selectedSkills";
+            selectedSkills: {
+                id: string;
+                name: string;
+            }[];
+        };
+        role: "agent";
     } | {
         conversationId: string;
         modelName: string;
@@ -256,6 +271,21 @@ export declare class AgentConversation implements AgentConversationProperties {
             result: Record<string, unknown>;
         };
         role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            chunkType: "selectedSkills";
+            selectedSkills: {
+                id: string;
+                name: string;
+            }[];
+        };
+        role: "agent";
     } | {
         conversationId: string;
         modelName: string;

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

@@ -147,6 +147,21 @@ export declare const createConversationMachine: (config: SonarV4Config) => impor
                 result: Record<string, unknown>;
             };
             role: "agent";
+        } | {
+            conversationId: string;
+            modelName: string;
+            modelProviderId: string;
+            runId: string;
+            createdAt: import("temporal-polyfill").Temporal.Instant;
+            id: string;
+            content: {
+                chunkType: "selectedSkills";
+                selectedSkills: {
+                    id: string;
+                    name: string;
+                }[];
+            };
+            role: "agent";
         } | {
             conversationId: string;
             modelName: string;
@@ -309,6 +324,21 @@ export declare const createConversationMachine: (config: SonarV4Config) => impor
                     result: Record<string, unknown>;
                 };
                 role: "agent";
+            } | {
+                conversationId: string;
+                modelName: string;
+                modelProviderId: string;
+                runId: string;
+                createdAt: import("temporal-polyfill").Temporal.Instant;
+                id: string;
+                content: {
+                    chunkType: "selectedSkills";
+                    selectedSkills: {
+                        id: string;
+                        name: string;
+                    }[];
+                };
+                role: "agent";
             } | {
                 conversationId: string;
                 modelName: string;
@@ -478,6 +508,21 @@ export declare const createConversationMachine: (config: SonarV4Config) => impor
             result: Record<string, unknown>;
         };
         role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            chunkType: "selectedSkills";
+            selectedSkills: {
+                id: string;
+                name: string;
+            }[];
+        };
+        role: "agent";
     } | {
         conversationId: string;
         modelName: string;
@@ -664,6 +709,21 @@ export declare const createConversationMachine: (config: SonarV4Config) => impor
                 result: Record<string, unknown>;
             };
             role: "agent";
+        } | {
+            conversationId: string;
+            modelName: string;
+            modelProviderId: string;
+            runId: string;
+            createdAt: import("temporal-polyfill").Temporal.Instant;
+            id: string;
+            content: {
+                chunkType: "selectedSkills";
+                selectedSkills: {
+                    id: string;
+                    name: string;
+                }[];
+            };
+            role: "agent";
         } | {
             conversationId: string;
             modelName: string;
@@ -830,6 +890,21 @@ export declare const createConversationMachine: (config: SonarV4Config) => impor
                     result: Record<string, unknown>;
                 };
                 role: "agent";
+            } | {
+                conversationId: string;
+                modelName: string;
+                modelProviderId: string;
+                runId: string;
+                createdAt: import("temporal-polyfill").Temporal.Instant;
+                id: string;
+                content: {
+                    chunkType: "selectedSkills";
+                    selectedSkills: {
+                        id: string;
+                        name: string;
+                    }[];
+                };
+                role: "agent";
             } | {
                 conversationId: string;
                 modelName: string;
@@ -1011,6 +1086,21 @@ export declare const createConversationMachine: (config: SonarV4Config) => impor
             result: Record<string, unknown>;
         };
         role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            chunkType: "selectedSkills";
+            selectedSkills: {
+                id: string;
+                name: string;
+            }[];
+        };
+        role: "agent";
     } | {
         conversationId: string;
         modelName: string;

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

@@ -23,9 +23,10 @@ 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 = {
@@ -85,7 +86,7 @@ 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,

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;AAqC5E,MAAM,yBAAyB,GAAG,MAAM,CAAC;AACzC,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,CACF,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 } 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_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        ),\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"]}
+{"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

@@ -80,6 +80,21 @@ export declare const retrieveConversationHistory: (config: SonarV4Config) => (id
             result: Record<string, unknown>;
         };
         role: "agent";
+    } | {
+        conversationId: string;
+        modelName: string;
+        modelProviderId: string;
+        runId: string;
+        createdAt: import("temporal-polyfill").Temporal.Instant;
+        id: string;
+        content: {
+            chunkType: "selectedSkills";
+            selectedSkills: {
+                id: string;
+                name: string;
+            }[];
+        };
+        role: "agent";
     } | {
         conversationId: string;
         modelName: string;

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

@@ -1,4 +1,4 @@
-import type { ChatEvent, ChatEventWithChunkType } from "../chat/chatEventSchema.ts";
+import type { ChatEvent, ChatEventWithChunkType, SelectedSkillInfo } from "../chat/chatEventSchema.ts";
 import type { UserChatMessage } from "../chat/UserChatMessage.ts";
 export type AgentToolCall = {
     id: string;
@@ -7,9 +7,13 @@ export type AgentToolCall = {
     result: ChatEventWithChunkType<"toolResponse"> | undefined;
 };
 export type ConversationExchangeMessage = ChatEventWithChunkType<"error"> | ChatEventWithChunkType<"model"> | ChatEventWithChunkType<"sandboxProgress"> | ChatEventWithChunkType<"sandboxStdout"> | ChatEventWithChunkType<"userMessage">;
+export type ConversationExchangeRunMetadata = {
+    selectedSkills?: SelectedSkillInfo[];
+};
 export type ConversationExchange = {
     id: string;
     messages: Map<string, ConversationExchangeMessage>;
+    runMetadata?: ConversationExchangeRunMetadata;
     submittedUserMessage?: UserChatMessage;
     toolCalls: Map<string, AgentToolCall>;
 };

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

@@ -14,6 +14,10 @@
  * limitations under the License.
  */
 import { create } from "mutative";
+import { Temporal } from "temporal-polyfill";
+// Mutative array drafts do not handle symbol-key reads reliably, so use a non-enumerable
+// string key for reducer bookkeeping while keeping JSON/deep equality output clean.
+const selectedSkillsCursorKey = "__dremioSelectedSkillsCursor";
 function applyChatEventToConversation(draft, chatEvent) {
     let conversationExchange = draft.findLast((exchange) => exchange.id === chatEvent.runId);
     if (!conversationExchange) {
@@ -37,6 +41,10 @@ function applyChatEventToConversationExchange(conversationExchange, chatEvent) {
             }
             break;
         }
+        case "selectedSkills": {
+            applySelectedSkillsMetadata(conversationExchange, chatEvent);
+            break;
+        }
         case "error":
         case "model":
         case "sandboxProgress":
@@ -76,6 +84,91 @@ function applyChatEventToConversationExchange(conversationExchange, chatEvent) {
         }
     }
 }
+function applySelectedSkillsMetadata(conversationExchange, chatEvent) {
+    if (!isSelectedSkillsEventNewer(conversationExchange.runMetadata, chatEvent)) {
+        return;
+    }
+    conversationExchange.runMetadata ??= {};
+    // Clone before attaching reducer bookkeeping so the input event payload stays immutable.
+    const selectedSkills = [...chatEvent.content.selectedSkills];
+    setSelectedSkillsCursor(selectedSkills, {
+        messagePosition: parseMessagePosition(chatEvent.id),
+        updatedAt: chatEvent.createdAt,
+    });
+    conversationExchange.runMetadata.selectedSkills = selectedSkills;
+}
+function isSelectedSkillsEventNewer(runMetadata, chatEvent) {
+    // selectedSkills is singleton run metadata, so unlike messages and tool calls it is not keyed by
+    // event id. Keep newer live-stream metadata when an older history replay is reduced later.
+    const cursor = getSelectedSkillsCursor(runMetadata);
+    if (!cursor) {
+        return true;
+    }
+    const timestampComparison = Temporal.Instant.compare(chatEvent.createdAt, cursor.updatedAt);
+    if (timestampComparison !== 0) {
+        return timestampComparison > 0;
+    }
+    const messagePositionComparison = compareMessagePositions(parseMessagePosition(chatEvent.id), cursor.messagePosition);
+    if (messagePositionComparison !== undefined) {
+        return messagePositionComparison >= 0;
+    }
+    return true;
+}
+function getSelectedSkillsCursor(runMetadata) {
+    return runMetadata?.selectedSkills?.[selectedSkillsCursorKey];
+}
+function setSelectedSkillsCursor(selectedSkills, cursor) {
+    Object.defineProperty(selectedSkills, selectedSkillsCursorKey, {
+        configurable: true,
+        enumerable: false,
+        value: cursor,
+        writable: true,
+    });
+}
+// Run-events message ids are chunkId:messageSubId; split message parts append :partId.
+function parseMessagePosition(messageId) {
+    const parts = messageId.split(":");
+    if (parts.length < 2 || parts.length > 3) {
+        return undefined;
+    }
+    const [chunkId, messageSubId, partId] = parts;
+    if (!chunkId) {
+        return undefined;
+    }
+    const parsedMessageSubId = parseNonNegativeInteger(messageSubId);
+    if (parsedMessageSubId === undefined) {
+        return undefined;
+    }
+    const parsedPartId = partId === undefined ? undefined : parseNonNegativeInteger(partId);
+    if (partId !== undefined && parsedPartId === undefined) {
+        return undefined;
+    }
+    return {
+        chunkId,
+        messageSubId: parsedMessageSubId,
+        partId: parsedPartId,
+    };
+}
+function parseNonNegativeInteger(value) {
+    if (value === undefined || !/^\d+$/.test(value)) {
+        return undefined;
+    }
+    const parsedValue = Number(value);
+    if (!Number.isSafeInteger(parsedValue)) {
+        return undefined;
+    }
+    return parsedValue;
+}
+function compareMessagePositions(incoming, current) {
+    if (!incoming || !current || incoming.chunkId !== current.chunkId) {
+        return undefined;
+    }
+    const messageSubIdComparison = incoming.messageSubId - current.messageSubId;
+    if (messageSubIdComparison !== 0) {
+        return messageSubIdComparison;
+    }
+    return (incoming.partId ?? 0) - (current.partId ?? 0);
+}
 export function reduceChatEvents(state = [], chatEvents) {
     return create(state, (draft) => {
         for (const chatEvent of chatEvents) {

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;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"]}
+{"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;AAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAsB7C,yFAAyF;AACzF,oFAAoF;AACpF,MAAM,uBAAuB,GAAG,8BAA8B,CAAC;AA+B/D,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,gBAAgB,CAAC,CAAC,CAAC;YACtB,2BAA2B,CACzB,oBAAoB,EACpB,SAAqD,CACtD,CAAC;YACF,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,SAAS,2BAA2B,CAClC,oBAAiD,EACjD,SAAmD;IAEnD,IAAI,CAAC,0BAA0B,CAAC,oBAAoB,CAAC,WAAW,EAAE,SAAS,CAAC,EAAE,CAAC;QAC7E,OAAO;IACT,CAAC;IAED,oBAAoB,CAAC,WAAW,KAAK,EAAE,CAAC;IACxC,yFAAyF;IACzF,MAAM,cAAc,GAAG,CAAC,GAAG,SAAS,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;IAC7D,uBAAuB,CAAC,cAAc,EAAE;QACtC,eAAe,EAAE,oBAAoB,CAAC,SAAS,CAAC,EAAE,CAAC;QACnD,SAAS,EAAE,SAAS,CAAC,SAAS;KAC/B,CAAC,CAAC;IACH,oBAAoB,CAAC,WAAW,CAAC,cAAc,GAAG,cAAc,CAAC;AACnE,CAAC;AAED,SAAS,0BAA0B,CACjC,WAA+D,EAC/D,SAAmD;IAEnD,iGAAiG;IACjG,2FAA2F;IAC3F,MAAM,MAAM,GAAG,uBAAuB,CAAC,WAAW,CAAC,CAAC;IACpD,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,mBAAmB,GAAG,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;IAC5F,IAAI,mBAAmB,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO,mBAAmB,GAAG,CAAC,CAAC;IACjC,CAAC;IAED,MAAM,yBAAyB,GAAG,uBAAuB,CACvD,oBAAoB,CAAC,SAAS,CAAC,EAAE,CAAC,EAClC,MAAM,CAAC,eAAe,CACvB,CAAC;IACF,IAAI,yBAAyB,KAAK,SAAS,EAAE,CAAC;QAC5C,OAAO,yBAAyB,IAAI,CAAC,CAAC;IACxC,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,uBAAuB,CAC9B,WAA+D;IAE/D,OAAQ,WAAW,EAAE,cAA8D,EAAE,CACnF,uBAAuB,CACxB,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAC9B,cAAmC,EACnC,MAA4B;IAE5B,MAAM,CAAC,cAAc,CAAC,cAAc,EAAE,uBAAuB,EAAE;QAC7D,YAAY,EAAE,IAAI;QAClB,UAAU,EAAE,KAAK;QACjB,KAAK,EAAE,MAAM;QACb,QAAQ,EAAE,IAAI;KACf,CAAC,CAAC;AACL,CAAC;AAED,uFAAuF;AACvF,SAAS,oBAAoB,CAAC,SAAiB;IAC7C,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACnC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,GAAG,KAAK,CAAC;IAC9C,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,kBAAkB,GAAG,uBAAuB,CAAC,YAAY,CAAC,CAAC;IACjE,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;QACrC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,YAAY,GAAG,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,uBAAuB,CAAC,MAAM,CAAC,CAAC;IACxF,IAAI,MAAM,KAAK,SAAS,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;QACvD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO;QACL,OAAO;QACP,YAAY,EAAE,kBAAkB;QAChC,MAAM,EAAE,YAAY;KACrB,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAAC,KAAyB;IACxD,IAAI,KAAK,KAAK,SAAS,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QAChD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAClC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,WAAW,CAAC,EAAE,CAAC;QACvC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,SAAS,uBAAuB,CAC9B,QAAqC,EACrC,OAA2C;IAE3C,IAAI,CAAC,QAAQ,IAAI,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,KAAK,OAAO,CAAC,OAAO,EAAE,CAAC;QAClE,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,sBAAsB,GAAG,QAAQ,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;IAC5E,IAAI,sBAAsB,KAAK,CAAC,EAAE,CAAC;QACjC,OAAO,sBAAsB,CAAC;IAChC,CAAC;IAED,OAAO,CAAC,QAAQ,CAAC,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC;AACxD,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 { Temporal } from \"temporal-polyfill\";\nimport type {\n  ChatEvent,\n  ChatEventWithChunkType,\n  SelectedSkillInfo,\n} 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\n// Mutative array drafts do not handle symbol-key reads reliably, so use a non-enumerable\n// string key for reducer bookkeeping while keeping JSON/deep equality output clean.\nconst selectedSkillsCursorKey = \"__dremioSelectedSkillsCursor\";\n\ntype SelectedSkillsCursor = {\n  messagePosition: MessagePosition | undefined;\n  updatedAt: Temporal.Instant;\n};\n\ntype SelectedSkillsWithCursor = SelectedSkillInfo[] & {\n  [selectedSkillsCursorKey]?: SelectedSkillsCursor;\n};\n\ntype MessagePosition = {\n  chunkId: string;\n  messageSubId: number;\n  partId: number | undefined;\n};\n\nexport type ConversationExchangeRunMetadata = {\n  selectedSkills?: SelectedSkillInfo[];\n};\n\nexport type ConversationExchange = {\n  id: string;\n  messages: Map<string, ConversationExchangeMessage>;\n  runMetadata?: ConversationExchangeRunMetadata;\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 \"selectedSkills\": {\n      applySelectedSkillsMetadata(\n        conversationExchange,\n        chatEvent as ChatEventWithChunkType<\"selectedSkills\">,\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\nfunction applySelectedSkillsMetadata(\n  conversationExchange: Draft<ConversationExchange>,\n  chatEvent: ChatEventWithChunkType<\"selectedSkills\">,\n): void {\n  if (!isSelectedSkillsEventNewer(conversationExchange.runMetadata, chatEvent)) {\n    return;\n  }\n\n  conversationExchange.runMetadata ??= {};\n  // Clone before attaching reducer bookkeeping so the input event payload stays immutable.\n  const selectedSkills = [...chatEvent.content.selectedSkills];\n  setSelectedSkillsCursor(selectedSkills, {\n    messagePosition: parseMessagePosition(chatEvent.id),\n    updatedAt: chatEvent.createdAt,\n  });\n  conversationExchange.runMetadata.selectedSkills = selectedSkills;\n}\n\nfunction isSelectedSkillsEventNewer(\n  runMetadata: Draft<ConversationExchangeRunMetadata> | undefined,\n  chatEvent: ChatEventWithChunkType<\"selectedSkills\">,\n): boolean {\n  // selectedSkills is singleton run metadata, so unlike messages and tool calls it is not keyed by\n  // event id. Keep newer live-stream metadata when an older history replay is reduced later.\n  const cursor = getSelectedSkillsCursor(runMetadata);\n  if (!cursor) {\n    return true;\n  }\n\n  const timestampComparison = Temporal.Instant.compare(chatEvent.createdAt, cursor.updatedAt);\n  if (timestampComparison !== 0) {\n    return timestampComparison > 0;\n  }\n\n  const messagePositionComparison = compareMessagePositions(\n    parseMessagePosition(chatEvent.id),\n    cursor.messagePosition,\n  );\n  if (messagePositionComparison !== undefined) {\n    return messagePositionComparison >= 0;\n  }\n\n  return true;\n}\n\nfunction getSelectedSkillsCursor(\n  runMetadata: Draft<ConversationExchangeRunMetadata> | undefined,\n): Draft<SelectedSkillsCursor> | undefined {\n  return (runMetadata?.selectedSkills as Draft<SelectedSkillsWithCursor> | undefined)?.[\n    selectedSkillsCursorKey\n  ];\n}\n\nfunction setSelectedSkillsCursor(\n  selectedSkills: SelectedSkillInfo[],\n  cursor: SelectedSkillsCursor,\n): void {\n  Object.defineProperty(selectedSkills, selectedSkillsCursorKey, {\n    configurable: true,\n    enumerable: false,\n    value: cursor,\n    writable: true,\n  });\n}\n\n// Run-events message ids are chunkId:messageSubId; split message parts append :partId.\nfunction parseMessagePosition(messageId: string): MessagePosition | undefined {\n  const parts = messageId.split(\":\");\n  if (parts.length < 2 || parts.length > 3) {\n    return undefined;\n  }\n\n  const [chunkId, messageSubId, partId] = parts;\n  if (!chunkId) {\n    return undefined;\n  }\n\n  const parsedMessageSubId = parseNonNegativeInteger(messageSubId);\n  if (parsedMessageSubId === undefined) {\n    return undefined;\n  }\n\n  const parsedPartId = partId === undefined ? undefined : parseNonNegativeInteger(partId);\n  if (partId !== undefined && parsedPartId === undefined) {\n    return undefined;\n  }\n\n  return {\n    chunkId,\n    messageSubId: parsedMessageSubId,\n    partId: parsedPartId,\n  };\n}\n\nfunction parseNonNegativeInteger(value: string | undefined): number | undefined {\n  if (value === undefined || !/^\\d+$/.test(value)) {\n    return undefined;\n  }\n\n  const parsedValue = Number(value);\n  if (!Number.isSafeInteger(parsedValue)) {\n    return undefined;\n  }\n\n  return parsedValue;\n}\n\nfunction compareMessagePositions(\n  incoming: MessagePosition | undefined,\n  current: Draft<MessagePosition> | undefined,\n): number | undefined {\n  if (!incoming || !current || incoming.chunkId !== current.chunkId) {\n    return undefined;\n  }\n\n  const messageSubIdComparison = incoming.messageSubId - current.messageSubId;\n  if (messageSubIdComparison !== 0) {\n    return messageSubIdComparison;\n  }\n\n  return (incoming.partId ?? 0) - (current.partId ?? 0);\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"]}

package/dist/enterprise/ai/index.d.ts

@@ -1,7 +1,7 @@
 import { AgentConversation } from "./conversations/AgentConversation.ts";
 import { ModelProvider } from "./modelProvider/ModelProvider.ts";
-import { type ChatEvent, type ChatEventWithChunkType } from "./chat/chatEventSchema.ts";
-import type { ConversationExchange, ConversationExchangeMessage, AgentToolCall } from "./conversations/reduceChatEvents.ts";
+import { type ChatEvent, type ChatEventWithChunkType, type SelectedSkillInfo } from "./chat/chatEventSchema.ts";
+import type { ConversationExchange, ConversationExchangeMessage, ConversationExchangeRunMetadata, AgentToolCall } from "./conversations/reduceChatEvents.ts";
 import { type AgentConversationMachine } from "./conversations/createConversationMachine.ts";
 export * from "./chat/index.ts";
-export { AgentConversation, ModelProvider, type ChatEvent, type ChatEventWithChunkType, type ConversationExchange, type ConversationExchangeMessage, type AgentToolCall, type AgentConversationMachine, };
+export { AgentConversation, ModelProvider, type ChatEvent, type ChatEventWithChunkType, type SelectedSkillInfo, type ConversationExchange, type ConversationExchangeMessage, type ConversationExchangeRunMetadata, type AgentToolCall, type AgentConversationMachine, };

package/dist/enterprise/ai/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/enterprise/ai/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AACzE,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AACjE,OAAO,EAA+C,MAAM,2BAA2B,CAAC;AAMxF,OAAO,EAAiC,MAAM,8CAA8C,CAAC;AAC7F,cAAc,iBAAiB,CAAC;AAChC,OAAO,EACL,iBAAiB,EACjB,aAAa,GAOd,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 { AgentConversation } from \"./conversations/AgentConversation.ts\";\nimport { ModelProvider } from \"./modelProvider/ModelProvider.ts\";\nimport { type ChatEvent, type ChatEventWithChunkType } from \"./chat/chatEventSchema.ts\";\nimport type {\n  ConversationExchange,\n  ConversationExchangeMessage,\n  AgentToolCall,\n} from \"./conversations/reduceChatEvents.ts\";\nimport { type AgentConversationMachine } from \"./conversations/createConversationMachine.ts\";\nexport * from \"./chat/index.ts\";\nexport {\n  AgentConversation,\n  ModelProvider,\n  type ChatEvent,\n  type ChatEventWithChunkType,\n  type ConversationExchange,\n  type ConversationExchangeMessage,\n  type AgentToolCall,\n  type AgentConversationMachine,\n};\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/enterprise/ai/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AACzE,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AACjE,OAAO,EAIN,MAAM,2BAA2B,CAAC;AAOnC,OAAO,EAAiC,MAAM,8CAA8C,CAAC;AAC7F,cAAc,iBAAiB,CAAC;AAChC,OAAO,EACL,iBAAiB,EACjB,aAAa,GASd,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 { AgentConversation } from \"./conversations/AgentConversation.ts\";\nimport { ModelProvider } from \"./modelProvider/ModelProvider.ts\";\nimport {\n  type ChatEvent,\n  type ChatEventWithChunkType,\n  type SelectedSkillInfo,\n} from \"./chat/chatEventSchema.ts\";\nimport type {\n  ConversationExchange,\n  ConversationExchangeMessage,\n  ConversationExchangeRunMetadata,\n  AgentToolCall,\n} from \"./conversations/reduceChatEvents.ts\";\nimport { type AgentConversationMachine } from \"./conversations/createConversationMachine.ts\";\nexport * from \"./chat/index.ts\";\nexport {\n  AgentConversation,\n  ModelProvider,\n  type ChatEvent,\n  type ChatEventWithChunkType,\n  type SelectedSkillInfo,\n  type ConversationExchange,\n  type ConversationExchangeMessage,\n  type ConversationExchangeRunMetadata,\n  type AgentToolCall,\n  type AgentConversationMachine,\n};\n"]}