@mondaydotcomorg/atp-server 0.20.0 → 0.20.2

package/dist/index.js

@@ -2,7 +2,7 @@ import { log, isReplayMode, setCurrentExecutionId, getAPIResultFromCache, approv
 import fs, { readFile } from 'fs/promises';
 import yaml from 'js-yaml';
 import { createServer as createServer$1 } from 'http';
-import { ProvenanceMode, ToolOperationType, ToolSensitivityLevel, CallbackType, ToolOperation, ExecutionStatus, sanitizeInput, MAX_CODE_SIZE, ExecutionErrorCode } from '@mondaydotcomorg/atp-protocol';
+import { ProvenanceMode, ToolOperationType, ToolSensitivityLevel, createToolStartEvent, createToolEndEvent, CallbackType, ToolOperation, ExecutionStatus, sanitizeInput, MAX_CODE_SIZE, ExecutionErrorCode } from '@mondaydotcomorg/atp-protocol';
 import { getProvenance, createProvenanceProxy, ProvenanceSource, setProvenanceExecutionId, SecurityPolicyEngine, createTrackingRuntime, registerProvenanceMetadata, instrumentCode, captureProvenanceSnapshot, cleanupProvenanceForExecution, clearProvenanceExecutionId, DynamicPolicyRegistry, getProvenanceForPrimitive, computeDigest, verifyProvenanceHints, restoreProvenanceSnapshot, markPrimitiveTainted, ProvenanceMode as ProvenanceMode$1, issueProvenanceToken } from '@mondaydotcomorg/atp-provenance';
 export { ProvenanceMode, ProvenanceSecurityError, ProvenanceSource, SecurityPolicyEngine, auditSensitiveAccess, blockLLMRecipients, blockLLMRecipientsWithApproval, canRead, captureProvenanceState, cleanupProvenanceForExecution, clearProvenanceExecutionId, createCustomPolicy, createProvenanceProxy, createTrackingRuntime, getAllProvenance, getBuiltInPolicies, getBuiltInPoliciesWithApproval, getProvenance, getProvenanceForPrimitive, hasProvenance, instrumentCode, isPrimitiveTainted, markPrimitiveTainted, preventDataExfiltration, preventDataExfiltrationWithApproval, requireUserOrigin, requireUserOriginWithApproval, restoreProvenanceState, setProvenanceExecutionId } from '@mondaydotcomorg/atp-provenance';
 import crypto, { randomBytes, randomUUID } from 'crypto';
@@ -6410,10 +6410,28 @@ function convertOperation(path, method, operation, spec, baseURL, options, auth)
       url.searchParams.append(key, value);
     }
     try {
-      const response = await fetch(url.toString(), {
-        method: method.toUpperCase(),
-        headers,
-        body: body ? JSON.stringify(body) : void 0
+      let finalUrl = url.toString();
+      let finalMethod = method.toUpperCase();
+      let finalHeaders = {
+        ...headers
+      };
+      let finalBody = body ? JSON.stringify(body) : void 0;
+      if (options.requestTransformer) {
+        const transformed = await options.requestTransformer({
+          url: finalUrl,
+          method: finalMethod,
+          headers: finalHeaders,
+          body
+        });
+        if (transformed.url) finalUrl = transformed.url;
+        if (transformed.method) finalMethod = transformed.method;
+        if (transformed.headers) finalHeaders = transformed.headers;
+        if (transformed.body !== void 0) finalBody = transformed.body;
+      }
+      const response = await fetch(finalUrl, {
+        method: finalMethod,
+        headers: finalHeaders,
+        body: finalBody
       });
       if (!response.ok) {
         const errorText = await response.text();
@@ -7050,6 +7068,24 @@ function reattachProvenanceFromHints(args, hintMap) {
 __name(reattachProvenanceFromHints, "reattachProvenanceFromHints");
 
 // src/executor/sandbox-builder.ts
+function createEventEmitter(eventCallback) {
+  return (eventOrType, data, runId) => {
+    if (!eventCallback) return;
+    const event = typeof eventOrType === "string" ? {
+      type: eventOrType,
+      data,
+      timestamp: Date.now(),
+      runId
+    } : eventOrType;
+    setImmediate(() => {
+      try {
+        eventCallback(event);
+      } catch (error) {
+      }
+    });
+  };
+}
+__name(createEventEmitter, "createEventEmitter");
 var SandboxBuilder = class {
   static {
     __name(this, "SandboxBuilder");
@@ -7408,21 +7444,25 @@ var SandboxBuilder = class {
                 sensitivityLevel: metadata?.sensitivityLevel
               });
             }
+            const emit = createEventEmitter(config.eventCallback);
             const handlerContext = {
               metadata,
-              requestContext: config.requestContext
+              requestContext: config.requestContext,
+              emit
             };
             const toolCallStartTime = Date.now();
             let result;
             let toolCallError;
+            emit(createToolStartEvent(func.name, group.name, input));
             try {
               result = await handler(input, handlerContext);
             } catch (error) {
               toolCallError = error instanceof Error ? error : new Error(String(error));
               throw error;
             } finally {
+              const duration = Date.now() - toolCallStartTime;
+              emit(createToolEndEvent(func.name, group.name, toolCallError ? void 0 : result, duration, !toolCallError, toolCallError?.message));
               if (config.onToolCall) {
-                const duration = Date.now() - toolCallStartTime;
                 config.onToolCall({
                   toolName: func.name,
                   apiGroup: group.name,
@@ -11250,7 +11290,8 @@ async function handleExecute(ctx, executor, stateManager, config, auditSink, ses
       path: ctx.path,
       method: ctx.method
     },
-    onToolCall
+    onToolCall,
+    eventCallback: requestConfig.eventCallback
   };
   let hintMap;
   const prelimExecutionId = crypto.randomUUID();