@dremio/js-sdk 0.43.0 → 0.44.0

package/dist-iife/cloud.js

@@ -23890,6 +23890,9 @@ Set the \`cycles\` parameter to \`"ref"\` to resolve cyclical schemas with defs.
     summary: optional(string2()),
     title: optional(string2())
   });
+  var endOfStreamChunkSchema = object({
+    chunkType: literal("endOfStream")
+  });
   var modelChunkSchema = object({
     chunkType: literal("model"),
     name: string2().check(_trim(), _minLength(1)),
@@ -23928,6 +23931,7 @@ Set the \`cycles\` parameter to \`"ref"\` to resolve cyclical schemas with defs.
   ]);
   var chatEventInputSchema = discriminatedUnion("chunkType", [
     errorChunkSchema,
+    extend2(endOfStreamChunkSchema, chatEventSharedSchema),
     extend2(conversationUpdateChunkSchema, chatEventSharedSchema),
     extend2(modelChunkSchema, chatEventSharedSchema),
     extend2(toolRequestChunkSchema, chatEventSharedSchema),
@@ -23944,6 +23948,10 @@ Set the \`cycles\` parameter to \`"ref"\` to resolve cyclical schemas with defs.
       role: literal("agent")
     }),
     extend2(chatEventOutputSharedSchema, { content: errorChunkSchema, role: literal("agent") }),
+    extend2(chatEventOutputSharedSchema, {
+      content: endOfStreamChunkSchema,
+      role: literal("agent")
+    }),
     extend2(chatEventOutputSharedSchema, { content: modelChunkSchema, role: literal("agent") }),
     extend2(chatEventOutputSharedSchema, {
       content: toolRequestChunkSchema,
@@ -23994,6 +24002,14 @@ Set the \`cycles\` parameter to \`"ref"\` to resolve cyclical schemas with defs.
             },
             role: "agent"
           };
+        case "endOfStream":
+          return {
+            ...sharedProperties,
+            content: {
+              chunkType: "endOfStream"
+            },
+            role: "agent"
+          };
         case "model":
           return {
             ...sharedProperties,
@@ -26042,6 +26058,14 @@ Set the \`cycles\` parameter to \`"ref"\` to resolve cyclical schemas with defs.
   }
   function applyChatEventToConversationExchange(conversationExchange, chatEvent) {
     switch (chatEvent.content.chunkType) {
+      case "endOfStream": {
+        for (const [, toolCall] of conversationExchange.toolCalls) {
+          if (!toolCall.result) {
+            toolCall.state = "canceled";
+          }
+        }
+        break;
+      }
       case "error":
       case "model":
       case "userMessage": {
@@ -30950,6 +30974,7 @@ ${err.message}`);
     },
     types: {
       context: {},
+      emitted: {},
       events: {},
       input: {}
     }
@@ -30971,21 +30996,27 @@ ${err.message}`);
           }),
           onDone: [
             {
-              actions: assign({
-                conversationId: ({ event }) => event.output.id,
-                // Add the message attempt into the snapshot manually
-                conversationSnapshot: ({ context: context2, event }) => [
-                  ...context2.conversationSnapshot ?? [],
-                  {
-                    id: event.output.currentRunId,
-                    messages: /* @__PURE__ */ new Map(),
-                    submittedUserMessage: context2.messageAttempt,
-                    toolCalls: /* @__PURE__ */ new Map()
-                  }
-                ],
-                currentRunId: ({ event }) => event.output.currentRunId,
-                messageAttempt: void 0
-              }),
+              actions: [
+                assign({
+                  conversationId: ({ event }) => event.output.id,
+                  // Add the message attempt into the snapshot manually
+                  conversationSnapshot: ({ context: context2, event }) => [
+                    ...context2.conversationSnapshot ?? [],
+                    {
+                      id: event.output.currentRunId,
+                      messages: /* @__PURE__ */ new Map(),
+                      submittedUserMessage: context2.messageAttempt,
+                      toolCalls: /* @__PURE__ */ new Map()
+                    }
+                  ],
+                  currentRunId: ({ event }) => event.output.currentRunId,
+                  messageAttempt: void 0
+                }),
+                emit(({ event }) => ({
+                  id: event.output.id,
+                  type: "conversationCreated"
+                }))
+              ],
               // Don't transition to streaming in a situation where we didn't receive a currentRunId
               guard: ({ event }) => !!event.output.currentRunId,
               target: "streaming"
@@ -31091,16 +31122,19 @@ ${err.message}`);
             target: "idle"
           },
           onSnapshot: {
-            actions: assign({
-              conversationSnapshot: ({ context: context2, event }) => {
-                const chatEvent = event.snapshot.context;
-                if (!chatEvent) {
-                  return context2.conversationSnapshot;
-                }
-                return AgentConversation.reduceChatEvents(context2.conversationSnapshot, [
-                  chatEvent
-                ]);
+            actions: enqueueActions(({ context: context2, enqueue, event }) => {
+              const chatEvent = event.snapshot.context;
+              if (!chatEvent) {
+                return;
               }
+              const nextSnapshot = AgentConversation.reduceChatEvents(context2.conversationSnapshot, [chatEvent]);
+              enqueue.assign({
+                conversationSnapshot: nextSnapshot
+              });
+              enqueue.emit({
+                chatEvent,
+                type: "chatEventReceived"
+              });
             })
           },
           src: "streamRun"

package/dist-iife/enterprise.js

@@ -24945,6 +24945,9 @@ Set the \`cycles\` parameter to \`"ref"\` to resolve cyclical schemas with defs.
     summary: optional(string2()),
     title: optional(string2())
   });
+  var endOfStreamChunkSchema = object({
+    chunkType: literal("endOfStream")
+  });
   var modelChunkSchema = object({
     chunkType: literal("model"),
     name: string2().check(_trim(), _minLength(1)),
@@ -24983,6 +24986,7 @@ Set the \`cycles\` parameter to \`"ref"\` to resolve cyclical schemas with defs.
   ]);
   var chatEventInputSchema = discriminatedUnion("chunkType", [
     errorChunkSchema,
+    extend2(endOfStreamChunkSchema, chatEventSharedSchema),
     extend2(conversationUpdateChunkSchema, chatEventSharedSchema),
     extend2(modelChunkSchema, chatEventSharedSchema),
     extend2(toolRequestChunkSchema, chatEventSharedSchema),
@@ -24999,6 +25003,10 @@ Set the \`cycles\` parameter to \`"ref"\` to resolve cyclical schemas with defs.
       role: literal("agent")
     }),
     extend2(chatEventOutputSharedSchema, { content: errorChunkSchema, role: literal("agent") }),
+    extend2(chatEventOutputSharedSchema, {
+      content: endOfStreamChunkSchema,
+      role: literal("agent")
+    }),
     extend2(chatEventOutputSharedSchema, { content: modelChunkSchema, role: literal("agent") }),
     extend2(chatEventOutputSharedSchema, {
       content: toolRequestChunkSchema,
@@ -25049,6 +25057,14 @@ Set the \`cycles\` parameter to \`"ref"\` to resolve cyclical schemas with defs.
             },
             role: "agent"
           };
+        case "endOfStream":
+          return {
+            ...sharedProperties,
+            content: {
+              chunkType: "endOfStream"
+            },
+            role: "agent"
+          };
         case "model":
           return {
             ...sharedProperties,
@@ -26748,6 +26764,14 @@ Set the \`cycles\` parameter to \`"ref"\` to resolve cyclical schemas with defs.
   }
   function applyChatEventToConversationExchange(conversationExchange, chatEvent) {
     switch (chatEvent.content.chunkType) {
+      case "endOfStream": {
+        for (const [, toolCall] of conversationExchange.toolCalls) {
+          if (!toolCall.result) {
+            toolCall.state = "canceled";
+          }
+        }
+        break;
+      }
       case "error":
       case "model":
       case "userMessage": {
@@ -31656,6 +31680,7 @@ ${err.message}`);
     },
     types: {
       context: {},
+      emitted: {},
       events: {},
       input: {}
     }
@@ -31677,21 +31702,27 @@ ${err.message}`);
           }),
           onDone: [
             {
-              actions: assign({
-                conversationId: ({ event }) => event.output.id,
-                // Add the message attempt into the snapshot manually
-                conversationSnapshot: ({ context: context2, event }) => [
-                  ...context2.conversationSnapshot ?? [],
-                  {
-                    id: event.output.currentRunId,
-                    messages: /* @__PURE__ */ new Map(),
-                    submittedUserMessage: context2.messageAttempt,
-                    toolCalls: /* @__PURE__ */ new Map()
-                  }
-                ],
-                currentRunId: ({ event }) => event.output.currentRunId,
-                messageAttempt: void 0
-              }),
+              actions: [
+                assign({
+                  conversationId: ({ event }) => event.output.id,
+                  // Add the message attempt into the snapshot manually
+                  conversationSnapshot: ({ context: context2, event }) => [
+                    ...context2.conversationSnapshot ?? [],
+                    {
+                      id: event.output.currentRunId,
+                      messages: /* @__PURE__ */ new Map(),
+                      submittedUserMessage: context2.messageAttempt,
+                      toolCalls: /* @__PURE__ */ new Map()
+                    }
+                  ],
+                  currentRunId: ({ event }) => event.output.currentRunId,
+                  messageAttempt: void 0
+                }),
+                emit(({ event }) => ({
+                  id: event.output.id,
+                  type: "conversationCreated"
+                }))
+              ],
               // Don't transition to streaming in a situation where we didn't receive a currentRunId
               guard: ({ event }) => !!event.output.currentRunId,
               target: "streaming"
@@ -31797,16 +31828,19 @@ ${err.message}`);
             target: "idle"
           },
           onSnapshot: {
-            actions: assign({
-              conversationSnapshot: ({ context: context2, event }) => {
-                const chatEvent = event.snapshot.context;
-                if (!chatEvent) {
-                  return context2.conversationSnapshot;
-                }
-                return AgentConversation.reduceChatEvents(context2.conversationSnapshot, [
-                  chatEvent
-                ]);
+            actions: enqueueActions(({ context: context2, enqueue, event }) => {
+              const chatEvent = event.snapshot.context;
+              if (!chatEvent) {
+                return;
               }
+              const nextSnapshot = AgentConversation.reduceChatEvents(context2.conversationSnapshot, [chatEvent]);
+              enqueue.assign({
+                conversationSnapshot: nextSnapshot
+              });
+              enqueue.emit({
+                chatEvent,
+                type: "chatEventReceived"
+              });
             })
           },
           src: "streamRun"

package/package.json

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