@mondaydotcomorg/atp-server 0.20.1 → 0.20.2

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@mondaydotcomorg/atp-server",
-	"version": "0.20.1",
+	"version": "0.20.2",
 	"description": "Server implementation for Agent Tool Protocol",
 	"type": "module",
 	"main": "./dist/index.cjs",
@@ -49,11 +49,11 @@
 		"@babel/parser": "^7.26.0",
 		"@babel/traverse": "^7.26.0",
 		"@babel/types": "^7.26.0",
-		"@mondaydotcomorg/atp-compiler": "0.19.16",
-		"@mondaydotcomorg/atp-protocol": "0.19.14",
-		"@mondaydotcomorg/atp-provenance": "0.19.13",
-		"@mondaydotcomorg/atp-providers": "0.19.14",
-		"@mondaydotcomorg/atp-runtime": "0.19.13",
+		"@mondaydotcomorg/atp-compiler": "0.19.17",
+		"@mondaydotcomorg/atp-protocol": "0.19.15",
+		"@mondaydotcomorg/atp-provenance": "0.19.14",
+		"@mondaydotcomorg/atp-providers": "0.19.15",
+		"@mondaydotcomorg/atp-runtime": "0.19.14",
 		"@opentelemetry/api": "^1.9.0",
 		"@opentelemetry/auto-instrumentations-node": "^0.66.0",
 		"@opentelemetry/core": "^2.2.0",

package/src/controllers/stream.controller.ts

@@ -1,5 +1,5 @@
 import type { IncomingMessage, ServerResponse } from 'node:http';
-import type { ExecutionConfig } from '@mondaydotcomorg/atp-protocol';
+import type { ExecutionConfig, ATPEvent } from '@mondaydotcomorg/atp-protocol';
 import {
 	ExecutionErrorCode,
 	validateExecutionConfig,
@@ -86,6 +86,9 @@ export async function handleExecuteStream(
 			progressCallback: (message: string, fraction: number) => {
 				sendEvent('progress', { message, fraction });
 			},
+			eventCallback: (event: ATPEvent) => {
+				sendEvent(event.type, event.data);
+			},
 			clientServices: request.config?.clientServices,
 			provenanceMode: request.config?.provenanceMode,
 			securityPolicies: request.config?.securityPolicies,

package/src/executor/sandbox-builder.ts

@@ -3,7 +3,10 @@ import type {
 	APIGroupConfig,
 	ClientToolDefinition,
 	ToolCallEvent,
+	ATPEvent,
+	ATPEventType,
 } from '@mondaydotcomorg/atp-protocol';
+import { createToolStartEvent, createToolEndEvent } from '@mondaydotcomorg/atp-protocol';
 import { filterApiGroups } from '../core/request-scope.js';
 import {
 	ToolOperationType,
@@ -41,6 +44,35 @@ import { ReaderPermissions } from '@mondaydotcomorg/atp-server';
 import { getHintMap, reattachProvenanceFromHints } from '../utils/provenance-reattachment.js';
 import { createASTProvenanceChecker } from './ast-provenance-bridge.js';
 
+/**
+ * Creates a non-blocking event emitter for streaming events to clients.
+ * Events are fired asynchronously without blocking execution.
+ */
+function createEventEmitter(
+	eventCallback?: (event: ATPEvent) => void
+): (
+	eventOrType: ATPEvent | ATPEventType | string,
+	data?: unknown,
+	runId?: string
+) => void {
+	return (eventOrType, data, runId) => {
+		if (!eventCallback) return;
+
+		const event: ATPEvent =
+			typeof eventOrType === 'string'
+				? { type: eventOrType, data, timestamp: Date.now(), runId }
+				: eventOrType;
+
+		setImmediate(() => {
+			try {
+				eventCallback(event);
+			} catch (error) {
+				// Silently ignore errors in event callbacks to avoid breaking execution
+			}
+		});
+	};
+}
+
 export class SandboxBuilder {
 	private policyEngine: SecurityPolicyEngine | null = null;
 
@@ -467,22 +499,39 @@ export class SandboxBuilder {
 							});
 						}
 
+						const emit = createEventEmitter(config.eventCallback);
+
 						const handlerContext = {
 							metadata: metadata,
 							requestContext: config.requestContext,
+							emit,
 						};
 						const toolCallStartTime = Date.now();
 						let result: unknown;
 						let toolCallError: Error | undefined;
 
+						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 ? undefined : result,
+									duration,
+									!toolCallError,
+									toolCallError?.message
+								)
+							);
+
 							if (config.onToolCall) {
-								const duration = Date.now() - toolCallStartTime;
 								config.onToolCall({
 									toolName: func.name,
 									apiGroup: group.name,

package/src/handlers/execute.handler.ts

@@ -127,6 +127,7 @@ export async function handleExecute(
 			method: ctx.method,
 		},
 		onToolCall,
+		eventCallback: requestConfig.eventCallback,
 	};
 
 	// Verify provenance hints if provided