@dremio/js-sdk 0.61.0 → 0.62.1

package/dist-iife/cloud.js

@@ -17251,6 +17251,14 @@ var DremioCloud = (() => {
     toolExecutionId: string2(),
     toolName: string2()
   });
+  var selectedSkillInfoSchema = object({
+    id: string2(),
+    name: string2()
+  });
+  var selectedSkillsChunkSchema = object({
+    chunkType: literal("selectedSkills"),
+    selectedSkills: array(selectedSkillInfoSchema)
+  });
   var userMessageChunkSchema = object({
     chunkType: literal("userMessage"),
     text: string2()
@@ -17268,6 +17276,7 @@ var DremioCloud = (() => {
     extend2(conversationUpdateChunkSchema, chatEventSharedSchema),
     extend2(jobUpdateChunkSchema, chatEventSharedSchema),
     extend2(modelChunkSchema, chatEventSharedSchema),
+    extend2(selectedSkillsChunkSchema, chatEventSharedSchema),
     extend2(sandboxProgressChunkSchema, chatEventSharedSchema),
     sandboxStdoutChatEventSchema,
     extend2(toolRequestChunkSchema, chatEventSharedSchema),
@@ -17298,6 +17307,10 @@ var DremioCloud = (() => {
       role: literal("agent")
     }),
     extend2(chatEventOutputSharedSchema, { content: modelChunkSchema, role: literal("agent") }),
+    extend2(chatEventOutputSharedSchema, {
+      content: selectedSkillsChunkSchema,
+      role: literal("agent")
+    }),
     extend2(chatEventOutputSharedSchema, {
       content: sandboxProgressChunkSchema,
       role: literal("agent")
@@ -17391,6 +17404,15 @@ var DremioCloud = (() => {
             },
             role: "agent"
           };
+        case "selectedSkills":
+          return {
+            ...sharedProperties,
+            content: {
+              chunkType: "selectedSkills",
+              selectedSkills: v2.selectedSkills
+            },
+            role: "agent"
+          };
         case "sandboxProgress":
           return {
             ...sharedProperties,
@@ -19457,6 +19479,7 @@ var DremioCloud = (() => {
   var constructorString = Object.prototype.constructor.toString();
 
   // dist/enterprise/ai/conversations/reduceChatEvents.js
+  var selectedSkillsCursorKey = "__dremioSelectedSkillsCursor";
   function applyChatEventToConversation(draft, chatEvent) {
     let conversationExchange = draft.findLast((exchange) => exchange.id === chatEvent.runId);
     if (!conversationExchange) {
@@ -19480,6 +19503,10 @@ var DremioCloud = (() => {
         }
         break;
       }
+      case "selectedSkills": {
+        applySelectedSkillsMetadata(conversationExchange, chatEvent);
+        break;
+      }
       case "error":
       case "model":
       case "sandboxProgress":
@@ -19517,6 +19544,87 @@ var DremioCloud = (() => {
       }
     }
   }
+  function applySelectedSkillsMetadata(conversationExchange, chatEvent) {
+    if (!isSelectedSkillsEventNewer(conversationExchange.runMetadata, chatEvent)) {
+      return;
+    }
+    conversationExchange.runMetadata ??= {};
+    const selectedSkills = [...chatEvent.content.selectedSkills];
+    setSelectedSkillsCursor(selectedSkills, {
+      messagePosition: parseMessagePosition(chatEvent.id),
+      updatedAt: chatEvent.createdAt
+    });
+    conversationExchange.runMetadata.selectedSkills = selectedSkills;
+  }
+  function isSelectedSkillsEventNewer(runMetadata, chatEvent) {
+    const cursor = getSelectedSkillsCursor(runMetadata);
+    if (!cursor) {
+      return true;
+    }
+    const timestampComparison = Xn.Instant.compare(chatEvent.createdAt, cursor.updatedAt);
+    if (timestampComparison !== 0) {
+      return timestampComparison > 0;
+    }
+    const messagePositionComparison = compareMessagePositions(parseMessagePosition(chatEvent.id), cursor.messagePosition);
+    if (messagePositionComparison !== void 0) {
+      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
+    });
+  }
+  function parseMessagePosition(messageId) {
+    const parts = messageId.split(":");
+    if (parts.length < 2 || parts.length > 3) {
+      return void 0;
+    }
+    const [chunkId, messageSubId, partId] = parts;
+    if (!chunkId) {
+      return void 0;
+    }
+    const parsedMessageSubId = parseNonNegativeInteger(messageSubId);
+    if (parsedMessageSubId === void 0) {
+      return void 0;
+    }
+    const parsedPartId = partId === void 0 ? void 0 : parseNonNegativeInteger(partId);
+    if (partId !== void 0 && parsedPartId === void 0) {
+      return void 0;
+    }
+    return {
+      chunkId,
+      messageSubId: parsedMessageSubId,
+      partId: parsedPartId
+    };
+  }
+  function parseNonNegativeInteger(value) {
+    if (value === void 0 || !/^\d+$/.test(value)) {
+      return void 0;
+    }
+    const parsedValue = Number(value);
+    if (!Number.isSafeInteger(parsedValue)) {
+      return void 0;
+    }
+    return parsedValue;
+  }
+  function compareMessagePositions(incoming, current2) {
+    if (!incoming || !current2 || incoming.chunkId !== current2.chunkId) {
+      return void 0;
+    }
+    const messageSubIdComparison = incoming.messageSubId - current2.messageSubId;
+    if (messageSubIdComparison !== 0) {
+      return messageSubIdComparison;
+    }
+    return (incoming.partId ?? 0) - (current2.partId ?? 0);
+  }
   function reduceChatEvents(state = [], chatEvents) {
     return create(state, (draft) => {
       for (const chatEvent of chatEvents) {
@@ -23155,7 +23263,8 @@ ${err.message}`);
               actions: assign({
                 conversationSnapshot: ({ context: context2, event }) => AgentConversation.reduceChatEvents(context2.conversationSnapshot, event.output.history.data),
                 currentRunId: ({ event }) => event.output.conversation.currentRunId,
-                receivedEndOfStream: false
+                receivedEndOfStream: false,
+                streamRetryCount: 0
               }),
               guard: ({ event }) => !!event.output.conversation.currentRunId,
               target: "streaming"

package/dist-iife/enterprise.js

@@ -16170,6 +16170,14 @@ var DremioEnterprise = (() => {
     toolExecutionId: string2(),
     toolName: string2()
   });
+  var selectedSkillInfoSchema = object({
+    id: string2(),
+    name: string2()
+  });
+  var selectedSkillsChunkSchema = object({
+    chunkType: literal("selectedSkills"),
+    selectedSkills: array(selectedSkillInfoSchema)
+  });
   var userMessageChunkSchema = object({
     chunkType: literal("userMessage"),
     text: string2()
@@ -16187,6 +16195,7 @@ var DremioEnterprise = (() => {
     extend2(conversationUpdateChunkSchema, chatEventSharedSchema),
     extend2(jobUpdateChunkSchema, chatEventSharedSchema),
     extend2(modelChunkSchema, chatEventSharedSchema),
+    extend2(selectedSkillsChunkSchema, chatEventSharedSchema),
     extend2(sandboxProgressChunkSchema, chatEventSharedSchema),
     sandboxStdoutChatEventSchema,
     extend2(toolRequestChunkSchema, chatEventSharedSchema),
@@ -16217,6 +16226,10 @@ var DremioEnterprise = (() => {
       role: literal("agent")
     }),
     extend2(chatEventOutputSharedSchema, { content: modelChunkSchema, role: literal("agent") }),
+    extend2(chatEventOutputSharedSchema, {
+      content: selectedSkillsChunkSchema,
+      role: literal("agent")
+    }),
     extend2(chatEventOutputSharedSchema, {
       content: sandboxProgressChunkSchema,
       role: literal("agent")
@@ -16310,6 +16323,15 @@ var DremioEnterprise = (() => {
             },
             role: "agent"
           };
+        case "selectedSkills":
+          return {
+            ...sharedProperties,
+            content: {
+              chunkType: "selectedSkills",
+              selectedSkills: v2.selectedSkills
+            },
+            role: "agent"
+          };
         case "sandboxProgress":
           return {
             ...sharedProperties,
@@ -18027,6 +18049,7 @@ var DremioEnterprise = (() => {
   var constructorString = Object.prototype.constructor.toString();
 
   // dist/enterprise/ai/conversations/reduceChatEvents.js
+  var selectedSkillsCursorKey = "__dremioSelectedSkillsCursor";
   function applyChatEventToConversation(draft, chatEvent) {
     let conversationExchange = draft.findLast((exchange) => exchange.id === chatEvent.runId);
     if (!conversationExchange) {
@@ -18050,6 +18073,10 @@ var DremioEnterprise = (() => {
         }
         break;
       }
+      case "selectedSkills": {
+        applySelectedSkillsMetadata(conversationExchange, chatEvent);
+        break;
+      }
       case "error":
       case "model":
       case "sandboxProgress":
@@ -18087,6 +18114,87 @@ var DremioEnterprise = (() => {
       }
     }
   }
+  function applySelectedSkillsMetadata(conversationExchange, chatEvent) {
+    if (!isSelectedSkillsEventNewer(conversationExchange.runMetadata, chatEvent)) {
+      return;
+    }
+    conversationExchange.runMetadata ??= {};
+    const selectedSkills = [...chatEvent.content.selectedSkills];
+    setSelectedSkillsCursor(selectedSkills, {
+      messagePosition: parseMessagePosition(chatEvent.id),
+      updatedAt: chatEvent.createdAt
+    });
+    conversationExchange.runMetadata.selectedSkills = selectedSkills;
+  }
+  function isSelectedSkillsEventNewer(runMetadata, chatEvent) {
+    const cursor = getSelectedSkillsCursor(runMetadata);
+    if (!cursor) {
+      return true;
+    }
+    const timestampComparison = Xn.Instant.compare(chatEvent.createdAt, cursor.updatedAt);
+    if (timestampComparison !== 0) {
+      return timestampComparison > 0;
+    }
+    const messagePositionComparison = compareMessagePositions(parseMessagePosition(chatEvent.id), cursor.messagePosition);
+    if (messagePositionComparison !== void 0) {
+      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
+    });
+  }
+  function parseMessagePosition(messageId) {
+    const parts = messageId.split(":");
+    if (parts.length < 2 || parts.length > 3) {
+      return void 0;
+    }
+    const [chunkId, messageSubId, partId] = parts;
+    if (!chunkId) {
+      return void 0;
+    }
+    const parsedMessageSubId = parseNonNegativeInteger(messageSubId);
+    if (parsedMessageSubId === void 0) {
+      return void 0;
+    }
+    const parsedPartId = partId === void 0 ? void 0 : parseNonNegativeInteger(partId);
+    if (partId !== void 0 && parsedPartId === void 0) {
+      return void 0;
+    }
+    return {
+      chunkId,
+      messageSubId: parsedMessageSubId,
+      partId: parsedPartId
+    };
+  }
+  function parseNonNegativeInteger(value) {
+    if (value === void 0 || !/^\d+$/.test(value)) {
+      return void 0;
+    }
+    const parsedValue = Number(value);
+    if (!Number.isSafeInteger(parsedValue)) {
+      return void 0;
+    }
+    return parsedValue;
+  }
+  function compareMessagePositions(incoming, current2) {
+    if (!incoming || !current2 || incoming.chunkId !== current2.chunkId) {
+      return void 0;
+    }
+    const messageSubIdComparison = incoming.messageSubId - current2.messageSubId;
+    if (messageSubIdComparison !== 0) {
+      return messageSubIdComparison;
+    }
+    return (incoming.partId ?? 0) - (current2.partId ?? 0);
+  }
   function reduceChatEvents(state = [], chatEvents) {
     return create(state, (draft) => {
       for (const chatEvent of chatEvents) {
@@ -21725,7 +21833,8 @@ ${err.message}`);
               actions: assign({
                 conversationSnapshot: ({ context: context2, event }) => AgentConversation.reduceChatEvents(context2.conversationSnapshot, event.output.history.data),
                 currentRunId: ({ event }) => event.output.conversation.currentRunId,
-                receivedEndOfStream: false
+                receivedEndOfStream: false,
+                streamRetryCount: 0
               }),
               guard: ({ event }) => !!event.output.conversation.currentRunId,
               target: "streaming"

package/package.json

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