@mondaydotcomorg/atp-vercel-sdk 0.20.1 → 0.20.3

package/dist/tools.js

@@ -3,6 +3,7 @@ import { VercelAIATPClient } from './client.js';
 import { createToolsFromATPClient, ToolNames } from '@mondaydotcomorg/atp-client';
 import { ExecutionStatus } from '@mondaydotcomorg/atp-protocol';
 import { tool } from 'ai';
+import { createVercelEventHandler } from './event-adapter.js';
 export async function createATPTools(options) {
     const { defaultExecutionConfig, ...clientOptions } = options;
     const client = new VercelAIATPClient(clientOptions);
@@ -97,4 +98,98 @@ export async function createATPTools(options) {
         tools: vercelTools,
     };
 }
+/**
+ * Creates ATP tools with streaming event support.
+ * Events from tool execution will be forwarded to the provided dataStream.
+ *
+ * @param options - Tool creation options including dataStream for event forwarding
+ * @returns Promise resolving to client and tools with streaming support
+ *
+ * @example
+ * ```typescript
+ * // In your Vercel AI SDK route handler
+ * const dataStream = createUIMessageStream({...});
+ *
+ * const { tools } = await createATPStreamingTools({
+ *   serverUrl: 'http://localhost:3333',
+ *   model: openai('gpt-4'),
+ *   dataStream,
+ * });
+ *
+ * // Use tools in streamText
+ * const result = streamText({
+ *   model: openai('gpt-4'),
+ *   messages,
+ *   tools,
+ * });
+ * ```
+ */
+export async function createATPStreamingTools(options) {
+    const { dataStream, defaultExecutionConfig, ...clientOptions } = options;
+    const client = new VercelAIATPClient(clientOptions);
+    await client.connect();
+    const eventHandler = createVercelEventHandler(dataStream);
+    const underlyingClient = client.getUnderlyingClient();
+    const vercelTools = {};
+    vercelTools.atp_execute_code = tool({
+        description: 'Execute TypeScript code in ATP sandbox with streaming events for thinking, tool execution, and text output',
+        parameters: z.object({
+            code: z.string().describe('TypeScript code to execute in the ATP sandbox'),
+        }),
+        execute: async ({ code }) => {
+            try {
+                const result = await underlyingClient.executeStream(code, defaultExecutionConfig, eventHandler);
+                if (result.status === ExecutionStatus.COMPLETED) {
+                    return {
+                        success: true,
+                        result: result.result,
+                        stats: result.stats,
+                    };
+                }
+                else if (result.status === ExecutionStatus.FAILED) {
+                    return {
+                        success: false,
+                        error: result.error,
+                        stats: result.stats,
+                    };
+                }
+                else {
+                    return {
+                        success: false,
+                        error: 'Execution in unexpected state: ' + result.status,
+                    };
+                }
+            }
+            catch (error) {
+                return {
+                    success: false,
+                    error: error.message || 'Unknown error',
+                };
+            }
+        },
+    });
+    vercelTools.atp_get_type_definitions = tool({
+        description: 'Get TypeScript type definitions for ATP runtime APIs to understand available functions',
+        parameters: z.object({}),
+        execute: async () => {
+            try {
+                const types = client.getTypeDefinitions();
+                return {
+                    success: true,
+                    types,
+                };
+            }
+            catch (error) {
+                return {
+                    success: false,
+                    error: error.message,
+                };
+            }
+        },
+    });
+    return {
+        client,
+        tools: vercelTools,
+    };
+}
 //# sourceMappingURL=tools.js.map

package/dist/tools.js.map

@@ -1 +1 @@
-{"version":3,"file":"tools.js","sourceRoot":"","sources":["../src/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAEhD,OAAO,EAAE,wBAAwB,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AAClF,OAAO,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAChE,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAE1B,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,OAA8B;IAClE,MAAM,EAAE,sBAAsB,EAAE,GAAG,aAAa,EAAE,GAAG,OAAO,CAAC;IAE7D,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,aAAa,CAAC,CAAC;IACpD,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IAEvB,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC;IAExE,MAAM,WAAW,GAAwB,EAAE,CAAC;IAE5C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAChC,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,CAAC,YAAY,EAAE,CAAC;YAC7C,WAAW,CAAC,gBAAgB,GAAG,IAAI,CAAC;gBACnC,WAAW,EACV,OAAO,CAAC,WAAW;oBACnB,iHAAiH;gBAClH,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;oBACpB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,+CAA+C,CAAC;iBAC1E,CAAC;gBACF,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAoB,EAAE,EAAE;oBAC7C,IAAI,CAAC;wBACJ,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,sBAAsB,CAAC,CAAC;wBAElE,IAAI,MAAM,CAAC,MAAM,KAAK,eAAe,CAAC,SAAS,EAAE,CAAC;4BACjD,OAAO;gCACN,OAAO,EAAE,IAAI;gCACb,MAAM,EAAE,MAAM,CAAC,MAAM;gCACrB,KAAK,EAAE,MAAM,CAAC,KAAK;6BACnB,CAAC;wBACH,CAAC;6BAAM,IAAI,MAAM,CAAC,MAAM,KAAK,eAAe,CAAC,MAAM,EAAE,CAAC;4BACrD,OAAO;gCACN,OAAO,EAAE,KAAK;gCACd,KAAK,EAAE,MAAM,CAAC,KAAK;gCACnB,KAAK,EAAE,MAAM,CAAC,KAAK;6BACnB,CAAC;wBACH,CAAC;6BAAM,CAAC;4BACP,OAAO;gCACN,OAAO,EAAE,KAAK;gCACd,KAAK,EAAE,iCAAiC,GAAG,MAAM,CAAC,MAAM;6BACxD,CAAC;wBACH,CAAC;oBACF,CAAC;oBAAC,OAAO,KAAU,EAAE,CAAC;wBACrB,OAAO;4BACN,OAAO,EAAE,KAAK;4BACd,KAAK,EAAE,KAAK,CAAC,OAAO,IAAI,eAAe;yBACvC,CAAC;oBACH,CAAC;gBACF,CAAC;aACD,CAAC,CAAC;QACJ,CAAC;aAAM,CAAC;YACP,MAAM,QAAQ,GAAG,OAAO,OAAO,CAAC,IAAI,EAAE,CAAC;YACvC,WAAW,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC;gBAC5B,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,EAAE;gBACtC,UAAU,EAAE,OAAO,CAAC,SAAS,IAAI,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC7C,OAAO,EAAE,KAAK,EAAE,KAAU,EAAE,EAAE;oBAC7B,IAAI,CAAC;wBACJ,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;wBACzC,OAAO;4BACN,OAAO,EAAE,IAAI;4BACb,MAAM;yBACN,CAAC;oBACH,CAAC;oBAAC,OAAO,KAAU,EAAE,CAAC;wBACrB,OAAO;4BACN,OAAO,EAAE,KAAK;4BACd,KAAK,EAAE,KAAK,CAAC,OAAO;yBACpB,CAAC;oBACH,CAAC;gBACF,CAAC;aACD,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;IAED,WAAW,CAAC,wBAAwB,GAAG,IAAI,CAAC;QAC3C,WAAW,EACV,wFAAwF;QACzF,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;QACxB,OAAO,EAAE,KAAK,IAAI,EAAE;YACnB,IAAI,CAAC;gBACJ,MAAM,KAAK,GAAG,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAC1C,OAAO;oBACN,OAAO,EAAE,IAAI;oBACb,KAAK;iBACL,CAAC;YACH,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACrB,OAAO;oBACN,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,KAAK,CAAC,OAAO;iBACpB,CAAC;YACH,CAAC;QACF,CAAC;KACD,CAAC,CAAC;IAEH,OAAO;QACN,MAAM;QACN,KAAK,EAAE,WAAW;KAClB,CAAC;AACH,CAAC"}
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../src/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAEhD,OAAO,EAAE,wBAAwB,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AAClF,OAAO,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAChE,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAC1B,OAAO,EAAE,wBAAwB,EAA8B,MAAM,oBAAoB,CAAC;AAE1F,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,OAA8B;IAClE,MAAM,EAAE,sBAAsB,EAAE,GAAG,aAAa,EAAE,GAAG,OAAO,CAAC;IAE7D,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,aAAa,CAAC,CAAC;IACpD,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IAEvB,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC;IAExE,MAAM,WAAW,GAAwB,EAAE,CAAC;IAE5C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAChC,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,CAAC,YAAY,EAAE,CAAC;YAC7C,WAAW,CAAC,gBAAgB,GAAG,IAAI,CAAC;gBACnC,WAAW,EACV,OAAO,CAAC,WAAW;oBACnB,iHAAiH;gBAClH,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;oBACpB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,+CAA+C,CAAC;iBAC1E,CAAC;gBACF,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAoB,EAAE,EAAE;oBAC7C,IAAI,CAAC;wBACJ,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,sBAAsB,CAAC,CAAC;wBAElE,IAAI,MAAM,CAAC,MAAM,KAAK,eAAe,CAAC,SAAS,EAAE,CAAC;4BACjD,OAAO;gCACN,OAAO,EAAE,IAAI;gCACb,MAAM,EAAE,MAAM,CAAC,MAAM;gCACrB,KAAK,EAAE,MAAM,CAAC,KAAK;6BACnB,CAAC;wBACH,CAAC;6BAAM,IAAI,MAAM,CAAC,MAAM,KAAK,eAAe,CAAC,MAAM,EAAE,CAAC;4BACrD,OAAO;gCACN,OAAO,EAAE,KAAK;gCACd,KAAK,EAAE,MAAM,CAAC,KAAK;gCACnB,KAAK,EAAE,MAAM,CAAC,KAAK;6BACnB,CAAC;wBACH,CAAC;6BAAM,CAAC;4BACP,OAAO;gCACN,OAAO,EAAE,KAAK;gCACd,KAAK,EAAE,iCAAiC,GAAG,MAAM,CAAC,MAAM;6BACxD,CAAC;wBACH,CAAC;oBACF,CAAC;oBAAC,OAAO,KAAU,EAAE,CAAC;wBACrB,OAAO;4BACN,OAAO,EAAE,KAAK;4BACd,KAAK,EAAE,KAAK,CAAC,OAAO,IAAI,eAAe;yBACvC,CAAC;oBACH,CAAC;gBACF,CAAC;aACD,CAAC,CAAC;QACJ,CAAC;aAAM,CAAC;YACP,MAAM,QAAQ,GAAG,OAAO,OAAO,CAAC,IAAI,EAAE,CAAC;YACvC,WAAW,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC;gBAC5B,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,EAAE;gBACtC,UAAU,EAAE,OAAO,CAAC,SAAS,IAAI,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC7C,OAAO,EAAE,KAAK,EAAE,KAAU,EAAE,EAAE;oBAC7B,IAAI,CAAC;wBACJ,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;wBACzC,OAAO;4BACN,OAAO,EAAE,IAAI;4BACb,MAAM;yBACN,CAAC;oBACH,CAAC;oBAAC,OAAO,KAAU,EAAE,CAAC;wBACrB,OAAO;4BACN,OAAO,EAAE,KAAK;4BACd,KAAK,EAAE,KAAK,CAAC,OAAO;yBACpB,CAAC;oBACH,CAAC;gBACF,CAAC;aACD,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;IAED,WAAW,CAAC,wBAAwB,GAAG,IAAI,CAAC;QAC3C,WAAW,EACV,wFAAwF;QACzF,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;QACxB,OAAO,EAAE,KAAK,IAAI,EAAE;YACnB,IAAI,CAAC;gBACJ,MAAM,KAAK,GAAG,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAC1C,OAAO;oBACN,OAAO,EAAE,IAAI;oBACb,KAAK;iBACL,CAAC;YACH,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACrB,OAAO;oBACN,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,KAAK,CAAC,OAAO;iBACpB,CAAC;YACH,CAAC;QACF,CAAC;KACD,CAAC,CAAC;IAEH,OAAO;QACN,MAAM;QACN,KAAK,EAAE,WAAW;KAClB,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC5C,OAA8B;IAE9B,MAAM,EAAE,UAAU,EAAE,sBAAsB,EAAE,GAAG,aAAa,EAAE,GAAG,OAAO,CAAC;IAEzE,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,aAAa,CAAC,CAAC;IACpD,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IAEvB,MAAM,YAAY,GAAG,wBAAwB,CAAC,UAAU,CAAC,CAAC;IAC1D,MAAM,gBAAgB,GAAG,MAAM,CAAC,mBAAmB,EAAE,CAAC;IAEtD,MAAM,WAAW,GAAwB,EAAE,CAAC;IAE5C,WAAW,CAAC,gBAAgB,GAAG,IAAI,CAAC;QACnC,WAAW,EACV,4GAA4G;QAC7G,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;YACpB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,+CAA+C,CAAC;SAC1E,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAoB,EAAE,EAAE;YAC7C,IAAI,CAAC;gBACJ,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,aAAa,CAClD,IAAI,EACJ,sBAAsB,EACtB,YAAY,CACZ,CAAC;gBAEF,IAAI,MAAM,CAAC,MAAM,KAAK,eAAe,CAAC,SAAS,EAAE,CAAC;oBACjD,OAAO;wBACN,OAAO,EAAE,IAAI;wBACb,MAAM,EAAE,MAAM,CAAC,MAAM;wBACrB,KAAK,EAAE,MAAM,CAAC,KAAK;qBACnB,CAAC;gBACH,CAAC;qBAAM,IAAI,MAAM,CAAC,MAAM,KAAK,eAAe,CAAC,MAAM,EAAE,CAAC;oBACrD,OAAO;wBACN,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,MAAM,CAAC,KAAK;wBACnB,KAAK,EAAE,MAAM,CAAC,KAAK;qBACnB,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACP,OAAO;wBACN,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,iCAAiC,GAAG,MAAM,CAAC,MAAM;qBACxD,CAAC;gBACH,CAAC;YACF,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACrB,OAAO;oBACN,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,KAAK,CAAC,OAAO,IAAI,eAAe;iBACvC,CAAC;YACH,CAAC;QACF,CAAC;KACD,CAAC,CAAC;IAEH,WAAW,CAAC,wBAAwB,GAAG,IAAI,CAAC;QAC3C,WAAW,EACV,wFAAwF;QACzF,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;QACxB,OAAO,EAAE,KAAK,IAAI,EAAE;YACnB,IAAI,CAAC;gBACJ,MAAM,KAAK,GAAG,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAC1C,OAAO;oBACN,OAAO,EAAE,IAAI;oBACb,KAAK;iBACL,CAAC;YACH,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACrB,OAAO;oBACN,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,KAAK,CAAC,OAAO;iBACpB,CAAC;YACH,CAAC;QACF,CAAC;KACD,CAAC,CAAC;IAEH,OAAO;QACN,MAAM;QACN,KAAK,EAAE,WAAW;KAClB,CAAC;AACH,CAAC"}

package/dist/types.d.ts

@@ -1,5 +1,6 @@
 import type { ExecutionConfig, ClientTool } from '@mondaydotcomorg/atp-protocol';
 import type { ClientHooks } from '@mondaydotcomorg/atp-client';
+import type { UIMessageStreamWriter } from './event-adapter.js';
 export interface ApprovalRequest {
     message: string;
     context?: Record<string, unknown>;
@@ -14,26 +15,54 @@ export type ApprovalHandler = (message: string, context?: Record<string, unknown
 export interface EmbeddingProvider {
     embed(text: string): Promise<number[]>;
 }
-export interface VercelAIATPClientOptions {
-    serverUrl: string;
-    headers?: Record<string, string>;
+export interface InProcessServer {
+    start(): Promise<void>;
+    handleInit(ctx: unknown): Promise<unknown>;
+    getDefinitions(ctx?: unknown): Promise<unknown>;
+    getRuntimeDefinitions(ctx?: unknown): Promise<string>;
+    getInfo(): unknown;
+    handleSearch(ctx: unknown): Promise<unknown>;
+    handleExplore(ctx: unknown): Promise<unknown>;
+    handleExecute(ctx: unknown): Promise<unknown>;
+    handleResume(ctx: unknown, executionId: string): Promise<unknown>;
+}
+interface BaseClientOptions {
     model: any;
     embeddings?: EmbeddingProvider;
     tools?: ClientTool[];
     approvalHandler?: ApprovalHandler;
     hooks?: ClientHooks;
 }
-export interface CreateATPToolsOptions {
+/** HTTP mode options */
+interface HttpModeOptions extends BaseClientOptions {
     serverUrl: string;
     headers?: Record<string, string>;
-    model: any;
-    embeddings?: EmbeddingProvider;
-    approvalHandler?: ApprovalHandler;
-    defaultExecutionConfig?: Partial<ExecutionConfig>;
-    hooks?: ClientHooks;
+    server?: never;
 }
+/** In-process mode options */
+interface InProcessModeOptions extends BaseClientOptions {
+    server: InProcessServer;
+    serverUrl?: never;
+    headers?: never;
+}
+export type VercelAIATPClientOptions = HttpModeOptions | InProcessModeOptions;
+export type CreateATPToolsOptions = (HttpModeOptions | InProcessModeOptions) & {
+    defaultExecutionConfig?: Partial<ExecutionConfig>;
+};
+/**
+ * Options for creating ATP tools with streaming event support
+ */
+export type StreamingToolsOptions = CreateATPToolsOptions & {
+    /**
+     * UIMessageStreamWriter to forward events to.
+     * Events like 'thinking', 'tool_start', 'tool_end', 'text', 'source' will be
+     * converted to Vercel AI SDK format and written to this stream.
+     */
+    dataStream: UIMessageStreamWriter;
+};
 export interface ATPToolsResult {
     client: any;
     tools: Record<string, any>;
 }
+export {};
 //# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,+BAA+B,CAAC;AACjF,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAE/D,MAAM,WAAW,eAAe;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,SAAS,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,gBAAgB;IAChC,QAAQ,EAAE,OAAO,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,MAAM,eAAe,GAAG,CAC7B,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC7B,OAAO,CAAC,OAAO,CAAC,CAAC;AAEtB,MAAM,WAAW,iBAAiB;IACjC,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;CACvC;AAED,MAAM,WAAW,wBAAwB;IACxC,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,KAAK,EAAE,GAAG,CAAC;IACX,UAAU,CAAC,EAAE,iBAAiB,CAAC;IAC/B,KAAK,CAAC,EAAE,UAAU,EAAE,CAAC;IACrB,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,KAAK,CAAC,EAAE,WAAW,CAAC;CACpB;AAED,MAAM,WAAW,qBAAqB;IACrC,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,KAAK,EAAE,GAAG,CAAC;IACX,UAAU,CAAC,EAAE,iBAAiB,CAAC;IAC/B,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,sBAAsB,CAAC,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC;IAClD,KAAK,CAAC,EAAE,WAAW,CAAC;CACpB;AAED,MAAM,WAAW,cAAc;IAC9B,MAAM,EAAE,GAAG,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC3B"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,+BAA+B,CAAC;AACjF,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAEhE,MAAM,WAAW,eAAe;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,SAAS,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,gBAAgB;IAChC,QAAQ,EAAE,OAAO,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,MAAM,eAAe,GAAG,CAC7B,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC7B,OAAO,CAAC,OAAO,CAAC,CAAC;AAEtB,MAAM,WAAW,iBAAiB;IACjC,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;CACvC;AAED,MAAM,WAAW,eAAe;IAC/B,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,UAAU,CAAC,GAAG,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC3C,cAAc,CAAC,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAChD,qBAAqB,CAAC,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IACtD,OAAO,IAAI,OAAO,CAAC;IACnB,YAAY,CAAC,GAAG,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC7C,aAAa,CAAC,GAAG,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC9C,aAAa,CAAC,GAAG,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC9C,YAAY,CAAC,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;CAClE;AAED,UAAU,iBAAiB;IAC1B,KAAK,EAAE,GAAG,CAAC;IACX,UAAU,CAAC,EAAE,iBAAiB,CAAC;IAC/B,KAAK,CAAC,EAAE,UAAU,EAAE,CAAC;IACrB,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,KAAK,CAAC,EAAE,WAAW,CAAC;CACpB;AAED,wBAAwB;AACxB,UAAU,eAAgB,SAAQ,iBAAiB;IAClD,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,MAAM,CAAC,EAAE,KAAK,CAAC;CACf;AAED,8BAA8B;AAC9B,UAAU,oBAAqB,SAAQ,iBAAiB;IACvD,MAAM,EAAE,eAAe,CAAC;IACxB,SAAS,CAAC,EAAE,KAAK,CAAC;IAClB,OAAO,CAAC,EAAE,KAAK,CAAC;CAChB;AAED,MAAM,MAAM,wBAAwB,GAAG,eAAe,GAAG,oBAAoB,CAAC;AAE9E,MAAM,MAAM,qBAAqB,GAAG,CAAC,eAAe,GAAG,oBAAoB,CAAC,GAAG;IAC9E,sBAAsB,CAAC,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC;CAClD,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG,qBAAqB,GAAG;IAC3D;;;;OAIG;IACH,UAAU,EAAE,qBAAqB,CAAC;CAClC,CAAC;AAEF,MAAM,WAAW,cAAc;IAC9B,MAAM,EAAE,GAAG,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC3B"}

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@mondaydotcomorg/atp-vercel-sdk",
-	"version": "0.20.1",
+	"version": "0.20.3",
 	"description": "Vercel AI SDK integration for Agent Tool Protocol",
 	"type": "module",
 	"main": "./dist/index.cjs",
@@ -44,8 +44,8 @@
 	],
 	"license": "MIT",
 	"dependencies": {
-		"@mondaydotcomorg/atp-client": "0.19.14",
-		"@mondaydotcomorg/atp-protocol": "0.19.14"
+		"@mondaydotcomorg/atp-client": "0.19.15",
+		"@mondaydotcomorg/atp-protocol": "0.19.15"
 	},
 	"peerDependencies": {
 		"ai": "^4.0.0",

package/src/client.ts

@@ -15,14 +15,25 @@ export class VercelAIATPClient {
 	private approvalHandler?: ApprovalHandler;
 
 	constructor(options: VercelAIATPClientOptions) {
-		const { serverUrl, headers, model, embeddings, tools, approvalHandler, hooks } = options;
+		const { model, embeddings, tools, approvalHandler, hooks } = options;
+
+		if ('server' in options && options.server) {
+			this.client = new AgentToolProtocolClient({
+				server: options.server,
+				hooks,
+				serviceProviders: tools ? { tools } : undefined,
+			});
+		} else if ('serverUrl' in options && options.serverUrl) {
+			this.client = new AgentToolProtocolClient({
+				baseUrl: options.serverUrl,
+				headers: options.headers,
+				hooks,
+				serviceProviders: tools ? { tools } : undefined,
+			});
+		} else {
+			throw new Error('Either serverUrl or server must be provided');
+		}
 
-		this.client = new AgentToolProtocolClient({
-			baseUrl: serverUrl,
-			headers,
-			hooks,
-			serviceProviders: tools ? { tools } : undefined,
-		});
 		this.model = model;
 		this.embeddings = embeddings;
 		this.approvalHandler = approvalHandler;

package/src/event-adapter.ts

@@ -0,0 +1,275 @@
+/**
+ * Vercel AI SDK Event Adapter
+ *
+ * Converts ATP streaming events to Vercel AI SDK UIMessageStream format.
+ * Use this to forward ATP events to the chat UI.
+ */
+
+import { type ATPEvent, ATPEventType } from '@mondaydotcomorg/atp-protocol';
+
+/**
+ * UIMessageStreamWriter interface (subset of Vercel AI SDK's UIMessageStreamWriter)
+ */
+export interface UIMessageStreamWriter {
+	write(event: UIStreamEvent): void;
+}
+
+/**
+ * UI Stream event types that Vercel AI SDK understands
+ */
+export type UIStreamEvent =
+	| { type: 'text-start'; id: string }
+	| { type: 'text-delta'; id: string; delta: string }
+	| { type: 'text-end'; id: string }
+	| { type: 'reasoning-start'; id: string }
+	| { type: 'reasoning-delta'; id: string; delta: string }
+	| { type: 'reasoning-end'; id: string }
+	| { type: 'source-url'; url: string; title: string; sourceId: string; providerMetadata?: unknown }
+	| { type: string; data?: unknown; transient?: boolean };
+
+/**
+ * Options for creating the Vercel event handler
+ */
+export interface CreateVercelEventHandlerOptions {
+	/**
+	 * Prefix for generated text run IDs
+	 * @default 'atp-text'
+	 */
+	textRunIdPrefix?: string;
+}
+
+/**
+ * Creates an event handler that forwards ATP events to a Vercel AI SDK UIMessageStreamWriter.
+ *
+ * Handles nested tool calls properly:
+ * - First tool_start triggers agentStepStart
+ * - Inner tool_start/end emit agentStep updates (not start/end)
+ * - Last tool_end triggers agentStepEnd
+ *
+ * @param dataStream - The Vercel AI SDK UIMessageStreamWriter to write events to
+ * @param options - Optional configuration
+ * @returns An event handler function to pass to ATP's executeStream
+ *
+ * @example
+ * ```typescript
+ * const handler = createVercelEventHandler(dataStream);
+ *
+ * const result = await client.executeStream(code, {}, handler);
+ * ```
+ */
+export function createVercelEventHandler(
+	dataStream: UIMessageStreamWriter,
+	options: CreateVercelEventHandlerOptions = {}
+): (event: ATPEvent) => void {
+	const { textRunIdPrefix = 'atp-text' } = options;
+
+	let currentTextRunId: string | undefined;
+	let textRunCounter = 0;
+
+	// Track reasoning state for proper start/delta/end events
+	let reasoningRunId: string | undefined;
+	let reasoningCounter = 0;
+
+	// Track nested tool calls with a stack
+	const toolStack: Array<{ toolName: string; apiGroup: string }> = [];
+
+	return (event: ATPEvent) => {
+		switch (event.type) {
+			case ATPEventType.THINKING: {
+				const data = event.data as { content: string; step?: string };
+
+				// Start new reasoning block if needed
+				if (!reasoningRunId) {
+					reasoningRunId = `atp-reasoning-${++reasoningCounter}`;
+					dataStream.write({
+						type: 'reasoning-start',
+						id: reasoningRunId,
+					} as UIStreamEvent);
+				}
+
+				// Write reasoning delta
+				dataStream.write({
+					type: 'reasoning-delta',
+					id: reasoningRunId,
+					delta: data.content + '\n',
+				} as UIStreamEvent);
+				break;
+			}
+
+			case ATPEventType.TOOL_START: {
+				const data = event.data as { toolName: string; apiGroup: string; input: unknown };
+				const isFirstTool = toolStack.length === 0;
+
+				toolStack.push({ toolName: data.toolName, apiGroup: data.apiGroup });
+
+				if (isFirstTool) {
+					// First tool - emit agentStepStart
+					dataStream.write({
+						type: 'data-agentStepStart',
+						data: `Executing ${data.apiGroup}.${data.toolName}...`,
+						transient: true,
+					});
+				} else {
+					// Nested tool - emit progress update instead
+					dataStream.write({
+						type: 'data-agentStep',
+						data: `→ ${data.apiGroup}.${data.toolName}`,
+						transient: true,
+					});
+				}
+				break;
+			}
+
+			case ATPEventType.TOOL_END: {
+				const data = event.data as {
+					toolName: string;
+					apiGroup: string;
+					output: unknown;
+					duration: number;
+					success: boolean;
+					error?: string;
+				};
+
+				toolStack.pop();
+				const isLastTool = toolStack.length === 0;
+
+				// Close reasoning block when outer tool ends
+				if (isLastTool && reasoningRunId) {
+					dataStream.write({
+						type: 'reasoning-end',
+						id: reasoningRunId,
+					} as UIStreamEvent);
+					reasoningRunId = undefined;
+				}
+
+				if (isLastTool) {
+					// Last tool - emit agentStepEnd
+					dataStream.write({
+						type: 'data-agentStepEnd',
+						data: data.success
+							? `Completed ${data.apiGroup}.${data.toolName} (${data.duration}ms)`
+							: `Failed ${data.apiGroup}.${data.toolName}: ${data.error}`,
+						transient: true,
+					});
+				} else {
+					// Nested tool finished - emit progress update
+					dataStream.write({
+						type: 'data-agentStep',
+						data: data.success
+							? `✓ ${data.apiGroup}.${data.toolName} (${data.duration}ms)`
+							: `✗ ${data.apiGroup}.${data.toolName}: ${data.error}`,
+						transient: true,
+					});
+				}
+				break;
+			}
+
+			case ATPEventType.TEXT: {
+				const data = event.data as { text: string };
+				const runId = event.runId || `${textRunIdPrefix}-${++textRunCounter}`;
+
+				// If we're starting a new text run, close the previous one
+				if (currentTextRunId && currentTextRunId !== runId) {
+					dataStream.write({ type: 'text-end', id: currentTextRunId });
+					currentTextRunId = undefined;
+				}
+
+				// Start new text run if needed
+				if (!currentTextRunId || currentTextRunId !== runId) {
+					currentTextRunId = runId;
+					dataStream.write({ type: 'text-start', id: currentTextRunId });
+				}
+
+				// Write text delta
+				dataStream.write({
+					type: 'text-delta',
+					id: currentTextRunId,
+					delta: data.text,
+				});
+				break;
+			}
+
+			case ATPEventType.TEXT_END:
+				if (currentTextRunId) {
+					dataStream.write({ type: 'text-end', id: currentTextRunId });
+					currentTextRunId = undefined;
+				}
+				break;
+
+			case ATPEventType.SOURCE: {
+				const data = event.data as {
+					url: string;
+					title: string;
+					summary?: string;
+					createdAt?: string;
+				};
+				dataStream.write({
+					type: 'source-url',
+					url: data.url,
+					title: data.title,
+					sourceId: `${data.title}-${event.timestamp}`,
+					providerMetadata: {
+						aiChat: {
+							summary: data.summary,
+							createdAt: data.createdAt,
+						},
+					},
+				});
+				break;
+			}
+
+			case ATPEventType.PROGRESS: {
+				const data = event.data as { message: string; fraction: number };
+				dataStream.write({
+					type: 'data-agentStep',
+					data: `${data.message} (${Math.round(data.fraction * 100)}%)`,
+					transient: true,
+				});
+				break;
+			}
+
+			case ATPEventType.ERROR: {
+				const data = event.data as { message: string; code?: string };
+				dataStream.write({
+					type: 'data-error',
+					data: data.message,
+				});
+				break;
+			}
+
+			default:
+				// Forward unknown events as custom data events
+				dataStream.write({
+					type: `data-${event.type}`,
+					data: event.data,
+					transient: true,
+				});
+				break;
+		}
+	};
+}
+
+/**
+ * Creates an event handler that collects events into an array.
+ * Useful for testing or post-processing events.
+ *
+ * @returns Object with handler function and collected events array
+ */
+export function createEventCollector(): {
+	handler: (event: ATPEvent) => void;
+	events: ATPEvent[];
+	clear: () => void;
+} {
+	const events: ATPEvent[] = [];
+
+	return {
+		handler: (event: ATPEvent) => {
+			events.push(event);
+		},
+		events,
+		clear: () => {
+			events.length = 0;
+		},
+	};
+}
+

package/src/index.ts

@@ -1,8 +1,16 @@
 export { VercelAIATPClient } from './client.js';
-export { createATPTools } from './tools.js';
+export { createATPTools, createATPStreamingTools } from './tools.js';
+export {
+	createVercelEventHandler,
+	createEventCollector,
+	type UIMessageStreamWriter,
+	type UIStreamEvent,
+	type CreateVercelEventHandlerOptions,
+} from './event-adapter.js';
 export type {
 	VercelAIATPClientOptions,
 	CreateATPToolsOptions,
+	StreamingToolsOptions,
 	ATPToolsResult,
 	ApprovalHandler,
 	ApprovalRequest,

package/src/tools.ts

@@ -1,9 +1,10 @@
 import { z } from 'zod';
 import { VercelAIATPClient } from './client.js';
-import type { CreateATPToolsOptions, ATPToolsResult } from './types.js';
+import type { CreateATPToolsOptions, ATPToolsResult, StreamingToolsOptions } from './types.js';
 import { createToolsFromATPClient, ToolNames } from '@mondaydotcomorg/atp-client';
 import { ExecutionStatus } from '@mondaydotcomorg/atp-protocol';
 import { tool } from 'ai';
+import { createVercelEventHandler, type UIMessageStreamWriter } from './event-adapter.js';
 
 export async function createATPTools(options: CreateATPToolsOptions): Promise<ATPToolsResult> {
 	const { defaultExecutionConfig, ...clientOptions } = options;
@@ -102,3 +103,109 @@ export async function createATPTools(options: CreateATPToolsOptions): Promise<AT
 		tools: vercelTools,
 	};
 }
+
+/**
+ * Creates ATP tools with streaming event support.
+ * Events from tool execution will be forwarded to the provided dataStream.
+ *
+ * @param options - Tool creation options including dataStream for event forwarding
+ * @returns Promise resolving to client and tools with streaming support
+ *
+ * @example
+ * ```typescript
+ * // In your Vercel AI SDK route handler
+ * const dataStream = createUIMessageStream({...});
+ *
+ * const { tools } = await createATPStreamingTools({
+ *   serverUrl: 'http://localhost:3333',
+ *   model: openai('gpt-4'),
+ *   dataStream,
+ * });
+ *
+ * // Use tools in streamText
+ * const result = streamText({
+ *   model: openai('gpt-4'),
+ *   messages,
+ *   tools,
+ * });
+ * ```
+ */
+export async function createATPStreamingTools(
+	options: StreamingToolsOptions
+): Promise<ATPToolsResult> {
+	const { dataStream, defaultExecutionConfig, ...clientOptions } = options;
+
+	const client = new VercelAIATPClient(clientOptions);
+	await client.connect();
+
+	const eventHandler = createVercelEventHandler(dataStream);
+	const underlyingClient = client.getUnderlyingClient();
+
+	const vercelTools: Record<string, any> = {};
+
+	vercelTools.atp_execute_code = tool({
+		description:
+			'Execute TypeScript code in ATP sandbox with streaming events for thinking, tool execution, and text output',
+		parameters: z.object({
+			code: z.string().describe('TypeScript code to execute in the ATP sandbox'),
+		}),
+		execute: async ({ code }: { code: string }) => {
+			try {
+				const result = await underlyingClient.executeStream(
+					code,
+					defaultExecutionConfig,
+					eventHandler
+				);
+
+				if (result.status === ExecutionStatus.COMPLETED) {
+					return {
+						success: true,
+						result: result.result,
+						stats: result.stats,
+					};
+				} else if (result.status === ExecutionStatus.FAILED) {
+					return {
+						success: false,
+						error: result.error,
+						stats: result.stats,
+					};
+				} else {
+					return {
+						success: false,
+						error: 'Execution in unexpected state: ' + result.status,
+					};
+				}
+			} catch (error: any) {
+				return {
+					success: false,
+					error: error.message || 'Unknown error',
+				};
+			}
+		},
+	});
+
+	vercelTools.atp_get_type_definitions = tool({
+		description:
+			'Get TypeScript type definitions for ATP runtime APIs to understand available functions',
+		parameters: z.object({}),
+		execute: async () => {
+			try {
+				const types = client.getTypeDefinitions();
+				return {
+					success: true,
+					types,
+				};
+			} catch (error: any) {
+				return {
+					success: false,
+					error: error.message,
+				};
+			}
+		},
+	});
+
+	return {
+		client,
+		tools: vercelTools,
+	};
+}