@mondaydotcomorg/atp-vercel-sdk 0.20.1 → 0.20.3

package/src/types.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;
@@ -22,9 +23,19 @@ 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[];
@@ -32,16 +43,38 @@ export interface VercelAIATPClientOptions {
 	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>;