@mondaydotcomorg/atp-compiler 0.19.12 → 0.19.13

package/dist/index.js

@@ -60,6 +60,10 @@ var PAUSABLE_CALL_PATTERNS = [
   {
     namespace: "atp.embedding",
     method: "encode"
+  },
+  {
+    namespace: "api.client",
+    method: "*"
   }
 ];
 var DEFAULT_COMPILER_CONFIG = {
@@ -86,7 +90,7 @@ function isPausableCallExpression(node) {
     return false;
   }
   const fullPath = getMemberExpressionPath(callee);
-  return PAUSABLE_CALL_PATTERNS.some((pattern) => fullPath === `${pattern.namespace}.${pattern.method}`);
+  return PAUSABLE_CALL_PATTERNS.some((pattern) => pattern.method === "*" ? fullPath.startsWith(`${pattern.namespace}.`) : fullPath === `${pattern.namespace}.${pattern.method}`);
 }
 __name(isPausableCallExpression, "isPausableCallExpression");
 function getMemberExpressionPath(node) {
@@ -729,10 +733,24 @@ var BatchParallelDetector = class {
       return null;
     }
     const [namespace, service, method] = parts;
-    if (namespace !== "atp" || !method) {
+    if (!service || !method) {
+      return null;
+    }
+    let type;
+    if (namespace === "atp") {
+      if (![
+        "llm",
+        "approval",
+        "embedding"
+      ].includes(service)) {
+        return null;
+      }
+      type = service;
+    } else if (namespace === "api" && service === "client") {
+      type = "tool";
+    } else {
       return null;
     }
-    const type = service;
     const payload = this.extractPayload(callNode.arguments);
     return {
       type,
@@ -1293,6 +1311,7 @@ function isInIsolateRuntimeFunction(name) {
 __name(isInIsolateRuntimeFunction, "isInIsolateRuntimeFunction");
 
 // src/transformer/promise-transformer.ts
+var TOOL_OPERATION_CALL = "call";
 var PromiseTransformer = class {
   static {
     __name(this, "PromiseTransformer");
@@ -1358,11 +1377,7 @@ var PromiseTransformer = class {
         return t7.nullLiteral();
       }
       const payloadArg = callNode.arguments[0];
-      return t7.objectExpression([
-        t7.objectProperty(t7.identifier("type"), t7.stringLiteral(callInfo.type)),
-        t7.objectProperty(t7.identifier("operation"), t7.stringLiteral(callInfo.operation)),
-        t7.objectProperty(t7.identifier("payload"), payloadArg && t7.isExpression(payloadArg) ? payloadArg : t7.objectExpression([]))
-      ]);
+      return this.buildBatchCallObject(callInfo, payloadArg);
     }));
     const runtimeCall = t7.awaitExpression(t7.callExpression(t7.memberExpression(t7.identifier("__runtime"), t7.identifier(RuntimeFunction.BATCH_PARALLEL)), [
       batchCallsArray,
@@ -1372,6 +1387,28 @@ var PromiseTransformer = class {
     this.transformCount++;
     return true;
   }
+  /**
+  * Builds the AST for a batch call object.
+  * For client tools, wraps payload with toolName and input structure.
+  */
+  buildBatchCallObject(callInfo, payloadArg) {
+    const payloadExpr = payloadArg && t7.isExpression(payloadArg) ? payloadArg : t7.objectExpression([]);
+    if (callInfo.type === "tool") {
+      return t7.objectExpression([
+        t7.objectProperty(t7.identifier("type"), t7.stringLiteral(callInfo.type)),
+        t7.objectProperty(t7.identifier("operation"), t7.stringLiteral(TOOL_OPERATION_CALL)),
+        t7.objectProperty(t7.identifier("payload"), t7.objectExpression([
+          t7.objectProperty(t7.identifier("toolName"), t7.stringLiteral(callInfo.operation)),
+          t7.objectProperty(t7.identifier("input"), payloadExpr)
+        ]))
+      ]);
+    }
+    return t7.objectExpression([
+      t7.objectProperty(t7.identifier("type"), t7.stringLiteral(callInfo.type)),
+      t7.objectProperty(t7.identifier("operation"), t7.stringLiteral(callInfo.operation)),
+      t7.objectProperty(t7.identifier("payload"), payloadExpr)
+    ]);
+  }
   transformToSequential(path, node) {
     const arrayArg = node.arguments[0];
     if (!t7.isArrayExpression(arrayArg)) {