@mondaydotcomorg/atp-server 0.20.1 → 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';
@@ -7068,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");
@@ -7426,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,
@@ -11268,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();