@botpress/runtime 1.14.0 → 1.14.1

package/dist/library.js

@@ -48,7 +48,7 @@ var init_define_BUILD = __esm({
 var define_PACKAGE_VERSIONS_default;
 var init_define_PACKAGE_VERSIONS = __esm({
   "<define:__PACKAGE_VERSIONS__>"() {
-    define_PACKAGE_VERSIONS_default = { runtime: "1.14.0", adk: "1.14.0", sdk: "5.0.2", llmz: "0.0.37", zai: "2.5.6", cognitive: "0.3.3" };
+    define_PACKAGE_VERSIONS_default = { runtime: "1.14.1", adk: "1.14.1", sdk: "5.0.2", llmz: "0.0.37", zai: "2.5.6", cognitive: "0.3.3" };
   }
 });
 
@@ -8433,6 +8433,7 @@ var init_context_manager = __esm({
 import { randomBytes } from "node:crypto";
 import http from "node:http";
 import https from "node:https";
+import { gunzipSync, inflateSync, brotliDecompressSync } from "node:zlib";
 import { Dispatcher, getGlobalDispatcher, setGlobalDispatcher } from "undici";
 function isBotpressUrl(fullUrl) {
   try {
@@ -8464,6 +8465,31 @@ function extractIds(fullUrl, body) {
   }
   return ids;
 }
+function decompressBody(buffer, contentEncoding) {
+  if (!contentEncoding) {
+    return buffer.toString("utf-8");
+  }
+  const encoding = contentEncoding.toLowerCase().trim();
+  try {
+    switch (encoding) {
+      case "gzip":
+      case "x-gzip":
+        return gunzipSync(buffer).toString("utf-8");
+      case "deflate":
+        return inflateSync(buffer).toString("utf-8");
+      case "br":
+        return brotliDecompressSync(buffer).toString("utf-8");
+      case "identity":
+      case "":
+        return buffer.toString("utf-8");
+      default:
+        return buffer.toString("utf-8");
+    }
+  } catch (error) {
+    console.warn(`Failed to decompress response with encoding "${encoding}":`, error);
+    return buffer.toString("utf-8");
+  }
+}
 function installHttpClientInstrumentation({ injectTraceHeader = true } = {}) {
   const restores = [];
   const genId = (n) => randomBytes(n).toString("hex");
@@ -8498,6 +8524,9 @@ function installHttpClientInstrumentation({ injectTraceHeader = true } = {}) {
         if (options.headers[INSTRUMENTED_HEADER]) {
           return requestFn.apply(this, args);
         }
+        if (isSilentTracing()) {
+          return requestFn.apply(this, args);
+        }
         options.headers[INSTRUMENTED_HEADER] = "true";
         const method = (options.method || "GET").toUpperCase();
         const isBotpress = isBotpressUrl(urlString);
@@ -8513,12 +8542,14 @@ function installHttpClientInstrumentation({ injectTraceHeader = true } = {}) {
         if (method === "POST" && urlString.includes("/v1/chat/actions") && bodyData?.type) {
           actionName = bodyData.type;
         }
+        const isWorkflowStatusCheck = method === "GET" && /\/workflows\/[^/?]+$/.test(urlString);
         const spanName = isBotpress ? "botpress.client" : "http.client";
         const spanAttributes = isBotpress ? {
           "botpress.method": method,
           "botpress.url": urlString,
           "botpress.via": "http",
-          importance: "medium",
+          importance: isWorkflowStatusCheck ? "low" : "medium",
+          ...isWorkflowStatusCheck && { "debug.type": "workflow-status-check" },
           ...ids,
           ...actionName && { "action.name": actionName }
         } : {
@@ -8563,12 +8594,14 @@ function installHttpClientInstrumentation({ injectTraceHeader = true } = {}) {
         req.on("response", (res) => {
           span2.setAttribute(isBotpress ? "botpress.status_code" : "http.status_code", res.statusCode || 0);
           const responseBodyChunks = [];
+          const contentEncoding = res.headers["content-encoding"];
           res.on("data", (chunk) => {
             responseBodyChunks.push(chunk);
           });
           res.on("end", () => {
             if (responseBodyChunks.length > 0) {
-              const bodyString = Buffer.concat(responseBodyChunks).toString("utf-8");
+              const bodyBuffer = Buffer.concat(responseBodyChunks);
+              const bodyString = decompressBody(bodyBuffer, contentEncoding);
               span2.setAttribute(isBotpress ? "botpress.response.body" : "http.response.body", bodyString);
             }
             if ((res.statusCode || 0) >= 400) {
@@ -8641,6 +8674,9 @@ function installHttpClientInstrumentation({ injectTraceHeader = true } = {}) {
             }
           }
         }
+        if (isSilentTracing()) {
+          return this.d.dispatch(opts, handler);
+        }
         const method = (opts.method || "GET").toUpperCase();
         const origin2 = String(opts.origin ?? "");
         const path4 = String(opts.path ?? "");
@@ -8659,12 +8695,14 @@ function installHttpClientInstrumentation({ injectTraceHeader = true } = {}) {
         if (method === "POST" && path4 === "/v1/chat/actions" && bodyData?.type) {
           actionName = bodyData.type;
         }
+        const isWorkflowStatusCheck = method === "GET" && /\/workflows\/[^/?]+$/.test(fullUrl);
         const spanName = isBotpress ? "botpress.client" : "http.client";
         const spanAttributes = isBotpress ? {
           "botpress.method": method,
           "botpress.url": path4,
           "botpress.via": "undici",
-          importance: "medium",
+          importance: isWorkflowStatusCheck ? "low" : "medium",
+          ...isWorkflowStatusCheck && { "debug.type": "workflow-status-check" },
           ...ids,
           ...actionName && { "action.name": actionName }
         } : {
@@ -8715,11 +8753,18 @@ function installHttpClientInstrumentation({ injectTraceHeader = true } = {}) {
         }
         const nextOpts = { ...opts, headers };
         const responseBodyChunks = [];
+        let contentEncoding;
         const wrap = {
           onConnect: handler.onConnect?.bind(handler),
           onUpgrade: handler.onUpgrade?.bind(handler),
           onHeaders: (statusCode, rawHeaders, resume) => {
             span2.setAttribute(isBotpress ? "botpress.status_code" : "http.status_code", statusCode);
+            for (let i = 0; i < rawHeaders.length; i += 2) {
+              if (String(rawHeaders[i]).toLowerCase() === "content-encoding") {
+                contentEncoding = String(rawHeaders[i + 1]);
+                break;
+              }
+            }
             if (statusCode >= 500) {
               span2.setStatus({
                 code: SpanStatusCode.ERROR,
@@ -8750,7 +8795,8 @@ function installHttpClientInstrumentation({ injectTraceHeader = true } = {}) {
           },
           onComplete: (trailers) => {
             if (responseBodyChunks.length > 0) {
-              const bodyString = Buffer.concat(responseBodyChunks).toString("utf-8");
+              const bodyBuffer = Buffer.concat(responseBodyChunks);
+              const bodyString = decompressBody(bodyBuffer, contentEncoding);
               span2.setAttribute(isBotpress ? "botpress.response.body" : "http.response.body", bodyString);
             }
             span2.end();
@@ -13144,7 +13190,7 @@ var require_lodash = __commonJS({
           return value === true || value === false || isObjectLike2(value) && baseGetTag(value) == boolTag;
         }
         var isBuffer2 = nativeIsBuffer || stubFalse;
-        var isDate2 = nodeIsDate ? baseUnary(nodeIsDate) : baseIsDate;
+        var isDate3 = nodeIsDate ? baseUnary(nodeIsDate) : baseIsDate;
         function isElement(value) {
           return isObjectLike2(value) && value.nodeType === 1 && !isPlainObject2(value);
         }
@@ -14279,7 +14325,7 @@ var require_lodash = __commonJS({
         lodash.isArrayLikeObject = isArrayLikeObject;
         lodash.isBoolean = isBoolean2;
         lodash.isBuffer = isBuffer2;
-        lodash.isDate = isDate2;
+        lodash.isDate = isDate3;
         lodash.isElement = isElement;
         lodash.isEmpty = isEmpty;
         lodash.isEqual = isEqual;
@@ -15142,7 +15188,11 @@ var init_spans = __esm({
         "workflow.step.attempt": { type: "number", required: true },
         "workflow.step.output": { type: "json" },
         "workflow.step.max_attempts": { type: "number" },
-        "workflow.step.error": { type: "string" }
+        "workflow.step.error": { type: "string" },
+        // Map-specific attributes
+        "workflow.map.total": { type: "number", description: "Total number of items to process in map" },
+        "workflow.map.concurrency": { type: "number", description: "Concurrency level for map operation" },
+        "workflow.map.item_index": { type: "number", description: "Index of item in map operation" }
       }
     };
     ActionHandlerSpan = {
@@ -15604,7 +15654,14 @@ var init_span_helpers = __esm({
 });
 
 // src/telemetry/tracing.ts
-var import_sdk_trace_base, import_sdk_trace_node, import_resources, import_core2, spanProcessors, environmentInfo, resource, provider, contextManager, tracer;
+import { AsyncLocalStorage as AsyncLocalStorage3 } from "async_hooks";
+function withSilentTracing(fn) {
+  return silentTracingFlag.run(true, fn);
+}
+function isSilentTracing() {
+  return silentTracingFlag.getStore() === true;
+}
+var import_sdk_trace_base, import_sdk_trace_node, import_resources, import_core2, spanProcessors, environmentInfo, resource, provider, contextManager, tracer, silentTracingFlag;
 var init_tracing = __esm({
   "src/telemetry/tracing.ts"() {
     "use strict";
@@ -15652,6 +15709,7 @@ var init_tracing = __esm({
       });
     }
     tracer = trace.getTracer("adk", "1.0.0");
+    silentTracingFlag = getSingleton("__ADK_GLOBAL_SILENT_TRACING", () => new AsyncLocalStorage3());
     for (const sig of ["SIGINT", "SIGTERM"]) {
       process.on(sig, async () => {
         await provider.forceFlush();
@@ -36050,14 +36108,70 @@ var init_workflow_utils = __esm({
   }
 });
 
+// src/primitives/date-serialization.ts
+function isDateMarker(value) {
+  return typeof value === "object" && value !== null && DATE_MARKER in value && typeof value[DATE_MARKER] === "string";
+}
+function isDate2(value) {
+  return value instanceof Date;
+}
+function serializeDates(data) {
+  if (data === null || data === void 0) {
+    return data;
+  }
+  if (isDate2(data)) {
+    return { [DATE_MARKER]: data.toISOString() };
+  }
+  if (Array.isArray(data)) {
+    return data.map((item) => serializeDates(item));
+  }
+  if (typeof data === "object" && data.constructor === Object) {
+    const result = {};
+    for (const [key, value] of Object.entries(data)) {
+      result[key] = serializeDates(value);
+    }
+    return result;
+  }
+  return data;
+}
+function deserializeDates(data) {
+  if (data === null || data === void 0) {
+    return data;
+  }
+  if (isDateMarker(data)) {
+    return new Date(data[DATE_MARKER]);
+  }
+  if (Array.isArray(data)) {
+    return data.map((item) => deserializeDates(item));
+  }
+  if (typeof data === "object" && data.constructor === Object) {
+    const result = {};
+    for (const [key, value] of Object.entries(data)) {
+      result[key] = deserializeDates(value);
+    }
+    return result;
+  }
+  return data;
+}
+var DATE_MARKER;
+var init_date_serialization = __esm({
+  "src/primitives/date-serialization.ts"() {
+    "use strict";
+    init_define_BUILD();
+    init_define_PACKAGE_VERSIONS();
+    DATE_MARKER = "__date__";
+  }
+});
+
 // src/primitives/workflow-step.ts
 import { ulid } from "ulid";
 import assert from "assert";
-import { AsyncLocalStorage as AsyncLocalStorage3 } from "async_hooks";
+import { AsyncLocalStorage as AsyncLocalStorage4 } from "async_hooks";
 import { transforms as transforms2 } from "@botpress/sdk";
 async function _step(name, run, { maxAttempts = DEFAULT_MAX_ATTEMPTS } = {}, {
   spanFunc,
-  stepType
+  stepType,
+  stepMeta
 } = {}) {
   const workflowControlContext = context.get("workflowControlContext");
   workflowControlContext.signal.throwIfAborted();
@@ -36081,14 +36195,19 @@ async function _step(name, run, { maxAttempts = DEFAULT_MAX_ATTEMPTS } = {}, {
   const steps = stepContext?.steps ?? state.value.steps;
   steps[name] ??= {
     attempts: 0,
+    maxAttempts,
     steps: {},
     startedAt: (/* @__PURE__ */ new Date()).toISOString()
   };
+  if (stepMeta) {
+    Object.assign(steps[name], stepMeta);
+    console.log(`[_step] Applied stepMeta to "${name}":`, JSON.stringify(stepMeta), "\u2192 keys now:", Object.keys(steps[name]));
+  }
   if (steps[name].finishedAt) {
     if (steps[name].error) {
       throw new Error(steps[name].error.message);
     }
-    return steps[name].output;
+    return deserializeDates(steps[name].output);
   }
   while (true) {
     let shouldRetry = false;
@@ -36170,7 +36289,7 @@ async function _step(name, run, { maxAttempts = DEFAULT_MAX_ATTEMPTS } = {}, {
       throw createStepSignal();
     }
     steps[name].finishedAt = (/* @__PURE__ */ new Date()).toISOString();
-    steps[name].output = result;
+    steps[name].output = serializeDates(result);
     steps[name].steps = {};
     state.value.revision++;
     await workflowControlContext.ack();
@@ -36218,9 +36337,10 @@ var init_workflow_step = __esm({
     init_adk();
     init_singletons();
     init_workflow_utils();
+    init_date_serialization();
     DEFAULT_MAX_ATTEMPTS = 5;
     MIN_STEP_REMAINING_TIME_MS = 1e4;
-    storage2 = getSingleton("__ADK_GLOBAL_CTX_WORKFLOW_STEP", () => new AsyncLocalStorage3());
+    storage2 = getSingleton("__ADK_GLOBAL_CTX_WORKFLOW_STEP", () => new AsyncLocalStorage4());
     step = (async (name, run, options = {}) => {
       return _step(name, run, options);
     });
@@ -36359,6 +36479,8 @@ var init_workflow_step = __esm({
       {}
     );
     step.map = async (name, items, run, opts = {}) => {
+      const concurrency = opts.concurrency ?? 1;
+      const maxAttempts = opts.maxAttempts ?? DEFAULT_MAX_ATTEMPTS;
       return _step(
         name,
         async () => {
@@ -36370,15 +36492,28 @@ var init_workflow_step = __esm({
                 async () => {
                   return await run(item, { i });
                 },
-                opts
+                opts,
+                {
+                  stepType: "map-item",
+                  spanFunc: (span2) => {
+                    span2.setAttribute("workflow.map.item_index", i);
+                    span2.setAttribute("workflow.map.total", items.length);
+                  }
+                }
               );
             },
-            opts.concurrency ?? 1
+            concurrency
           );
         },
         opts,
         {
-          stepType: "map"
+          stepType: "map",
+          stepMeta: { mapTotal: items.length, mapConcurrency: concurrency },
+          spanFunc: (span2) => {
+            span2.setAttribute("workflow.map.total", items.length);
+            span2.setAttribute("workflow.map.concurrency", concurrency);
+            span2.setAttribute("workflow.step.max_attempts", maxAttempts);
+          }
         }
       );
     };
@@ -36473,7 +36608,7 @@ function startWorkflowCancellationMonitor(props) {
       return;
     }
     try {
-      const { workflow } = await client2.getWorkflow({ id: workflowId });
+      const { workflow } = await withSilentTracing(() => client2.getWorkflow({ id: workflowId }));
       const isTerminated = workflow.status === "cancelled" || workflow.status === "failed" || workflow.status === "timedout";
       if (isTerminated) {
         workflowControlContext.aborted = true;
@@ -36496,6 +36631,7 @@ var init_workflow_cancellation_monitor = __esm({
     "use strict";
     init_define_BUILD();
     init_define_PACKAGE_VERSIONS();
+    init_tracing();
   }
 });
 
@@ -36548,7 +36684,9 @@ var init_workflow_instance = __esm({
         i: z12.number().optional(),
         startedAt: z12.string(),
         finishedAt: z12.string().optional(),
+        maxAttempts: z12.number().optional(),
         steps: z12.record(z12.string(), workflowStepContextSchema).optional(),
+        mapTotal: z12.number().optional(),
         error: z12.object({
           message: z12.string(),
           failedAt: z12.string(),