@dremio/js-sdk 0.62.1 → 0.63.0

package/dist-iife/enterprise.js

@@ -3323,6 +3323,7 @@ var DremioEnterprise = (() => {
     createConversationPromptSchema: () => createConversationPromptSchema,
     fromPat: () => fromPat,
     fromTextEventStream: () => fromTextEventStream,
+    fromTextEventStreamActivity: () => fromTextEventStreamActivity,
     fromUsernamePassword: () => fromUsernamePassword,
     isToolRequest: () => isToolRequest,
     isToolResult: () => isToolResult
@@ -9842,6 +9843,67 @@ var DremioEnterprise = (() => {
     inst._zod.def = def;
     (_a = inst._zod).onattach ?? (_a.onattach = []);
   });
+  var numericOriginMap = {
+    number: "number",
+    bigint: "bigint",
+    object: "date"
+  };
+  var $ZodCheckLessThan = /* @__PURE__ */ $constructor("$ZodCheckLessThan", (inst, def) => {
+    $ZodCheck.init(inst, def);
+    const origin = numericOriginMap[typeof def.value];
+    inst._zod.onattach.push((inst2) => {
+      const bag = inst2._zod.bag;
+      const curr = (def.inclusive ? bag.maximum : bag.exclusiveMaximum) ?? Number.POSITIVE_INFINITY;
+      if (def.value < curr) {
+        if (def.inclusive)
+          bag.maximum = def.value;
+        else
+          bag.exclusiveMaximum = def.value;
+      }
+    });
+    inst._zod.check = (payload) => {
+      if (def.inclusive ? payload.value <= def.value : payload.value < def.value) {
+        return;
+      }
+      payload.issues.push({
+        origin,
+        code: "too_big",
+        maximum: def.value,
+        input: payload.value,
+        inclusive: def.inclusive,
+        inst,
+        continue: !def.abort
+      });
+    };
+  });
+  var $ZodCheckGreaterThan = /* @__PURE__ */ $constructor("$ZodCheckGreaterThan", (inst, def) => {
+    $ZodCheck.init(inst, def);
+    const origin = numericOriginMap[typeof def.value];
+    inst._zod.onattach.push((inst2) => {
+      const bag = inst2._zod.bag;
+      const curr = (def.inclusive ? bag.minimum : bag.exclusiveMinimum) ?? Number.NEGATIVE_INFINITY;
+      if (def.value > curr) {
+        if (def.inclusive)
+          bag.minimum = def.value;
+        else
+          bag.exclusiveMinimum = def.value;
+      }
+    });
+    inst._zod.check = (payload) => {
+      if (def.inclusive ? payload.value >= def.value : payload.value > def.value) {
+        return;
+      }
+      payload.issues.push({
+        origin,
+        code: "too_small",
+        minimum: def.value,
+        input: payload.value,
+        inclusive: def.inclusive,
+        inst,
+        continue: !def.abort
+      });
+    };
+  });
   var $ZodCheckMinSize = /* @__PURE__ */ $constructor("$ZodCheckMinSize", (inst, def) => {
     var _a;
     $ZodCheck.init(inst, def);
@@ -10946,6 +11008,22 @@ var DremioEnterprise = (() => {
       ...normalizeParams(params)
     });
   }
+  function _lte(value, params) {
+    return new $ZodCheckLessThan({
+      check: "less_than",
+      ...normalizeParams(params),
+      value,
+      inclusive: true
+    });
+  }
+  function _gte(value, params) {
+    return new $ZodCheckGreaterThan({
+      check: "greater_than",
+      ...normalizeParams(params),
+      value,
+      inclusive: true
+    });
+  }
   function _minSize(minimum, params) {
     return new $ZodCheckMinSize({
       check: "min_size",
@@ -13652,6 +13730,16 @@ var DremioEnterprise = (() => {
     });
   }
 
+  // node_modules/.pnpm/rxjs@7.8.2/node_modules/rxjs/dist/esm5/internal/operators/filter.js
+  function filter(predicate, thisArg) {
+    return operate(function(source, subscriber) {
+      var index = 0;
+      source.subscribe(createOperatorSubscriber(subscriber, function(value) {
+        return predicate.call(thisArg, value, index++) && subscriber.next(value);
+      }));
+    });
+  }
+
   // node_modules/.pnpm/rxjs@7.8.2/node_modules/rxjs/dist/esm5/internal/operators/catchError.js
   function catchError(selector) {
     return operate(function(source, subscriber) {
@@ -14183,29 +14271,29 @@ var DremioEnterprise = (() => {
     user: set(string2().check(_trim(), _minLength(1)))
   }));
   function mapParams(params) {
-    const filter = [];
+    const filter2 = [];
     if (params.contains) {
-      filter.push(`*=contains=${JSON.stringify(params.contains)}`);
+      filter2.push(`*=contains=${JSON.stringify(params.contains)}`);
     }
     if (params.queryType?.size) {
-      filter.push(`(${Array.from(params.queryType).map((v2) => `qt=="${v2}"`).join(",")})`);
+      filter2.push(`(${Array.from(params.queryType).map((v2) => `qt=="${v2}"`).join(",")})`);
     }
     if (params.status?.size) {
-      filter.push(`(${Array.from(params.status).map((v2) => `jst=="${v2}"`).join(",")})`);
+      filter2.push(`(${Array.from(params.status).map((v2) => `jst=="${v2}"`).join(",")})`);
     }
     if (params.user?.size) {
-      filter.push(`(${Array.from(params.user).map((v2) => `usr=="${v2}"`).join(",")})`);
+      filter2.push(`(${Array.from(params.user).map((v2) => `usr=="${v2}"`).join(",")})`);
     }
     if (params.dateRange?.start) {
       const { end, start } = params.dateRange;
-      filter.push(`(${[
+      filter2.push(`(${[
         start && `st=gt=${start.epochMilliseconds}`,
         end && `st=lt=${end.epochMilliseconds}`
       ].join(";")})`);
     }
     return {
       detailLevel: "1",
-      filter,
+      filter: filter2,
       order: params.sort ? params.sort.direction === "DESC" ? "DESCENDING" : "ASCENDING" : void 0,
       sort: params.sort ? fieldToV2Mapping[params.sort.field] : void 0
     };
@@ -15973,7 +16061,8 @@ var DremioEnterprise = (() => {
   };
 
   // dist/common/fromTextEventStream.js
-  var fromTextEventStream = (input) => new Observable((subscriber) => {
+  var fromTextEventStream = (input) => fromTextEventStreamActivity(input).pipe(filter((activity) => activity.type === "event"), map((activity) => activity.event));
+  var fromTextEventStreamActivity = (input) => new Observable((subscriber) => {
     const controller = new AbortController();
     const { signal } = controller;
     input({ signal }).then(async (response) => {
@@ -15986,7 +16075,13 @@ var DremioEnterprise = (() => {
       if (!response.body) {
         throw new Error(`Response body was undefined`);
       }
-      return response.body.pipeThrough(new TextDecoderStream()).pipeThrough(new EventSourceParserStream());
+      return response.body.pipeThrough(new TextDecoderStream()).pipeThrough(new EventSourceParserStream({
+        onComment(comment) {
+          if (!signal.aborted) {
+            subscriber.next({ comment, type: "comment" });
+          }
+        }
+      }));
     }).then(async (readableStream) => {
       const reader = readableStream.getReader();
       while (!signal.aborted) {
@@ -15998,7 +16093,7 @@ var DremioEnterprise = (() => {
           subscriber.complete();
           break;
         }
-        subscriber.next(value);
+        subscriber.next({ event: value, type: "event" });
       }
       await reader.cancel();
     }).catch((e2) => {
@@ -16170,6 +16265,10 @@ var DremioEnterprise = (() => {
     toolExecutionId: string2(),
     toolName: string2()
   });
+  var contextUsageChunkSchema = object({
+    chunkType: literal("contextUsage"),
+    contextUsagePercentage: number2().check(_gte(0), _lte(100))
+  });
   var selectedSkillInfoSchema = object({
     id: string2(),
     name: string2()
@@ -16192,6 +16291,7 @@ var DremioEnterprise = (() => {
     extend2(errorChunkV2Schema, chatEventSharedSchema),
     endOfStreamInputSchema,
     extend2(interruptChunkSchema, chatEventSharedSchema),
+    extend2(contextUsageChunkSchema, chatEventSharedSchema),
     extend2(conversationUpdateChunkSchema, chatEventSharedSchema),
     extend2(jobUpdateChunkSchema, chatEventSharedSchema),
     extend2(modelChunkSchema, chatEventSharedSchema),
@@ -16221,6 +16321,10 @@ var DremioEnterprise = (() => {
       content: interruptChunkSchema,
       role: literal("agent")
     }),
+    extend2(chatEventOutputSharedSchema, {
+      content: contextUsageChunkSchema,
+      role: literal("agent")
+    }),
     extend2(chatEventOutputSharedSchema, {
       content: jobUpdateChunkSchema,
       role: literal("agent")
@@ -16273,6 +16377,15 @@ var DremioEnterprise = (() => {
         runId: v2.runId
       };
       switch (v2.chunkType) {
+        case "contextUsage":
+          return {
+            ...sharedProperties,
+            content: {
+              chunkType: "contextUsage",
+              contextUsagePercentage: v2.contextUsagePercentage
+            },
+            role: "agent"
+          };
         case "conversationUpdate":
           return {
             ...sharedProperties,
@@ -21664,7 +21777,7 @@ ${err.message}`);
 
   // dist/enterprise/ai/conversations/createConversationMachine.js
   var RETRIEVE_STATE_TIMEOUT_MS = 1e4;
-  var STREAM_IDLE_TIMEOUT_MS = 6e4;
+  var STREAM_IDLE_TIMEOUT_MS = 15e3;
   var STREAM_RETRY_DELAY_MS = 2500;
   var MAX_STREAM_RETRIES = 5;
   var STREAM_STATE_RESET = {
@@ -21724,7 +21837,7 @@ ${err.message}`);
         }
         return result.value;
       })),
-      streamRun: fromObservable(({ input }) => fromTextEventStream(({ signal }) => streamRunEvents(config3)({ conversationId: input.conversationId, runId: input.runId }, { signal })).pipe(map((event) => decode(chatEventCodec, JSON.parse(event.data))), timeout({ each: STREAM_IDLE_TIMEOUT_MS })))
+      streamRun: fromObservable(({ input }) => fromTextEventStreamActivity(({ signal }) => streamRunEvents(config3)({ conversationId: input.conversationId, runId: input.runId }, { signal })).pipe(timeout({ each: STREAM_IDLE_TIMEOUT_MS }), filter((activity) => activity.type === "event"), map(({ event }) => decode(chatEventCodec, JSON.parse(event.data)))))
     },
     delays: {
       STREAM_RETRY_DELAY: STREAM_RETRY_DELAY_MS

package/package.json

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