@mondaydotcomorg/atp-compiler 0.19.12 → 0.19.14

package/__tests__/unit/detector.test.ts

@@ -248,6 +248,36 @@ describe('AsyncIterationDetector', () => {
 			expect(result.batchableParallel).toBe(true);
 		});
 
+		it('should detect batchable Promise.all with api.client.* calls', () => {
+			const code = `
+        const results = await Promise.all([
+          api.client.toolA({ a: 1 }),
+          api.client.toolB({ b: 2 }),
+        ]);
+      `;
+
+			const result = detector.detect(code);
+
+			expect(result.needsTransform).toBe(true);
+			expect(result.patterns).toContain('promise-all');
+			expect(result.batchableParallel).toBe(true);
+		});
+
+		it('should detect batchable Promise.all with mixed atp.llm and api.client calls', () => {
+			const code = `
+        const results = await Promise.all([
+          atp.llm.call({ prompt: 'Q1' }),
+          api.client.myTool({ input: 'data' }),
+        ]);
+      `;
+
+			const result = detector.detect(code);
+
+			expect(result.needsTransform).toBe(true);
+			expect(result.patterns).toContain('promise-all');
+			expect(result.batchableParallel).toBe(true);
+		});
+
 		it('should not mark Promise.all as batchable with complex logic', () => {
 			const code = `
         const results = await Promise.all(

package/dist/index.cjs

@@ -86,6 +86,10 @@ var PAUSABLE_CALL_PATTERNS = [
   {
     namespace: "atp.embedding",
     method: "encode"
+  },
+  {
+    namespace: "api.client",
+    method: "*"
   }
 ];
 var DEFAULT_COMPILER_CONFIG = {
@@ -112,7 +116,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) {
@@ -755,10 +759,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,
@@ -1319,6 +1337,7 @@ function isInIsolateRuntimeFunction(name) {
 __name(isInIsolateRuntimeFunction, "isInIsolateRuntimeFunction");
 
 // src/transformer/promise-transformer.ts
+var TOOL_OPERATION_CALL = "call";
 var PromiseTransformer = class {
   static {
     __name(this, "PromiseTransformer");
@@ -1384,11 +1403,7 @@ var PromiseTransformer = class {
         return t7__namespace.nullLiteral();
       }
       const payloadArg = callNode.arguments[0];
-      return t7__namespace.objectExpression([
-        t7__namespace.objectProperty(t7__namespace.identifier("type"), t7__namespace.stringLiteral(callInfo.type)),
-        t7__namespace.objectProperty(t7__namespace.identifier("operation"), t7__namespace.stringLiteral(callInfo.operation)),
-        t7__namespace.objectProperty(t7__namespace.identifier("payload"), payloadArg && t7__namespace.isExpression(payloadArg) ? payloadArg : t7__namespace.objectExpression([]))
-      ]);
+      return this.buildBatchCallObject(callInfo, payloadArg);
     }));
     const runtimeCall = t7__namespace.awaitExpression(t7__namespace.callExpression(t7__namespace.memberExpression(t7__namespace.identifier("__runtime"), t7__namespace.identifier(RuntimeFunction.BATCH_PARALLEL)), [
       batchCallsArray,
@@ -1398,6 +1413,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__namespace.isExpression(payloadArg) ? payloadArg : t7__namespace.objectExpression([]);
+    if (callInfo.type === "tool") {
+      return t7__namespace.objectExpression([
+        t7__namespace.objectProperty(t7__namespace.identifier("type"), t7__namespace.stringLiteral(callInfo.type)),
+        t7__namespace.objectProperty(t7__namespace.identifier("operation"), t7__namespace.stringLiteral(TOOL_OPERATION_CALL)),
+        t7__namespace.objectProperty(t7__namespace.identifier("payload"), t7__namespace.objectExpression([
+          t7__namespace.objectProperty(t7__namespace.identifier("toolName"), t7__namespace.stringLiteral(callInfo.operation)),
+          t7__namespace.objectProperty(t7__namespace.identifier("input"), payloadExpr)
+        ]))
+      ]);
+    }
+    return t7__namespace.objectExpression([
+      t7__namespace.objectProperty(t7__namespace.identifier("type"), t7__namespace.stringLiteral(callInfo.type)),
+      t7__namespace.objectProperty(t7__namespace.identifier("operation"), t7__namespace.stringLiteral(callInfo.operation)),
+      t7__namespace.objectProperty(t7__namespace.identifier("payload"), payloadExpr)
+    ]);
+  }
   transformToSequential(path, node) {
     const arrayArg = node.arguments[0];
     if (!t7__namespace.isArrayExpression(arrayArg)) {