@dremio/js-sdk 0.59.0 → 0.60.0

package/dist-iife/enterprise.js

@@ -16056,6 +16056,14 @@ var DremioEnterprise = (() => {
   var endOfStreamChunkSchema = object({
     chunkType: literal("endOfStream")
   });
+  var endOfStreamInputSchema = object({
+    chunkType: literal("endOfStream"),
+    conversationId: string2(),
+    createdAt: string2().check(iso_exports.datetime()),
+    modelName: string2(),
+    modelProviderId: string2(),
+    runId: string2()
+  });
   var interruptChunkSchema = object({
     chunkType: literal("interrupt")
   });
@@ -16075,6 +16083,14 @@ var DremioEnterprise = (() => {
     stepName: string2(),
     text: string2()
   });
+  var sandboxStdoutChunkSchema = object({
+    callId: string2().check(_trim(), _minLength(1)),
+    chunkType: literal("sandboxStdout"),
+    isFinal: boolean2(),
+    text: string2()
+  });
+  var sandboxStdoutContentSchema = sandboxStdoutChunkSchema;
+  var sandboxStdoutChatEventSchema = extend2(sandboxStdoutChunkSchema, chatEventSharedSchema);
   var toolPlanSchema = record(string2(), unknown());
   var toolRequestChunkSchema = object({
     arguments: record(string2(), unknown()),
@@ -16112,12 +16128,13 @@ var DremioEnterprise = (() => {
   ]);
   var chatEventInputSchema = discriminatedUnion("chunkType", [
     extend2(errorChunkV2Schema, chatEventSharedSchema),
-    extend2(endOfStreamChunkSchema, chatEventSharedSchema),
+    endOfStreamInputSchema,
     extend2(interruptChunkSchema, chatEventSharedSchema),
     extend2(conversationUpdateChunkSchema, chatEventSharedSchema),
     extend2(jobUpdateChunkSchema, chatEventSharedSchema),
     extend2(modelChunkSchema, chatEventSharedSchema),
     extend2(sandboxProgressChunkSchema, chatEventSharedSchema),
+    sandboxStdoutChatEventSchema,
     extend2(toolRequestChunkSchema, chatEventSharedSchema),
     extend2(toolResponseChunkSchema, chatEventSharedSchema),
     extend2(userMessageChunkSchema, chatEventSharedSchema)
@@ -16126,13 +16143,14 @@ var DremioEnterprise = (() => {
     createdAt: _instanceof(Xn.Instant),
     id: string2()
   });
+  var endOfStreamOutputSchema = omit2(chatEventOutputSharedSchema, { id: true });
   var chatEventOutputSchema = union([
     extend2(chatEventOutputSharedSchema, {
       content: conversationUpdateChunkSchema,
       role: literal("agent")
     }),
     extend2(chatEventOutputSharedSchema, { content: errorChunkV2Schema, role: literal("agent") }),
-    extend2(chatEventOutputSharedSchema, {
+    extend2(endOfStreamOutputSchema, {
       content: endOfStreamChunkSchema,
       role: literal("agent")
     }),
@@ -16149,6 +16167,10 @@ var DremioEnterprise = (() => {
       content: sandboxProgressChunkSchema,
       role: literal("agent")
     }),
+    extend2(chatEventOutputSharedSchema, {
+      content: sandboxStdoutContentSchema,
+      role: literal("agent")
+    }),
     extend2(chatEventOutputSharedSchema, {
       content: toolRequestChunkSchema,
       role: literal("agent")
@@ -16164,6 +16186,17 @@ var DremioEnterprise = (() => {
   ]);
   var chatEventCodec = codec(chatEventInputSchema, chatEventOutputSchema, {
     decode(v2) {
+      if (v2.chunkType === "endOfStream") {
+        return {
+          content: { chunkType: "endOfStream" },
+          conversationId: v2.conversationId,
+          createdAt: Xn.Instant.from(v2.createdAt),
+          modelName: v2.modelName,
+          modelProviderId: v2.modelProviderId,
+          role: "agent",
+          runId: v2.runId
+        };
+      }
       const sharedProperties = {
         conversationId: v2.conversationId,
         createdAt: Xn.Instant.from(v2.createdAt),
@@ -16193,14 +16226,6 @@ var DremioEnterprise = (() => {
             },
             role: "agent"
           };
-        case "endOfStream":
-          return {
-            ...sharedProperties,
-            content: {
-              chunkType: "endOfStream"
-            },
-            role: "agent"
-          };
         case "interrupt":
           return {
             ...sharedProperties,
@@ -16241,6 +16266,17 @@ var DremioEnterprise = (() => {
             },
             role: "agent"
           };
+        case "sandboxStdout":
+          return {
+            ...sharedProperties,
+            content: {
+              callId: v2.callId,
+              chunkType: "sandboxStdout",
+              isFinal: v2.isFinal,
+              text: v2.text
+            },
+            role: "agent"
+          };
         case "toolRequest":
           return {
             ...sharedProperties,
@@ -16277,10 +16313,21 @@ var DremioEnterprise = (() => {
       }
     },
     encode(v2) {
+      if (v2.content.chunkType === "endOfStream") {
+        return {
+          chunkType: "endOfStream",
+          conversationId: v2.conversationId,
+          createdAt: v2.createdAt.toString(),
+          modelName: v2.modelName,
+          modelProviderId: v2.modelProviderId,
+          runId: v2.runId
+        };
+      }
+      const id = v2.id;
       return {
         conversationId: v2.conversationId,
         createdAt: v2.createdAt.toString(),
-        messageId: v2.id,
+        messageId: id,
         modelName: v2.modelName,
         modelProviderId: v2.modelProviderId,
         runId: v2.runId,
@@ -17952,8 +17999,10 @@ var DremioEnterprise = (() => {
       case "error":
       case "model":
       case "sandboxProgress":
+      case "sandboxStdout":
       case "userMessage": {
-        conversationExchange.messages.set(chatEvent.id, chatEvent);
+        const event = chatEvent;
+        conversationExchange.messages.set(event.id, event);
         break;
       }
       case "toolRequest":
@@ -21452,6 +21501,14 @@ ${err.message}`);
   }
 
   // dist/enterprise/ai/conversations/createConversationMachine.js
+  var RETRIEVE_STATE_TIMEOUT_MS = 1e4;
+  var STREAM_RETRY_DELAY_MS = 2500;
+  var MAX_STREAM_RETRIES = 5;
+  var STREAM_STATE_RESET = {
+    receivedEndOfStream: false,
+    streamRetryCount: 0,
+    userRequestedStop: false
+  };
   var createConversationMachine = (config3) => setup({
     actors: {
       createConversation: fromPromise2(({ input }) => createConversation(config3)(input).then((result) => {
@@ -21473,9 +21530,13 @@ ${err.message}`);
         return result.value;
       })),
       retrieveState: fromPromise2(async ({ input, signal }) => {
+        const timeoutSignal = AbortSignal.any([
+          signal,
+          AbortSignal.timeout(RETRIEVE_STATE_TIMEOUT_MS)
+        ]);
         const [conversationResult, historyResult] = await Promise.all([
-          retrieveConversation(config3)(input.conversationId, { signal }),
-          retrieveConversationHistory(config3)(input.conversationId, { signal })
+          retrieveConversation(config3)(input.conversationId, { signal: timeoutSignal }),
+          retrieveConversationHistory(config3)(input.conversationId, { signal: timeoutSignal })
         ]);
         if (conversationResult.isErr()) {
           throw conversationResult.error;
@@ -21502,6 +21563,9 @@ ${err.message}`);
       })),
       streamRun: fromObservable(({ input }) => fromTextEventStream(({ signal }) => streamRunEvents(config3)({ conversationId: input.conversationId, runId: input.runId }, { signal })).pipe(map((event) => decode(chatEventCodec, JSON.parse(event.data)))))
     },
+    delays: {
+      STREAM_RETRY_DELAY: STREAM_RETRY_DELAY_MS
+    },
     types: {
       context: {},
       emitted: {},
@@ -21513,7 +21577,8 @@ ${err.message}`);
       conversationId: input.conversationId,
       conversationSnapshot: void 0,
       currentRunId: void 0,
-      messageAttempt: void 0
+      messageAttempt: void 0,
+      ...STREAM_STATE_RESET
     }),
     id: "agentConversation",
     initial: "uninitialized",
@@ -21540,7 +21605,8 @@ ${err.message}`);
                     }
                   ],
                   currentRunId: ({ event }) => event.output.currentRunId,
-                  messageAttempt: void 0
+                  messageAttempt: void 0,
+                  ...STREAM_STATE_RESET
                 }),
                 emit(({ event }) => ({
                   id: event.output.id,
@@ -21603,7 +21669,8 @@ ${err.message}`);
             {
               actions: assign({
                 conversationSnapshot: ({ context: context2, event }) => AgentConversation.reduceChatEvents(context2.conversationSnapshot, event.output.history.data),
-                currentRunId: ({ event }) => event.output.conversation.currentRunId
+                currentRunId: ({ event }) => event.output.conversation.currentRunId,
+                receivedEndOfStream: false
               }),
               guard: ({ event }) => !!event.output.conversation.currentRunId,
               target: "streaming"
@@ -21611,7 +21678,8 @@ ${err.message}`);
             {
               actions: assign({
                 conversationSnapshot: ({ context: context2, event }) => AgentConversation.reduceChatEvents(context2.conversationSnapshot, event.output.history.data),
-                currentRunId: ({ event }) => event.output.conversation.currentRunId
+                currentRunId: ({ event }) => event.output.conversation.currentRunId,
+                ...STREAM_STATE_RESET
               }),
               target: "idle"
             }
@@ -21620,10 +21688,16 @@ ${err.message}`);
             {
               guard: ({ event }) => {
                 const e2 = event.error;
+                if (e2?.name === "TimeoutError")
+                  return false;
                 return e2?.name === "AbortError" || e2?.message?.includes("aborted") || e2?.code === "ABORT_ERR";
               },
               target: "uninitialized"
             },
+            {
+              guard: ({ context: context2 }) => context2.streamRetryCount > 0,
+              target: "stream_interrupted"
+            },
             {
               actions: [
                 assign({
@@ -21640,6 +21714,30 @@ ${err.message}`);
           src: "retrieveState"
         }
       },
+      stream_interrupted: {
+        after: {
+          STREAM_RETRY_DELAY: [
+            {
+              actions: assign({
+                streamRetryCount: ({ context: context2 }) => context2.streamRetryCount + 1
+              }),
+              guard: ({ context: context2 }) => context2.streamRetryCount < MAX_STREAM_RETRIES,
+              target: "retrieving_history"
+            },
+            {
+              target: "stream_recovery_failed"
+            }
+          ]
+        }
+      },
+      stream_recovery_failed: {
+        on: {
+          REFRESH_HISTORY: {
+            actions: assign({ lastError: void 0, streamRetryCount: 0 }),
+            target: "retrieving_history"
+          }
+        }
+      },
       streaming: {
         description: "Manages monitoring and canceling the current run",
         initial: "monitoring_replies",
@@ -21653,20 +21751,37 @@ ${err.message}`);
               runId: context2.currentRunId
             };
           },
-          onDone: {
-            actions: assign({ currentRunId: null }),
-            target: "idle"
-          },
-          onError: {
-            actions: assign({ currentRunId: null }),
-            target: "idle"
-          },
+          onDone: [
+            {
+              actions: assign({ currentRunId: null, userRequestedStop: false }),
+              guard: ({ context: context2 }) => !!context2.receivedEndOfStream || !!context2.userRequestedStop,
+              target: "idle"
+            },
+            {
+              actions: assign({ currentRunId: null }),
+              target: "stream_interrupted"
+            }
+          ],
+          onError: [
+            {
+              actions: assign({ currentRunId: null, userRequestedStop: false }),
+              guard: ({ context: context2 }) => !!context2.receivedEndOfStream || !!context2.userRequestedStop,
+              target: "idle"
+            },
+            {
+              actions: assign({ currentRunId: null }),
+              target: "stream_interrupted"
+            }
+          ],
           onSnapshot: {
             actions: enqueueActions(({ context: context2, enqueue, event }) => {
               const chatEvent = event.snapshot.context;
               if (!chatEvent) {
                 return;
               }
+              if (chatEvent.content.chunkType === "endOfStream" && !context2.receivedEndOfStream) {
+                enqueue.assign({ receivedEndOfStream: true });
+              }
               const nextSnapshot = AgentConversation.reduceChatEvents(context2.conversationSnapshot, [chatEvent]);
               enqueue.assign({
                 conversationSnapshot: nextSnapshot
@@ -21684,6 +21799,7 @@ ${err.message}`);
             description: "Attached to the run event stream",
             on: {
               STOP_RUN: {
+                actions: assign({ userRequestedStop: true }),
                 target: "stopping"
               }
             }
@@ -21696,10 +21812,11 @@ ${err.message}`);
                 runId: context2.currentRunId
               }),
               onDone: {
-                actions: assign({ currentRunId: void 0 }),
+                actions: assign({ currentRunId: void 0, userRequestedStop: false }),
                 target: "#agentConversation.idle"
               },
               onError: {
+                actions: assign({ userRequestedStop: false }),
                 target: "#agentConversation.retrieving_history"
               },
               src: "stopRun"
@@ -21741,7 +21858,8 @@ ${err.message}`);
                 }
               ],
               currentRunId: ({ event }) => event.output.currentRunId,
-              messageAttempt: void 0
+              messageAttempt: void 0,
+              ...STREAM_STATE_RESET
             }),
             target: "streaming"
           },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dremio/js-sdk",
-  "version": "0.59.0",
+  "version": "0.60.0",
   "description": "JavaScript library for the Dremio API",
   "keywords": [
     "dremio",