@mondaydotcomorg/atp-vercel-sdk 0.20.2 → 0.20.4
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/index.cjs +537 -0
- package/dist/index.cjs.map +1 -0
- package/dist/index.js +530 -3
- package/dist/index.js.map +1 -1
- package/dist/tools.d.ts.map +1 -1
- package/dist/tools.js +132 -69
- package/dist/tools.js.map +1 -1
- package/package.json +2 -2
- package/src/tools.ts +142 -67
package/dist/index.js
CHANGED
|
@@ -1,4 +1,531 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
1
|
+
import { AgentToolProtocolClient, ToolNames, fetchAllApisInputSchema, searchApiInputSchema, exploreApiInputSchema, executeCodeInputSchema } from '@mondaydotcomorg/atp-client';
|
|
2
|
+
import { generateText, generateObject, tool } from 'ai';
|
|
3
|
+
import { z } from 'zod';
|
|
4
|
+
import { ATPEventType, ExecutionStatus } from '@mondaydotcomorg/atp-protocol';
|
|
5
|
+
|
|
6
|
+
var __defProp = Object.defineProperty;
|
|
7
|
+
var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
|
|
8
|
+
var VercelAIATPClient = class {
|
|
9
|
+
static {
|
|
10
|
+
__name(this, "VercelAIATPClient");
|
|
11
|
+
}
|
|
12
|
+
client;
|
|
13
|
+
model;
|
|
14
|
+
embeddings;
|
|
15
|
+
approvalHandler;
|
|
16
|
+
constructor(options) {
|
|
17
|
+
const { model, embeddings, tools, approvalHandler, hooks } = options;
|
|
18
|
+
if ("server" in options && options.server) {
|
|
19
|
+
this.client = new AgentToolProtocolClient({
|
|
20
|
+
server: options.server,
|
|
21
|
+
hooks,
|
|
22
|
+
serviceProviders: tools ? {
|
|
23
|
+
tools
|
|
24
|
+
} : void 0
|
|
25
|
+
});
|
|
26
|
+
} else if ("serverUrl" in options && options.serverUrl) {
|
|
27
|
+
this.client = new AgentToolProtocolClient({
|
|
28
|
+
baseUrl: options.serverUrl,
|
|
29
|
+
headers: options.headers,
|
|
30
|
+
hooks,
|
|
31
|
+
serviceProviders: tools ? {
|
|
32
|
+
tools
|
|
33
|
+
} : void 0
|
|
34
|
+
});
|
|
35
|
+
} else {
|
|
36
|
+
throw new Error("Either serverUrl or server must be provided");
|
|
37
|
+
}
|
|
38
|
+
this.model = model;
|
|
39
|
+
this.embeddings = embeddings;
|
|
40
|
+
this.approvalHandler = approvalHandler;
|
|
41
|
+
this.client.provideLLM({
|
|
42
|
+
call: /* @__PURE__ */ __name(async (prompt, options2) => {
|
|
43
|
+
return await this.handleLLMCall(prompt, options2);
|
|
44
|
+
}, "call"),
|
|
45
|
+
extract: /* @__PURE__ */ __name(async (prompt, schema, options2) => {
|
|
46
|
+
return await this.handleLLMExtract(prompt, schema, options2);
|
|
47
|
+
}, "extract"),
|
|
48
|
+
classify: /* @__PURE__ */ __name(async (text, categories, options2) => {
|
|
49
|
+
return await this.handleLLMClassify(text, categories, options2);
|
|
50
|
+
}, "classify")
|
|
51
|
+
});
|
|
52
|
+
if (this.embeddings) {
|
|
53
|
+
this.client.provideEmbedding({
|
|
54
|
+
embed: /* @__PURE__ */ __name(async (text) => {
|
|
55
|
+
return await this.handleEmbedding(text);
|
|
56
|
+
}, "embed")
|
|
57
|
+
});
|
|
58
|
+
}
|
|
59
|
+
this.client.provideApproval({
|
|
60
|
+
request: /* @__PURE__ */ __name(async (message, context) => {
|
|
61
|
+
return await this.handleApprovalRequest(message, context);
|
|
62
|
+
}, "request")
|
|
63
|
+
});
|
|
64
|
+
}
|
|
65
|
+
async connect() {
|
|
66
|
+
await this.client.init({
|
|
67
|
+
name: "vercel-ai-sdk-atp-client",
|
|
68
|
+
version: "1.0.0"
|
|
69
|
+
});
|
|
70
|
+
await this.client.connect();
|
|
71
|
+
}
|
|
72
|
+
getTypeDefinitions() {
|
|
73
|
+
return this.client.getTypeDefinitions();
|
|
74
|
+
}
|
|
75
|
+
async execute(code, config) {
|
|
76
|
+
return await this.client.execute(code, config);
|
|
77
|
+
}
|
|
78
|
+
getUnderlyingClient() {
|
|
79
|
+
return this.client;
|
|
80
|
+
}
|
|
81
|
+
async handleLLMCall(prompt, options) {
|
|
82
|
+
const result = await generateText({
|
|
83
|
+
model: this.model,
|
|
84
|
+
prompt,
|
|
85
|
+
temperature: options?.temperature,
|
|
86
|
+
maxTokens: options?.maxTokens,
|
|
87
|
+
system: options?.systemPrompt
|
|
88
|
+
});
|
|
89
|
+
return result.text;
|
|
90
|
+
}
|
|
91
|
+
async handleLLMExtract(prompt, schema, options) {
|
|
92
|
+
const result = await generateObject({
|
|
93
|
+
model: this.model,
|
|
94
|
+
prompt,
|
|
95
|
+
schema,
|
|
96
|
+
system: options?.systemPrompt,
|
|
97
|
+
temperature: options?.temperature,
|
|
98
|
+
maxTokens: options?.maxTokens
|
|
99
|
+
});
|
|
100
|
+
return result.object;
|
|
101
|
+
}
|
|
102
|
+
async handleLLMClassify(text, categories, options) {
|
|
103
|
+
const promptText = `Classify the following text into one of these categories: ${categories.join(", ")}
|
|
104
|
+
|
|
105
|
+
Text: ${text}
|
|
106
|
+
|
|
107
|
+
Respond with ONLY the category name, nothing else.`;
|
|
108
|
+
const result = await generateText({
|
|
109
|
+
model: this.model,
|
|
110
|
+
prompt: promptText,
|
|
111
|
+
system: options?.systemPrompt,
|
|
112
|
+
temperature: 0
|
|
113
|
+
});
|
|
114
|
+
const classification = result.text.trim();
|
|
115
|
+
if (categories.includes(classification)) {
|
|
116
|
+
return classification;
|
|
117
|
+
}
|
|
118
|
+
for (const category of categories) {
|
|
119
|
+
if (classification.toLowerCase().includes(category.toLowerCase())) {
|
|
120
|
+
return category;
|
|
121
|
+
}
|
|
122
|
+
}
|
|
123
|
+
const fallback = categories[0];
|
|
124
|
+
if (!fallback) {
|
|
125
|
+
throw new Error("No categories provided for classification");
|
|
126
|
+
}
|
|
127
|
+
return fallback;
|
|
128
|
+
}
|
|
129
|
+
async handleEmbedding(text) {
|
|
130
|
+
if (!this.embeddings) {
|
|
131
|
+
throw new Error("Embeddings provider not configured. Pass embeddings option when creating VercelAIATPClient.");
|
|
132
|
+
}
|
|
133
|
+
return await this.embeddings.embed(text);
|
|
134
|
+
}
|
|
135
|
+
async handleApprovalRequest(message, context) {
|
|
136
|
+
if (!this.approvalHandler) {
|
|
137
|
+
throw new Error("No approval handler configured. Pass approvalHandler option when creating VercelAIATPClient.");
|
|
138
|
+
}
|
|
139
|
+
const approved = await this.approvalHandler(message, context);
|
|
140
|
+
return {
|
|
141
|
+
approved,
|
|
142
|
+
timestamp: Date.now()
|
|
143
|
+
};
|
|
144
|
+
}
|
|
145
|
+
};
|
|
146
|
+
function createVercelEventHandler(dataStream, options = {}) {
|
|
147
|
+
const { textRunIdPrefix = "atp-text" } = options;
|
|
148
|
+
let currentTextRunId;
|
|
149
|
+
let textRunCounter = 0;
|
|
150
|
+
let reasoningRunId;
|
|
151
|
+
let reasoningCounter = 0;
|
|
152
|
+
const toolStack = [];
|
|
153
|
+
return (event) => {
|
|
154
|
+
switch (event.type) {
|
|
155
|
+
case ATPEventType.THINKING: {
|
|
156
|
+
const data = event.data;
|
|
157
|
+
if (!reasoningRunId) {
|
|
158
|
+
reasoningRunId = `atp-reasoning-${++reasoningCounter}`;
|
|
159
|
+
dataStream.write({
|
|
160
|
+
type: "reasoning-start",
|
|
161
|
+
id: reasoningRunId
|
|
162
|
+
});
|
|
163
|
+
}
|
|
164
|
+
dataStream.write({
|
|
165
|
+
type: "reasoning-delta",
|
|
166
|
+
id: reasoningRunId,
|
|
167
|
+
delta: data.content + "\n"
|
|
168
|
+
});
|
|
169
|
+
break;
|
|
170
|
+
}
|
|
171
|
+
case ATPEventType.TOOL_START: {
|
|
172
|
+
const data = event.data;
|
|
173
|
+
const isFirstTool = toolStack.length === 0;
|
|
174
|
+
toolStack.push({
|
|
175
|
+
toolName: data.toolName,
|
|
176
|
+
apiGroup: data.apiGroup
|
|
177
|
+
});
|
|
178
|
+
if (isFirstTool) {
|
|
179
|
+
dataStream.write({
|
|
180
|
+
type: "data-agentStepStart",
|
|
181
|
+
data: `Executing ${data.apiGroup}.${data.toolName}...`,
|
|
182
|
+
transient: true
|
|
183
|
+
});
|
|
184
|
+
} else {
|
|
185
|
+
dataStream.write({
|
|
186
|
+
type: "data-agentStep",
|
|
187
|
+
data: `\u2192 ${data.apiGroup}.${data.toolName}`,
|
|
188
|
+
transient: true
|
|
189
|
+
});
|
|
190
|
+
}
|
|
191
|
+
break;
|
|
192
|
+
}
|
|
193
|
+
case ATPEventType.TOOL_END: {
|
|
194
|
+
const data = event.data;
|
|
195
|
+
toolStack.pop();
|
|
196
|
+
const isLastTool = toolStack.length === 0;
|
|
197
|
+
if (isLastTool && reasoningRunId) {
|
|
198
|
+
dataStream.write({
|
|
199
|
+
type: "reasoning-end",
|
|
200
|
+
id: reasoningRunId
|
|
201
|
+
});
|
|
202
|
+
reasoningRunId = void 0;
|
|
203
|
+
}
|
|
204
|
+
if (isLastTool) {
|
|
205
|
+
dataStream.write({
|
|
206
|
+
type: "data-agentStepEnd",
|
|
207
|
+
data: data.success ? `Completed ${data.apiGroup}.${data.toolName} (${data.duration}ms)` : `Failed ${data.apiGroup}.${data.toolName}: ${data.error}`,
|
|
208
|
+
transient: true
|
|
209
|
+
});
|
|
210
|
+
} else {
|
|
211
|
+
dataStream.write({
|
|
212
|
+
type: "data-agentStep",
|
|
213
|
+
data: data.success ? `\u2713 ${data.apiGroup}.${data.toolName} (${data.duration}ms)` : `\u2717 ${data.apiGroup}.${data.toolName}: ${data.error}`,
|
|
214
|
+
transient: true
|
|
215
|
+
});
|
|
216
|
+
}
|
|
217
|
+
break;
|
|
218
|
+
}
|
|
219
|
+
case ATPEventType.TEXT: {
|
|
220
|
+
const data = event.data;
|
|
221
|
+
const runId = event.runId || `${textRunIdPrefix}-${++textRunCounter}`;
|
|
222
|
+
if (currentTextRunId && currentTextRunId !== runId) {
|
|
223
|
+
dataStream.write({
|
|
224
|
+
type: "text-end",
|
|
225
|
+
id: currentTextRunId
|
|
226
|
+
});
|
|
227
|
+
currentTextRunId = void 0;
|
|
228
|
+
}
|
|
229
|
+
if (!currentTextRunId || currentTextRunId !== runId) {
|
|
230
|
+
currentTextRunId = runId;
|
|
231
|
+
dataStream.write({
|
|
232
|
+
type: "text-start",
|
|
233
|
+
id: currentTextRunId
|
|
234
|
+
});
|
|
235
|
+
}
|
|
236
|
+
dataStream.write({
|
|
237
|
+
type: "text-delta",
|
|
238
|
+
id: currentTextRunId,
|
|
239
|
+
delta: data.text
|
|
240
|
+
});
|
|
241
|
+
break;
|
|
242
|
+
}
|
|
243
|
+
case ATPEventType.TEXT_END:
|
|
244
|
+
if (currentTextRunId) {
|
|
245
|
+
dataStream.write({
|
|
246
|
+
type: "text-end",
|
|
247
|
+
id: currentTextRunId
|
|
248
|
+
});
|
|
249
|
+
currentTextRunId = void 0;
|
|
250
|
+
}
|
|
251
|
+
break;
|
|
252
|
+
case ATPEventType.SOURCE: {
|
|
253
|
+
const data = event.data;
|
|
254
|
+
dataStream.write({
|
|
255
|
+
type: "source-url",
|
|
256
|
+
url: data.url,
|
|
257
|
+
title: data.title,
|
|
258
|
+
sourceId: `${data.title}-${event.timestamp}`,
|
|
259
|
+
providerMetadata: {
|
|
260
|
+
aiChat: {
|
|
261
|
+
summary: data.summary,
|
|
262
|
+
createdAt: data.createdAt
|
|
263
|
+
}
|
|
264
|
+
}
|
|
265
|
+
});
|
|
266
|
+
break;
|
|
267
|
+
}
|
|
268
|
+
case ATPEventType.PROGRESS: {
|
|
269
|
+
const data = event.data;
|
|
270
|
+
dataStream.write({
|
|
271
|
+
type: "data-agentStep",
|
|
272
|
+
data: `${data.message} (${Math.round(data.fraction * 100)}%)`,
|
|
273
|
+
transient: true
|
|
274
|
+
});
|
|
275
|
+
break;
|
|
276
|
+
}
|
|
277
|
+
case ATPEventType.ERROR: {
|
|
278
|
+
const data = event.data;
|
|
279
|
+
dataStream.write({
|
|
280
|
+
type: "data-error",
|
|
281
|
+
data: data.message
|
|
282
|
+
});
|
|
283
|
+
break;
|
|
284
|
+
}
|
|
285
|
+
default:
|
|
286
|
+
dataStream.write({
|
|
287
|
+
type: `data-${event.type}`,
|
|
288
|
+
data: event.data,
|
|
289
|
+
transient: true
|
|
290
|
+
});
|
|
291
|
+
break;
|
|
292
|
+
}
|
|
293
|
+
};
|
|
294
|
+
}
|
|
295
|
+
__name(createVercelEventHandler, "createVercelEventHandler");
|
|
296
|
+
function createEventCollector() {
|
|
297
|
+
const events = [];
|
|
298
|
+
return {
|
|
299
|
+
handler: /* @__PURE__ */ __name((event) => {
|
|
300
|
+
events.push(event);
|
|
301
|
+
}, "handler"),
|
|
302
|
+
events,
|
|
303
|
+
clear: /* @__PURE__ */ __name(() => {
|
|
304
|
+
events.length = 0;
|
|
305
|
+
}, "clear")
|
|
306
|
+
};
|
|
307
|
+
}
|
|
308
|
+
__name(createEventCollector, "createEventCollector");
|
|
309
|
+
|
|
310
|
+
// src/tools.ts
|
|
311
|
+
var TOOL_SCHEMAS = {
|
|
312
|
+
[ToolNames.EXECUTE_CODE]: executeCodeInputSchema.pick({
|
|
313
|
+
code: true
|
|
314
|
+
}),
|
|
315
|
+
[ToolNames.EXPLORE_API]: exploreApiInputSchema,
|
|
316
|
+
[ToolNames.SEARCH_API]: searchApiInputSchema,
|
|
317
|
+
[ToolNames.FETCH_ALL_APIS]: fetchAllApisInputSchema
|
|
318
|
+
};
|
|
319
|
+
async function createATPTools(options) {
|
|
320
|
+
const { defaultExecutionConfig, ...clientOptions } = options;
|
|
321
|
+
const client = new VercelAIATPClient(clientOptions);
|
|
322
|
+
await client.connect();
|
|
323
|
+
const underlyingClient = client.getUnderlyingClient();
|
|
324
|
+
const vercelTools = {};
|
|
325
|
+
vercelTools.atp_execute_code = tool({
|
|
326
|
+
description: "Execute TypeScript code in ATP sandbox with access to runtime APIs (atp.llm.*, atp.embedding.*, atp.approval.*)",
|
|
327
|
+
parameters: TOOL_SCHEMAS[ToolNames.EXECUTE_CODE],
|
|
328
|
+
execute: /* @__PURE__ */ __name(async ({ code }) => {
|
|
329
|
+
try {
|
|
330
|
+
const result = await client.execute(code, defaultExecutionConfig);
|
|
331
|
+
if (result.status === ExecutionStatus.COMPLETED) {
|
|
332
|
+
return {
|
|
333
|
+
success: true,
|
|
334
|
+
result: result.result,
|
|
335
|
+
stats: result.stats
|
|
336
|
+
};
|
|
337
|
+
} else if (result.status === ExecutionStatus.FAILED) {
|
|
338
|
+
return {
|
|
339
|
+
success: false,
|
|
340
|
+
error: result.error,
|
|
341
|
+
stats: result.stats
|
|
342
|
+
};
|
|
343
|
+
} else {
|
|
344
|
+
return {
|
|
345
|
+
success: false,
|
|
346
|
+
error: "Execution in unexpected state: " + result.status
|
|
347
|
+
};
|
|
348
|
+
}
|
|
349
|
+
} catch (error) {
|
|
350
|
+
return {
|
|
351
|
+
success: false,
|
|
352
|
+
error: error.message || "Unknown error"
|
|
353
|
+
};
|
|
354
|
+
}
|
|
355
|
+
}, "execute")
|
|
356
|
+
});
|
|
357
|
+
vercelTools.atp_explore_api = tool({
|
|
358
|
+
description: "Explore APIs using filesystem-like navigation. Navigate through directories to discover available functions.",
|
|
359
|
+
parameters: TOOL_SCHEMAS[ToolNames.EXPLORE_API],
|
|
360
|
+
execute: /* @__PURE__ */ __name(async ({ path }) => {
|
|
361
|
+
try {
|
|
362
|
+
const result = await underlyingClient.exploreAPI(path);
|
|
363
|
+
return {
|
|
364
|
+
success: true,
|
|
365
|
+
result
|
|
366
|
+
};
|
|
367
|
+
} catch (error) {
|
|
368
|
+
return {
|
|
369
|
+
success: false,
|
|
370
|
+
error: error.message
|
|
371
|
+
};
|
|
372
|
+
}
|
|
373
|
+
}, "execute")
|
|
374
|
+
});
|
|
375
|
+
vercelTools.atp_search_api = tool({
|
|
376
|
+
description: 'Search for APIs by keyword. Provide search term as string like "add", "math", "user", etc.',
|
|
377
|
+
parameters: TOOL_SCHEMAS[ToolNames.SEARCH_API],
|
|
378
|
+
execute: /* @__PURE__ */ __name(async ({ query }) => {
|
|
379
|
+
try {
|
|
380
|
+
const results = await underlyingClient.searchAPI(query);
|
|
381
|
+
return {
|
|
382
|
+
success: true,
|
|
383
|
+
results: results.map((r) => ({
|
|
384
|
+
apiGroup: r.apiGroup,
|
|
385
|
+
functionName: r.functionName,
|
|
386
|
+
description: r.description,
|
|
387
|
+
signature: r.signature
|
|
388
|
+
})),
|
|
389
|
+
count: results.length
|
|
390
|
+
};
|
|
391
|
+
} catch (error) {
|
|
392
|
+
return {
|
|
393
|
+
success: false,
|
|
394
|
+
error: error.message
|
|
395
|
+
};
|
|
396
|
+
}
|
|
397
|
+
}, "execute")
|
|
398
|
+
});
|
|
399
|
+
vercelTools.atp_get_type_definitions = tool({
|
|
400
|
+
description: "Get TypeScript type definitions for ATP runtime APIs to understand available functions",
|
|
401
|
+
parameters: z.object({}),
|
|
402
|
+
execute: /* @__PURE__ */ __name(async () => {
|
|
403
|
+
try {
|
|
404
|
+
const types = client.getTypeDefinitions();
|
|
405
|
+
return {
|
|
406
|
+
success: true,
|
|
407
|
+
types
|
|
408
|
+
};
|
|
409
|
+
} catch (error) {
|
|
410
|
+
return {
|
|
411
|
+
success: false,
|
|
412
|
+
error: error.message
|
|
413
|
+
};
|
|
414
|
+
}
|
|
415
|
+
}, "execute")
|
|
416
|
+
});
|
|
417
|
+
return {
|
|
418
|
+
client,
|
|
419
|
+
tools: vercelTools
|
|
420
|
+
};
|
|
421
|
+
}
|
|
422
|
+
__name(createATPTools, "createATPTools");
|
|
423
|
+
async function createATPStreamingTools(options) {
|
|
424
|
+
const { dataStream, defaultExecutionConfig, ...clientOptions } = options;
|
|
425
|
+
const client = new VercelAIATPClient(clientOptions);
|
|
426
|
+
await client.connect();
|
|
427
|
+
const eventHandler = createVercelEventHandler(dataStream);
|
|
428
|
+
const underlyingClient = client.getUnderlyingClient();
|
|
429
|
+
const vercelTools = {};
|
|
430
|
+
vercelTools.atp_execute_code = tool({
|
|
431
|
+
description: "Execute TypeScript code in ATP sandbox with streaming events for thinking, tool execution, and text output",
|
|
432
|
+
parameters: TOOL_SCHEMAS[ToolNames.EXECUTE_CODE],
|
|
433
|
+
execute: /* @__PURE__ */ __name(async ({ code }) => {
|
|
434
|
+
try {
|
|
435
|
+
const result = await underlyingClient.executeStream(code, defaultExecutionConfig, eventHandler);
|
|
436
|
+
if (result.status === ExecutionStatus.COMPLETED) {
|
|
437
|
+
return {
|
|
438
|
+
success: true,
|
|
439
|
+
result: result.result,
|
|
440
|
+
stats: result.stats
|
|
441
|
+
};
|
|
442
|
+
} else if (result.status === ExecutionStatus.FAILED) {
|
|
443
|
+
return {
|
|
444
|
+
success: false,
|
|
445
|
+
error: result.error,
|
|
446
|
+
stats: result.stats
|
|
447
|
+
};
|
|
448
|
+
} else {
|
|
449
|
+
return {
|
|
450
|
+
success: false,
|
|
451
|
+
error: "Execution in unexpected state: " + result.status
|
|
452
|
+
};
|
|
453
|
+
}
|
|
454
|
+
} catch (error) {
|
|
455
|
+
return {
|
|
456
|
+
success: false,
|
|
457
|
+
error: error.message || "Unknown error"
|
|
458
|
+
};
|
|
459
|
+
}
|
|
460
|
+
}, "execute")
|
|
461
|
+
});
|
|
462
|
+
vercelTools.atp_explore_api = tool({
|
|
463
|
+
description: "Explore APIs using filesystem-like navigation. Navigate through directories to discover available functions.",
|
|
464
|
+
parameters: TOOL_SCHEMAS[ToolNames.EXPLORE_API],
|
|
465
|
+
execute: /* @__PURE__ */ __name(async ({ path }) => {
|
|
466
|
+
try {
|
|
467
|
+
const result = await underlyingClient.exploreAPI(path);
|
|
468
|
+
return {
|
|
469
|
+
success: true,
|
|
470
|
+
result
|
|
471
|
+
};
|
|
472
|
+
} catch (error) {
|
|
473
|
+
return {
|
|
474
|
+
success: false,
|
|
475
|
+
error: error.message
|
|
476
|
+
};
|
|
477
|
+
}
|
|
478
|
+
}, "execute")
|
|
479
|
+
});
|
|
480
|
+
vercelTools.atp_search_api = tool({
|
|
481
|
+
description: 'Search for APIs by keyword. Provide search term as string like "add", "math", "user", etc.',
|
|
482
|
+
parameters: TOOL_SCHEMAS[ToolNames.SEARCH_API],
|
|
483
|
+
execute: /* @__PURE__ */ __name(async ({ query }) => {
|
|
484
|
+
try {
|
|
485
|
+
const results = await underlyingClient.searchAPI(query);
|
|
486
|
+
return {
|
|
487
|
+
success: true,
|
|
488
|
+
results: results.map((r) => ({
|
|
489
|
+
apiGroup: r.apiGroup,
|
|
490
|
+
functionName: r.functionName,
|
|
491
|
+
description: r.description,
|
|
492
|
+
signature: r.signature
|
|
493
|
+
})),
|
|
494
|
+
count: results.length
|
|
495
|
+
};
|
|
496
|
+
} catch (error) {
|
|
497
|
+
return {
|
|
498
|
+
success: false,
|
|
499
|
+
error: error.message
|
|
500
|
+
};
|
|
501
|
+
}
|
|
502
|
+
}, "execute")
|
|
503
|
+
});
|
|
504
|
+
vercelTools.atp_get_type_definitions = tool({
|
|
505
|
+
description: "Get TypeScript type definitions for ATP runtime APIs to understand available functions",
|
|
506
|
+
parameters: z.object({}),
|
|
507
|
+
execute: /* @__PURE__ */ __name(async () => {
|
|
508
|
+
try {
|
|
509
|
+
const types = client.getTypeDefinitions();
|
|
510
|
+
return {
|
|
511
|
+
success: true,
|
|
512
|
+
types
|
|
513
|
+
};
|
|
514
|
+
} catch (error) {
|
|
515
|
+
return {
|
|
516
|
+
success: false,
|
|
517
|
+
error: error.message
|
|
518
|
+
};
|
|
519
|
+
}
|
|
520
|
+
}, "execute")
|
|
521
|
+
});
|
|
522
|
+
return {
|
|
523
|
+
client,
|
|
524
|
+
tools: vercelTools
|
|
525
|
+
};
|
|
526
|
+
}
|
|
527
|
+
__name(createATPStreamingTools, "createATPStreamingTools");
|
|
528
|
+
|
|
529
|
+
export { VercelAIATPClient, createATPStreamingTools, createATPTools, createEventCollector, createVercelEventHandler };
|
|
530
|
+
//# sourceMappingURL=index.js.map
|
|
4
531
|
//# sourceMappingURL=index.js.map
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,uBAAuB,EAAE,MAAM,YAAY,CAAC;AACrE,OAAO,EACN,wBAAwB,EACxB,oBAAoB,GAIpB,MAAM,oBAAoB,CAAC"}
|
|
1
|
+
{"version":3,"sources":["../src/client.ts","../src/event-adapter.ts","../src/tools.ts"],"names":["VercelAIATPClient","client","model","embeddings","approvalHandler","constructor","options","tools","hooks","server","AgentToolProtocolClient","serviceProviders","undefined","serverUrl","baseUrl","headers","Error","provideLLM","call","prompt","handleLLMCall","extract","schema","handleLLMExtract","classify","text","categories","handleLLMClassify","provideEmbedding","embed","handleEmbedding","provideApproval","request","message","context","handleApprovalRequest","connect","init","name","version","getTypeDefinitions","execute","code","config","getUnderlyingClient","result","generateText","temperature","maxTokens","system","systemPrompt","generateObject","object","promptText","join","classification","trim","includes","category","toLowerCase","fallback","approved","timestamp","Date","now","createVercelEventHandler","dataStream","textRunIdPrefix","currentTextRunId","textRunCounter","reasoningRunId","reasoningCounter","toolStack","event","type","ATPEventType","THINKING","data","write","id","delta","content","TOOL_START","isFirstTool","length","push","toolName","apiGroup","transient","TOOL_END","pop","isLastTool","success","duration","error","TEXT","runId","TEXT_END","SOURCE","url","title","sourceId","providerMetadata","aiChat","summary","createdAt","PROGRESS","Math","round","fraction","ERROR","createEventCollector","events","handler","clear","TOOL_SCHEMAS","ToolNames","EXECUTE_CODE","executeCodeInputSchema","pick","EXPLORE_API","exploreApiInputSchema","SEARCH_API","searchApiInputSchema","FETCH_ALL_APIS","fetchAllApisInputSchema","createATPTools","defaultExecutionConfig","clientOptions","underlyingClient","vercelTools","atp_execute_code","tool","description","parameters","status","ExecutionStatus","COMPLETED","stats","FAILED","atp_explore_api","path","exploreAPI","atp_search_api","query","results","searchAPI","map","r","functionName","signature","count","atp_get_type_definitions","z","types","createATPStreamingTools","eventHandler","executeStream"],"mappings":";;;;;;;AAUO,IAAMA,oBAAN,MAAMA;EAVb;;;AAWSC,EAAAA,MAAAA;AACAC,EAAAA,KAAAA;AACAC,EAAAA,UAAAA;AACAC,EAAAA,eAAAA;AAERC,EAAAA,WAAAA,CAAYC,OAAAA,EAAmC;AAC9C,IAAA,MAAM,EAAEJ,KAAAA,EAAOC,UAAAA,EAAYI,KAAAA,EAAOH,eAAAA,EAAiBI,OAAK,GAAKF,OAAAA;AAE7D,IAAA,IAAI,QAAA,IAAYA,OAAAA,IAAWA,OAAAA,CAAQG,MAAAA,EAAQ;AAC1C,MAAA,IAAA,CAAKR,MAAAA,GAAS,IAAIS,uBAAAA,CAAwB;AACzCD,QAAAA,MAAAA,EAAQH,OAAAA,CAAQG,MAAAA;AAChBD,QAAAA,KAAAA;AACAG,QAAAA,gBAAAA,EAAkBJ,KAAAA,GAAQ;AAAEA,UAAAA;SAAM,GAAIK;OACvC,CAAA;IACD,CAAA,MAAA,IAAW,WAAA,IAAeN,OAAAA,IAAWA,OAAAA,CAAQO,SAAAA,EAAW;AACvD,MAAA,IAAA,CAAKZ,MAAAA,GAAS,IAAIS,uBAAAA,CAAwB;AACzCI,QAAAA,OAAAA,EAASR,OAAAA,CAAQO,SAAAA;AACjBE,QAAAA,OAAAA,EAAST,OAAAA,CAAQS,OAAAA;AACjBP,QAAAA,KAAAA;AACAG,QAAAA,gBAAAA,EAAkBJ,KAAAA,GAAQ;AAAEA,UAAAA;SAAM,GAAIK;OACvC,CAAA;IACD,CAAA,MAAO;AACN,MAAA,MAAM,IAAII,MAAM,6CAAA,CAAA;AACjB,IAAA;AAEA,IAAA,IAAA,CAAKd,KAAAA,GAAQA,KAAAA;AACb,IAAA,IAAA,CAAKC,UAAAA,GAAaA,UAAAA;AAClB,IAAA,IAAA,CAAKC,eAAAA,GAAkBA,eAAAA;AAEvB,IAAA,IAAA,CAAKH,OAAOgB,UAAAA,CAAW;MACtBC,IAAAA,kBAAM,MAAA,CAAA,OAAOC,QAAgBb,QAAAA,KAAAA;AAC5B,QAAA,OAAO,MAAM,IAAA,CAAKc,aAAAA,CAAcD,MAAAA,EAAQb,QAAAA,CAAAA;MACzC,CAAA,EAFM,MAAA,CAAA;MAGNe,OAAAA,kBAAS,MAAA,CAAA,OAAOF,MAAAA,EAAgBG,MAAAA,EAAahB,QAAAA,KAAAA;AAC5C,QAAA,OAAO,MAAM,IAAA,CAAKiB,gBAAAA,CAAiBJ,MAAAA,EAAQG,QAAQhB,QAAAA,CAAAA;MACpD,CAAA,EAFS,SAAA,CAAA;MAGTkB,QAAAA,kBAAU,MAAA,CAAA,OAAOC,IAAAA,EAAcC,UAAAA,EAAsBpB,QAAAA,KAAAA;AACpD,QAAA,OAAO,MAAM,IAAA,CAAKqB,iBAAAA,CAAkBF,IAAAA,EAAMC,YAAYpB,QAAAA,CAAAA;MACvD,CAAA,EAFU,UAAA;KAGX,CAAA;AAEA,IAAA,IAAI,KAAKH,UAAAA,EAAY;AACpB,MAAA,IAAA,CAAKF,OAAO2B,gBAAAA,CAAiB;AAC5BC,QAAAA,KAAAA,gCAAcJ,IAAAA,KAAAA;AACb,UAAA,OAAO,MAAM,IAAA,CAAKK,eAAAA,CAAgBL,IAAAA,CAAAA;QACnC,CAAA,EAFO,OAAA;OAGR,CAAA;AACD,IAAA;AAEA,IAAA,IAAA,CAAKxB,OAAO8B,eAAAA,CAAgB;MAC3BC,OAAAA,kBAAS,MAAA,CAAA,OAAOC,SAAiBC,OAAAA,KAAAA;AAChC,QAAA,OAAO,MAAM,IAAA,CAAKC,qBAAAA,CAAsBF,OAAAA,EAASC,OAAAA,CAAAA;MAClD,CAAA,EAFS,SAAA;KAGV,CAAA;AACD,EAAA;AAEA,EAAA,MAAME,OAAAA,GAAyB;AAC9B,IAAA,MAAM,IAAA,CAAKnC,OAAOoC,IAAAA,CAAK;MAAEC,IAAAA,EAAM,0BAAA;MAA4BC,OAAAA,EAAS;KAAQ,CAAA;AAC5E,IAAA,MAAM,IAAA,CAAKtC,OAAOmC,OAAAA,EAAO;AAC1B,EAAA;EAEAI,kBAAAA,GAA6B;AAC5B,IAAA,OAAO,IAAA,CAAKvC,OAAOuC,kBAAAA,EAAkB;AACtC,EAAA;EAEA,MAAMC,OAAAA,CAAQC,MAAcC,MAAAA,EAA6D;AACxF,IAAA,OAAO,MAAM,IAAA,CAAK1C,MAAAA,CAAOwC,OAAAA,CAAQC,MAAMC,MAAAA,CAAAA;AACxC,EAAA;EAEAC,mBAAAA,GAA+C;AAC9C,IAAA,OAAO,IAAA,CAAK3C,MAAAA;AACb,EAAA;EAEA,MAAcmB,aAAAA,CAAcD,QAAgBb,OAAAA,EAAgC;AAC3E,IAAA,MAAMuC,MAAAA,GAAS,MAAMC,YAAAA,CAAa;AACjC5C,MAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZiB,MAAAA,MAAAA;AACA4B,MAAAA,WAAAA,EAAazC,OAAAA,EAASyC,WAAAA;AACtBC,MAAAA,SAAAA,EAAW1C,OAAAA,EAAS0C,SAAAA;AACpBC,MAAAA,MAAAA,EAAQ3C,OAAAA,EAAS4C;KAClB,CAAA;AAEA,IAAA,OAAOL,MAAAA,CAAOpB,IAAAA;AACf,EAAA;EAEA,MAAcF,gBAAAA,CAAiBJ,MAAAA,EAAgBG,MAAAA,EAAahB,OAAAA,EAA6B;AACxF,IAAA,MAAMuC,MAAAA,GAAS,MAAMM,cAAAA,CAAe;AACnCjD,MAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;AACZiB,MAAAA,MAAAA;AACAG,MAAAA,MAAAA;AACA2B,MAAAA,MAAAA,EAAQ3C,OAAAA,EAAS4C,YAAAA;AACjBH,MAAAA,WAAAA,EAAazC,OAAAA,EAASyC,WAAAA;AACtBC,MAAAA,SAAAA,EAAW1C,OAAAA,EAAS0C;KACrB,CAAA;AAEA,IAAA,OAAOH,MAAAA,CAAOO,MAAAA;AACf,EAAA;EAEA,MAAczB,iBAAAA,CACbF,IAAAA,EACAC,UAAAA,EACApB,OAAAA,EACkB;AAClB,IAAA,MAAM+C,UAAAA,GAAa,CAAA,0DAAA,EAA6D3B,UAAAA,CAAW4B,IAAAA,CAAK,IAAA,CAAA;;QAAkB7B,IAAAA;;;AAElH,IAAA,MAAMoB,MAAAA,GAAS,MAAMC,YAAAA,CAAa;AACjC5C,MAAAA,KAAAA,EAAO,IAAA,CAAKA,KAAAA;MACZiB,MAAAA,EAAQkC,UAAAA;AACRJ,MAAAA,MAAAA,EAAQ3C,OAAAA,EAAS4C,YAAAA;MACjBH,WAAAA,EAAa;KACd,CAAA;AAEA,IAAA,MAAMQ,cAAAA,GAAiBV,MAAAA,CAAOpB,IAAAA,CAAK+B,IAAAA,EAAI;AAEvC,IAAA,IAAI9B,UAAAA,CAAW+B,QAAAA,CAASF,cAAAA,CAAAA,EAAiB;AACxC,MAAA,OAAOA,cAAAA;AACR,IAAA;AAEA,IAAA,KAAA,MAAWG,YAAYhC,UAAAA,EAAY;AAClC,MAAA,IAAI6B,eAAeI,WAAAA,EAAW,CAAGF,SAASC,QAAAA,CAASC,WAAAA,EAAW,CAAA,EAAK;AAClE,QAAA,OAAOD,QAAAA;AACR,MAAA;AACD,IAAA;AAEA,IAAA,MAAME,QAAAA,GAAWlC,WAAW,CAAA,CAAA;AAC5B,IAAA,IAAI,CAACkC,QAAAA,EAAU;AACd,MAAA,MAAM,IAAI5C,MAAM,2CAAA,CAAA;AACjB,IAAA;AACA,IAAA,OAAO4C,QAAAA;AACR,EAAA;AAEA,EAAA,MAAc9B,gBAAgBL,IAAAA,EAAiC;AAC9D,IAAA,IAAI,CAAC,KAAKtB,UAAAA,EAAY;AACrB,MAAA,MAAM,IAAIa,MACT,6FAAA,CAAA;AAEF,IAAA;AAEA,IAAA,OAAO,MAAM,IAAA,CAAKb,UAAAA,CAAW0B,KAAAA,CAAMJ,IAAAA,CAAAA;AACpC,EAAA;EAEA,MAAcU,qBAAAA,CACbF,SACAC,OAAAA,EAC4B;AAC5B,IAAA,IAAI,CAAC,KAAK9B,eAAAA,EAAiB;AAC1B,MAAA,MAAM,IAAIY,MACT,8FAAA,CAAA;AAEF,IAAA;AAEA,IAAA,MAAM6C,QAAAA,GAAW,MAAM,IAAA,CAAKzD,eAAAA,CAAgB6B,SAASC,OAAAA,CAAAA;AACrD,IAAA,OAAO;AACN2B,MAAAA,QAAAA;AACAC,MAAAA,SAAAA,EAAWC,KAAKC,GAAAA;AACjB,KAAA;AACD,EAAA;AACD;AC7GO,SAASC,wBAAAA,CACfC,UAAAA,EACA5D,OAAAA,GAA2C,EAAC,EAAC;AAE7C,EAAA,MAAM,EAAE6D,eAAAA,GAAkB,UAAA,EAAU,GAAK7D,OAAAA;AAEzC,EAAA,IAAI8D,gBAAAA;AACJ,EAAA,IAAIC,cAAAA,GAAiB,CAAA;AAGrB,EAAA,IAAIC,cAAAA;AACJ,EAAA,IAAIC,gBAAAA,GAAmB,CAAA;AAGvB,EAAA,MAAMC,YAA2D,EAAA;AAEjE,EAAA,OAAO,CAACC,KAAAA,KAAAA;AACP,IAAA,QAAQA,MAAMC,IAAAA;AACb,MAAA,KAAKC,aAAaC,QAAAA,EAAU;AAC3B,QAAA,MAAMC,OAAOJ,KAAAA,CAAMI,IAAAA;AAGnB,QAAA,IAAI,CAACP,cAAAA,EAAgB;AACpBA,UAAAA,cAAAA,GAAiB,CAAA,cAAA,EAAiB,EAAEC,gBAAAA,CAAAA,CAAAA;AACpCL,UAAAA,UAAAA,CAAWY,KAAAA,CAAM;YAChBJ,IAAAA,EAAM,iBAAA;YACNK,EAAAA,EAAIT;WACL,CAAA;AACD,QAAA;AAGAJ,QAAAA,UAAAA,CAAWY,KAAAA,CAAM;UAChBJ,IAAAA,EAAM,iBAAA;UACNK,EAAAA,EAAIT,cAAAA;AACJU,UAAAA,KAAAA,EAAOH,KAAKI,OAAAA,GAAU;SACvB,CAAA;AACA,QAAA;AACD,MAAA;AAEA,MAAA,KAAKN,aAAaO,UAAAA,EAAY;AAC7B,QAAA,MAAML,OAAOJ,KAAAA,CAAMI,IAAAA;AACnB,QAAA,MAAMM,WAAAA,GAAcX,UAAUY,MAAAA,KAAW,CAAA;AAEzCZ,QAAAA,SAAAA,CAAUa,IAAAA,CAAK;AAAEC,UAAAA,QAAAA,EAAUT,IAAAA,CAAKS,QAAAA;AAAUC,UAAAA,QAAAA,EAAUV,IAAAA,CAAKU;SAAS,CAAA;AAElE,QAAA,IAAIJ,WAAAA,EAAa;AAEhBjB,UAAAA,UAAAA,CAAWY,KAAAA,CAAM;YAChBJ,IAAAA,EAAM,qBAAA;AACNG,YAAAA,IAAAA,EAAM,CAAA,UAAA,EAAaA,IAAAA,CAAKU,QAAQ,CAAA,CAAA,EAAIV,KAAKS,QAAQ,CAAA,GAAA,CAAA;YACjDE,SAAAA,EAAW;WACZ,CAAA;QACD,CAAA,MAAO;AAENtB,UAAAA,UAAAA,CAAWY,KAAAA,CAAM;YAChBJ,IAAAA,EAAM,gBAAA;AACNG,YAAAA,IAAAA,EAAM,CAAA,OAAA,EAAKA,IAAAA,CAAKU,QAAQ,CAAA,CAAA,EAAIV,KAAKS,QAAQ,CAAA,CAAA;YACzCE,SAAAA,EAAW;WACZ,CAAA;AACD,QAAA;AACA,QAAA;AACD,MAAA;AAEA,MAAA,KAAKb,aAAac,QAAAA,EAAU;AAC3B,QAAA,MAAMZ,OAAOJ,KAAAA,CAAMI,IAAAA;AASnBL,QAAAA,SAAAA,CAAUkB,GAAAA,EAAG;AACb,QAAA,MAAMC,UAAAA,GAAanB,UAAUY,MAAAA,KAAW,CAAA;AAGxC,QAAA,IAAIO,cAAcrB,cAAAA,EAAgB;AACjCJ,UAAAA,UAAAA,CAAWY,KAAAA,CAAM;YAChBJ,IAAAA,EAAM,eAAA;YACNK,EAAAA,EAAIT;WACL,CAAA;AACAA,UAAAA,cAAAA,GAAiB1D,MAAAA;AAClB,QAAA;AAEA,QAAA,IAAI+E,UAAAA,EAAY;AAEfzB,UAAAA,UAAAA,CAAWY,KAAAA,CAAM;YAChBJ,IAAAA,EAAM,mBAAA;YACNG,IAAAA,EAAMA,IAAAA,CAAKe,UACR,CAAA,UAAA,EAAaf,IAAAA,CAAKU,QAAQ,CAAA,CAAA,EAAIV,IAAAA,CAAKS,QAAQ,CAAA,EAAA,EAAKT,IAAAA,CAAKgB,QAAQ,CAAA,GAAA,CAAA,GAC7D,CAAA,OAAA,EAAUhB,KAAKU,QAAQ,CAAA,CAAA,EAAIV,KAAKS,QAAQ,CAAA,EAAA,EAAKT,KAAKiB,KAAK,CAAA,CAAA;YAC1DN,SAAAA,EAAW;WACZ,CAAA;QACD,CAAA,MAAO;AAENtB,UAAAA,UAAAA,CAAWY,KAAAA,CAAM;YAChBJ,IAAAA,EAAM,gBAAA;YACNG,IAAAA,EAAMA,IAAAA,CAAKe,UACR,CAAA,OAAA,EAAKf,IAAAA,CAAKU,QAAQ,CAAA,CAAA,EAAIV,IAAAA,CAAKS,QAAQ,CAAA,EAAA,EAAKT,IAAAA,CAAKgB,QAAQ,CAAA,GAAA,CAAA,GACrD,CAAA,OAAA,EAAKhB,KAAKU,QAAQ,CAAA,CAAA,EAAIV,KAAKS,QAAQ,CAAA,EAAA,EAAKT,KAAKiB,KAAK,CAAA,CAAA;YACrDN,SAAAA,EAAW;WACZ,CAAA;AACD,QAAA;AACA,QAAA;AACD,MAAA;AAEA,MAAA,KAAKb,aAAaoB,IAAAA,EAAM;AACvB,QAAA,MAAMlB,OAAOJ,KAAAA,CAAMI,IAAAA;AACnB,QAAA,MAAMmB,QAAQvB,KAAAA,CAAMuB,KAAAA,IAAS,GAAG7B,eAAAA,CAAAA,CAAAA,EAAmB,EAAEE,cAAAA,CAAAA,CAAAA;AAGrD,QAAA,IAAID,gBAAAA,IAAoBA,qBAAqB4B,KAAAA,EAAO;AACnD9B,UAAAA,UAAAA,CAAWY,KAAAA,CAAM;YAAEJ,IAAAA,EAAM,UAAA;YAAYK,EAAAA,EAAIX;WAAiB,CAAA;AAC1DA,UAAAA,gBAAAA,GAAmBxD,MAAAA;AACpB,QAAA;AAGA,QAAA,IAAI,CAACwD,gBAAAA,IAAoBA,gBAAAA,KAAqB4B,KAAAA,EAAO;AACpD5B,UAAAA,gBAAAA,GAAmB4B,KAAAA;AACnB9B,UAAAA,UAAAA,CAAWY,KAAAA,CAAM;YAAEJ,IAAAA,EAAM,YAAA;YAAcK,EAAAA,EAAIX;WAAiB,CAAA;AAC7D,QAAA;AAGAF,QAAAA,UAAAA,CAAWY,KAAAA,CAAM;UAChBJ,IAAAA,EAAM,YAAA;UACNK,EAAAA,EAAIX,gBAAAA;AACJY,UAAAA,KAAAA,EAAOH,IAAAA,CAAKpD;SACb,CAAA;AACA,QAAA;AACD,MAAA;AAEA,MAAA,KAAKkD,YAAAA,CAAasB,QAAAA;AACjB,QAAA,IAAI7B,gBAAAA,EAAkB;AACrBF,UAAAA,UAAAA,CAAWY,KAAAA,CAAM;YAAEJ,IAAAA,EAAM,UAAA;YAAYK,EAAAA,EAAIX;WAAiB,CAAA;AAC1DA,UAAAA,gBAAAA,GAAmBxD,MAAAA;AACpB,QAAA;AACA,QAAA;AAED,MAAA,KAAK+D,aAAauB,MAAAA,EAAQ;AACzB,QAAA,MAAMrB,OAAOJ,KAAAA,CAAMI,IAAAA;AAMnBX,QAAAA,UAAAA,CAAWY,KAAAA,CAAM;UAChBJ,IAAAA,EAAM,YAAA;AACNyB,UAAAA,GAAAA,EAAKtB,IAAAA,CAAKsB,GAAAA;AACVC,UAAAA,KAAAA,EAAOvB,IAAAA,CAAKuB,KAAAA;AACZC,UAAAA,QAAAA,EAAU,CAAA,EAAGxB,IAAAA,CAAKuB,KAAK,CAAA,CAAA,EAAI3B,MAAMX,SAAS,CAAA,CAAA;UAC1CwC,gBAAAA,EAAkB;YACjBC,MAAAA,EAAQ;AACPC,cAAAA,OAAAA,EAAS3B,IAAAA,CAAK2B,OAAAA;AACdC,cAAAA,SAAAA,EAAW5B,IAAAA,CAAK4B;AACjB;AACD;SACD,CAAA;AACA,QAAA;AACD,MAAA;AAEA,MAAA,KAAK9B,aAAa+B,QAAAA,EAAU;AAC3B,QAAA,MAAM7B,OAAOJ,KAAAA,CAAMI,IAAAA;AACnBX,QAAAA,UAAAA,CAAWY,KAAAA,CAAM;UAChBJ,IAAAA,EAAM,gBAAA;UACNG,IAAAA,EAAM,CAAA,EAAGA,KAAK5C,OAAO,CAAA,EAAA,EAAK0E,KAAKC,KAAAA,CAAM/B,IAAAA,CAAKgC,QAAAA,GAAW,GAAA,CAAA,CAAA,EAAA,CAAA;UACrDrB,SAAAA,EAAW;SACZ,CAAA;AACA,QAAA;AACD,MAAA;AAEA,MAAA,KAAKb,aAAamC,KAAAA,EAAO;AACxB,QAAA,MAAMjC,OAAOJ,KAAAA,CAAMI,IAAAA;AACnBX,QAAAA,UAAAA,CAAWY,KAAAA,CAAM;UAChBJ,IAAAA,EAAM,YAAA;AACNG,UAAAA,IAAAA,EAAMA,IAAAA,CAAK5C;SACZ,CAAA;AACA,QAAA;AACD,MAAA;AAEA,MAAA;AAECiC,QAAAA,UAAAA,CAAWY,KAAAA,CAAM;UAChBJ,IAAAA,EAAM,CAAA,KAAA,EAAQD,MAAMC,IAAI,CAAA,CAAA;AACxBG,UAAAA,IAAAA,EAAMJ,KAAAA,CAAMI,IAAAA;UACZW,SAAAA,EAAW;SACZ,CAAA;AACA,QAAA;AACF;AACD,EAAA,CAAA;AACD;AA9LgBvB,MAAAA,CAAAA,wBAAAA,EAAAA,0BAAAA,CAAAA;AAsMT,SAAS8C,oBAAAA,GAAAA;AAKf,EAAA,MAAMC,SAAqB,EAAA;AAE3B,EAAA,OAAO;AACNC,IAAAA,OAAAA,0BAAUxC,KAAAA,KAAAA;AACTuC,MAAAA,MAAAA,CAAO3B,KAAKZ,KAAAA,CAAAA;IACb,CAAA,EAFS,SAAA,CAAA;AAGTuC,IAAAA,MAAAA;AACAE,IAAAA,KAAAA,kBAAO,MAAA,CAAA,MAAA;AACNF,MAAAA,MAAAA,CAAO5B,MAAAA,GAAS,CAAA;IACjB,CAAA,EAFO,OAAA;AAGR,GAAA;AACD;AAhBgB2B,MAAAA,CAAAA,oBAAAA,EAAAA,sBAAAA,CAAAA;;;ACnPhB,IAAMI,YAAAA,GAAe;AACpB,EAAA,CAACC,SAAAA,CAAUC,YAAY,GAAGC,sBAAAA,CAAuBC,IAAAA,CAAK;IAAE7E,IAAAA,EAAM;GAAK,CAAA;EACnE,CAAC0E,SAAAA,CAAUI,WAAW,GAAGC,qBAAAA;EACzB,CAACL,SAAAA,CAAUM,UAAU,GAAGC,oBAAAA;EACxB,CAACP,SAAAA,CAAUQ,cAAc,GAAGC;AAC7B,CAAA;AAEA,eAAsBC,eAAexH,OAAAA,EAA8B;AAClE,EAAA,MAAM,EAAEyH,sBAAAA,EAAwB,GAAGC,aAAAA,EAAAA,GAAkB1H,OAAAA;AAErD,EAAA,MAAML,MAAAA,GAAS,IAAID,iBAAAA,CAAkBgI,aAAAA,CAAAA;AACrC,EAAA,MAAM/H,OAAOmC,OAAAA,EAAO;AAEpB,EAAA,MAAM6F,gBAAAA,GAAmBhI,OAAO2C,mBAAAA,EAAmB;AACnD,EAAA,MAAMsF,cAAmC,EAAC;AAE1CA,EAAAA,WAAAA,CAAYC,mBAAmBC,IAAAA,CAAK;IACnCC,WAAAA,EACC,iHAAA;IACDC,UAAAA,EAAYnB,YAAAA,CAAaC,UAAUC,YAAY,CAAA;IAC/C5E,OAAAA,kBAAS,MAAA,CAAA,OAAO,EAAEC,IAAAA,EAAI,KAAoB;AACzC,MAAA,IAAI;AACH,QAAA,MAAMG,MAAAA,GAAS,MAAM5C,MAAAA,CAAOwC,OAAAA,CAAQC,MAAMqF,sBAAAA,CAAAA;AAE1C,QAAA,IAAIlF,MAAAA,CAAO0F,MAAAA,KAAWC,eAAAA,CAAgBC,SAAAA,EAAW;AAChD,UAAA,OAAO;YACN7C,OAAAA,EAAS,IAAA;AACT/C,YAAAA,MAAAA,EAAQA,MAAAA,CAAOA,MAAAA;AACf6F,YAAAA,KAAAA,EAAO7F,MAAAA,CAAO6F;AACf,WAAA;QACD,CAAA,MAAA,IAAW7F,MAAAA,CAAO0F,MAAAA,KAAWC,eAAAA,CAAgBG,MAAAA,EAAQ;AACpD,UAAA,OAAO;YACN/C,OAAAA,EAAS,KAAA;AACTE,YAAAA,KAAAA,EAAOjD,MAAAA,CAAOiD,KAAAA;AACd4C,YAAAA,KAAAA,EAAO7F,MAAAA,CAAO6F;AACf,WAAA;QACD,CAAA,MAAO;AACN,UAAA,OAAO;YACN9C,OAAAA,EAAS,KAAA;AACTE,YAAAA,KAAAA,EAAO,oCAAoCjD,MAAAA,CAAO0F;AACnD,WAAA;AACD,QAAA;AACD,MAAA,CAAA,CAAA,OAASzC,KAAAA,EAAY;AACpB,QAAA,OAAO;UACNF,OAAAA,EAAS,KAAA;AACTE,UAAAA,KAAAA,EAAOA,MAAM7D,OAAAA,IAAW;AACzB,SAAA;AACD,MAAA;IACD,CAAA,EA5BS,SAAA;GA6BV,CAAA;AAEAiG,EAAAA,WAAAA,CAAYU,kBAAkBR,IAAAA,CAAK;IAClCC,WAAAA,EACC,8GAAA;IACDC,UAAAA,EAAYnB,YAAAA,CAAaC,UAAUI,WAAW,CAAA;IAC9C/E,OAAAA,kBAAS,MAAA,CAAA,OAAO,EAAEoG,IAAAA,EAAI,KAAoB;AACzC,MAAA,IAAI;AACH,QAAA,MAAMhG,MAAAA,GAAS,MAAMoF,gBAAAA,CAAiBa,UAAAA,CAAWD,IAAAA,CAAAA;AACjD,QAAA,OAAO;UACNjD,OAAAA,EAAS,IAAA;AACT/C,UAAAA;AACD,SAAA;AACD,MAAA,CAAA,CAAA,OAASiD,KAAAA,EAAY;AACpB,QAAA,OAAO;UACNF,OAAAA,EAAS,KAAA;AACTE,UAAAA,KAAAA,EAAOA,KAAAA,CAAM7D;AACd,SAAA;AACD,MAAA;IACD,CAAA,EAbS,SAAA;GAcV,CAAA;AAEAiG,EAAAA,WAAAA,CAAYa,iBAAiBX,IAAAA,CAAK;IACjCC,WAAAA,EACC,4FAAA;IACDC,UAAAA,EAAYnB,YAAAA,CAAaC,UAAUM,UAAU,CAAA;IAC7CjF,OAAAA,kBAAS,MAAA,CAAA,OAAO,EAAEuG,KAAAA,EAAK,KAAqB;AAC3C,MAAA,IAAI;AACH,QAAA,MAAMC,OAAAA,GAAU,MAAMhB,gBAAAA,CAAiBiB,SAAAA,CAAUF,KAAAA,CAAAA;AACjD,QAAA,OAAO;UACNpD,OAAAA,EAAS,IAAA;UACTqD,OAAAA,EAASA,OAAAA,CAAQE,GAAAA,CAAI,CAACC,CAAAA,MAAY;AACjC7D,YAAAA,QAAAA,EAAU6D,CAAAA,CAAE7D,QAAAA;AACZ8D,YAAAA,YAAAA,EAAcD,CAAAA,CAAEC,YAAAA;AAChBhB,YAAAA,WAAAA,EAAae,CAAAA,CAAEf,WAAAA;AACfiB,YAAAA,SAAAA,EAAWF,CAAAA,CAAEE;WACd,CAAA,CAAA;AACAC,UAAAA,KAAAA,EAAON,OAAAA,CAAQ7D;AAChB,SAAA;AACD,MAAA,CAAA,CAAA,OAASU,KAAAA,EAAY;AACpB,QAAA,OAAO;UACNF,OAAAA,EAAS,KAAA;AACTE,UAAAA,KAAAA,EAAOA,KAAAA,CAAM7D;AACd,SAAA;AACD,MAAA;IACD,CAAA,EAnBS,SAAA;GAoBV,CAAA;AAEAiG,EAAAA,WAAAA,CAAYsB,2BAA2BpB,IAAAA,CAAK;IAC3CC,WAAAA,EACC,wFAAA;IACDC,UAAAA,EAAYmB,CAAAA,CAAErG,MAAAA,CAAO,EAAC,CAAA;AACtBX,IAAAA,OAAAA,kBAAS,MAAA,CAAA,YAAA;AACR,MAAA,IAAI;AACH,QAAA,MAAMiH,KAAAA,GAAQzJ,OAAOuC,kBAAAA,EAAkB;AACvC,QAAA,OAAO;UACNoD,OAAAA,EAAS,IAAA;AACT8D,UAAAA;AACD,SAAA;AACD,MAAA,CAAA,CAAA,OAAS5D,KAAAA,EAAY;AACpB,QAAA,OAAO;UACNF,OAAAA,EAAS,KAAA;AACTE,UAAAA,KAAAA,EAAOA,KAAAA,CAAM7D;AACd,SAAA;AACD,MAAA;IACD,CAAA,EAbS,SAAA;GAcV,CAAA;AAEA,EAAA,OAAO;AACNhC,IAAAA,MAAAA;IACAM,KAAAA,EAAO2H;AACR,GAAA;AACD;AAlHsBJ,MAAAA,CAAAA,cAAAA,EAAAA,gBAAAA,CAAAA;AA8ItB,eAAsB6B,wBACrBrJ,OAAAA,EAA8B;AAE9B,EAAA,MAAM,EAAE4D,UAAAA,EAAY6D,sBAAAA,EAAwB,GAAGC,eAAAA,GAAkB1H,OAAAA;AAEjE,EAAA,MAAML,MAAAA,GAAS,IAAID,iBAAAA,CAAkBgI,aAAAA,CAAAA;AACrC,EAAA,MAAM/H,OAAOmC,OAAAA,EAAO;AAEpB,EAAA,MAAMwH,YAAAA,GAAe3F,yBAAyBC,UAAAA,CAAAA;AAC9C,EAAA,MAAM+D,gBAAAA,GAAmBhI,OAAO2C,mBAAAA,EAAmB;AAEnD,EAAA,MAAMsF,cAAmC,EAAC;AAE1CA,EAAAA,WAAAA,CAAYC,mBAAmBC,IAAAA,CAAK;IACnCC,WAAAA,EACC,4GAAA;IACDC,UAAAA,EAAYnB,YAAAA,CAAaC,UAAUC,YAAY,CAAA;IAC/C5E,OAAAA,kBAAS,MAAA,CAAA,OAAO,EAAEC,IAAAA,EAAI,KAAoB;AACzC,MAAA,IAAI;AACH,QAAA,MAAMG,SAAS,MAAMoF,gBAAAA,CAAiB4B,aAAAA,CACrCnH,IAAAA,EACAqF,wBACA6B,YAAAA,CAAAA;AAGD,QAAA,IAAI/G,MAAAA,CAAO0F,MAAAA,KAAWC,eAAAA,CAAgBC,SAAAA,EAAW;AAChD,UAAA,OAAO;YACN7C,OAAAA,EAAS,IAAA;AACT/C,YAAAA,MAAAA,EAAQA,MAAAA,CAAOA,MAAAA;AACf6F,YAAAA,KAAAA,EAAO7F,MAAAA,CAAO6F;AACf,WAAA;QACD,CAAA,MAAA,IAAW7F,MAAAA,CAAO0F,MAAAA,KAAWC,eAAAA,CAAgBG,MAAAA,EAAQ;AACpD,UAAA,OAAO;YACN/C,OAAAA,EAAS,KAAA;AACTE,YAAAA,KAAAA,EAAOjD,MAAAA,CAAOiD,KAAAA;AACd4C,YAAAA,KAAAA,EAAO7F,MAAAA,CAAO6F;AACf,WAAA;QACD,CAAA,MAAO;AACN,UAAA,OAAO;YACN9C,OAAAA,EAAS,KAAA;AACTE,YAAAA,KAAAA,EAAO,oCAAoCjD,MAAAA,CAAO0F;AACnD,WAAA;AACD,QAAA;AACD,MAAA,CAAA,CAAA,OAASzC,KAAAA,EAAY;AACpB,QAAA,OAAO;UACNF,OAAAA,EAAS,KAAA;AACTE,UAAAA,KAAAA,EAAOA,MAAM7D,OAAAA,IAAW;AACzB,SAAA;AACD,MAAA;IACD,CAAA,EAhCS,SAAA;GAiCV,CAAA;AAEAiG,EAAAA,WAAAA,CAAYU,kBAAkBR,IAAAA,CAAK;IAClCC,WAAAA,EACC,8GAAA;IACDC,UAAAA,EAAYnB,YAAAA,CAAaC,UAAUI,WAAW,CAAA;IAC9C/E,OAAAA,kBAAS,MAAA,CAAA,OAAO,EAAEoG,IAAAA,EAAI,KAAoB;AACzC,MAAA,IAAI;AACH,QAAA,MAAMhG,MAAAA,GAAS,MAAMoF,gBAAAA,CAAiBa,UAAAA,CAAWD,IAAAA,CAAAA;AACjD,QAAA,OAAO;UACNjD,OAAAA,EAAS,IAAA;AACT/C,UAAAA;AACD,SAAA;AACD,MAAA,CAAA,CAAA,OAASiD,KAAAA,EAAY;AACpB,QAAA,OAAO;UACNF,OAAAA,EAAS,KAAA;AACTE,UAAAA,KAAAA,EAAOA,KAAAA,CAAM7D;AACd,SAAA;AACD,MAAA;IACD,CAAA,EAbS,SAAA;GAcV,CAAA;AAEAiG,EAAAA,WAAAA,CAAYa,iBAAiBX,IAAAA,CAAK;IACjCC,WAAAA,EACC,4FAAA;IACDC,UAAAA,EAAYnB,YAAAA,CAAaC,UAAUM,UAAU,CAAA;IAC7CjF,OAAAA,kBAAS,MAAA,CAAA,OAAO,EAAEuG,KAAAA,EAAK,KAAqB;AAC3C,MAAA,IAAI;AACH,QAAA,MAAMC,OAAAA,GAAU,MAAMhB,gBAAAA,CAAiBiB,SAAAA,CAAUF,KAAAA,CAAAA;AACjD,QAAA,OAAO;UACNpD,OAAAA,EAAS,IAAA;UACTqD,OAAAA,EAASA,OAAAA,CAAQE,GAAAA,CAAI,CAACC,CAAAA,MAAY;AACjC7D,YAAAA,QAAAA,EAAU6D,CAAAA,CAAE7D,QAAAA;AACZ8D,YAAAA,YAAAA,EAAcD,CAAAA,CAAEC,YAAAA;AAChBhB,YAAAA,WAAAA,EAAae,CAAAA,CAAEf,WAAAA;AACfiB,YAAAA,SAAAA,EAAWF,CAAAA,CAAEE;WACd,CAAA,CAAA;AACAC,UAAAA,KAAAA,EAAON,OAAAA,CAAQ7D;AAChB,SAAA;AACD,MAAA,CAAA,CAAA,OAASU,KAAAA,EAAY;AACpB,QAAA,OAAO;UACNF,OAAAA,EAAS,KAAA;AACTE,UAAAA,KAAAA,EAAOA,KAAAA,CAAM7D;AACd,SAAA;AACD,MAAA;IACD,CAAA,EAnBS,SAAA;GAoBV,CAAA;AAEAiG,EAAAA,WAAAA,CAAYsB,2BAA2BpB,IAAAA,CAAK;IAC3CC,WAAAA,EACC,wFAAA;IACDC,UAAAA,EAAYmB,CAAAA,CAAErG,MAAAA,CAAO,EAAC,CAAA;AACtBX,IAAAA,OAAAA,kBAAS,MAAA,CAAA,YAAA;AACR,MAAA,IAAI;AACH,QAAA,MAAMiH,KAAAA,GAAQzJ,OAAOuC,kBAAAA,EAAkB;AACvC,QAAA,OAAO;UACNoD,OAAAA,EAAS,IAAA;AACT8D,UAAAA;AACD,SAAA;AACD,MAAA,CAAA,CAAA,OAAS5D,KAAAA,EAAY;AACpB,QAAA,OAAO;UACNF,OAAAA,EAAS,KAAA;AACTE,UAAAA,KAAAA,EAAOA,KAAAA,CAAM7D;AACd,SAAA;AACD,MAAA;IACD,CAAA,EAbS,SAAA;GAcV,CAAA;AAEA,EAAA,OAAO;AACNhC,IAAAA,MAAAA;IACAM,KAAAA,EAAO2H;AACR,GAAA;AACD;AA1HsByB,MAAAA,CAAAA,uBAAAA,EAAAA,yBAAAA,CAAAA","file":"index.js","sourcesContent":["import { AgentToolProtocolClient } from '@mondaydotcomorg/atp-client';\nimport type { ExecutionResult, ExecutionConfig } from '@mondaydotcomorg/atp-protocol';\nimport { generateObject, generateText } from 'ai';\nimport type {\n\tVercelAIATPClientOptions,\n\tApprovalResponse,\n\tEmbeddingProvider,\n\tApprovalHandler,\n} from './types.js';\n\nexport class VercelAIATPClient {\n\tprivate client: AgentToolProtocolClient;\n\tprivate model: any;\n\tprivate embeddings?: EmbeddingProvider;\n\tprivate approvalHandler?: ApprovalHandler;\n\n\tconstructor(options: VercelAIATPClientOptions) {\n\t\tconst { model, embeddings, tools, approvalHandler, hooks } = options;\n\n\t\tif ('server' in options && options.server) {\n\t\t\tthis.client = new AgentToolProtocolClient({\n\t\t\t\tserver: options.server,\n\t\t\t\thooks,\n\t\t\t\tserviceProviders: tools ? { tools } : undefined,\n\t\t\t});\n\t\t} else if ('serverUrl' in options && options.serverUrl) {\n\t\t\tthis.client = new AgentToolProtocolClient({\n\t\t\t\tbaseUrl: options.serverUrl,\n\t\t\t\theaders: options.headers,\n\t\t\t\thooks,\n\t\t\t\tserviceProviders: tools ? { tools } : undefined,\n\t\t\t});\n\t\t} else {\n\t\t\tthrow new Error('Either serverUrl or server must be provided');\n\t\t}\n\n\t\tthis.model = model;\n\t\tthis.embeddings = embeddings;\n\t\tthis.approvalHandler = approvalHandler;\n\n\t\tthis.client.provideLLM({\n\t\t\tcall: async (prompt: string, options?: any) => {\n\t\t\t\treturn await this.handleLLMCall(prompt, options);\n\t\t\t},\n\t\t\textract: async (prompt: string, schema: any, options?: any) => {\n\t\t\t\treturn await this.handleLLMExtract(prompt, schema, options);\n\t\t\t},\n\t\t\tclassify: async (text: string, categories: string[], options?: any) => {\n\t\t\t\treturn await this.handleLLMClassify(text, categories, options);\n\t\t\t},\n\t\t});\n\n\t\tif (this.embeddings) {\n\t\t\tthis.client.provideEmbedding({\n\t\t\t\tembed: async (text: string) => {\n\t\t\t\t\treturn await this.handleEmbedding(text);\n\t\t\t\t},\n\t\t\t});\n\t\t}\n\n\t\tthis.client.provideApproval({\n\t\t\trequest: async (message: string, context?: Record<string, unknown>) => {\n\t\t\t\treturn await this.handleApprovalRequest(message, context);\n\t\t\t},\n\t\t});\n\t}\n\n\tasync connect(): Promise<void> {\n\t\tawait this.client.init({ name: 'vercel-ai-sdk-atp-client', version: '1.0.0' });\n\t\tawait this.client.connect();\n\t}\n\n\tgetTypeDefinitions(): string {\n\t\treturn this.client.getTypeDefinitions();\n\t}\n\n\tasync execute(code: string, config?: Partial<ExecutionConfig>): Promise<ExecutionResult> {\n\t\treturn await this.client.execute(code, config);\n\t}\n\n\tgetUnderlyingClient(): AgentToolProtocolClient {\n\t\treturn this.client;\n\t}\n\n\tprivate async handleLLMCall(prompt: string, options?: any): Promise<string> {\n\t\tconst result = await generateText({\n\t\t\tmodel: this.model,\n\t\t\tprompt,\n\t\t\ttemperature: options?.temperature,\n\t\t\tmaxTokens: options?.maxTokens,\n\t\t\tsystem: options?.systemPrompt,\n\t\t});\n\n\t\treturn result.text;\n\t}\n\n\tprivate async handleLLMExtract(prompt: string, schema: any, options?: any): Promise<any> {\n\t\tconst result = await generateObject({\n\t\t\tmodel: this.model,\n\t\t\tprompt,\n\t\t\tschema,\n\t\t\tsystem: options?.systemPrompt,\n\t\t\ttemperature: options?.temperature,\n\t\t\tmaxTokens: options?.maxTokens,\n\t\t});\n\n\t\treturn result.object;\n\t}\n\n\tprivate async handleLLMClassify(\n\t\ttext: string,\n\t\tcategories: string[],\n\t\toptions?: any\n\t): Promise<string> {\n\t\tconst promptText = `Classify the following text into one of these categories: ${categories.join(', ')}\\n\\nText: ${text}\\n\\nRespond with ONLY the category name, nothing else.`;\n\n\t\tconst result = await generateText({\n\t\t\tmodel: this.model,\n\t\t\tprompt: promptText,\n\t\t\tsystem: options?.systemPrompt,\n\t\t\ttemperature: 0,\n\t\t});\n\n\t\tconst classification = result.text.trim();\n\n\t\tif (categories.includes(classification)) {\n\t\t\treturn classification;\n\t\t}\n\n\t\tfor (const category of categories) {\n\t\t\tif (classification.toLowerCase().includes(category.toLowerCase())) {\n\t\t\t\treturn category;\n\t\t\t}\n\t\t}\n\n\t\tconst fallback = categories[0];\n\t\tif (!fallback) {\n\t\t\tthrow new Error('No categories provided for classification');\n\t\t}\n\t\treturn fallback;\n\t}\n\n\tprivate async handleEmbedding(text: string): Promise<number[]> {\n\t\tif (!this.embeddings) {\n\t\t\tthrow new Error(\n\t\t\t\t'Embeddings provider not configured. Pass embeddings option when creating VercelAIATPClient.'\n\t\t\t);\n\t\t}\n\n\t\treturn await this.embeddings.embed(text);\n\t}\n\n\tprivate async handleApprovalRequest(\n\t\tmessage: string,\n\t\tcontext?: Record<string, unknown>\n\t): Promise<ApprovalResponse> {\n\t\tif (!this.approvalHandler) {\n\t\t\tthrow new Error(\n\t\t\t\t'No approval handler configured. Pass approvalHandler option when creating VercelAIATPClient.'\n\t\t\t);\n\t\t}\n\n\t\tconst approved = await this.approvalHandler(message, context);\n\t\treturn {\n\t\t\tapproved,\n\t\t\ttimestamp: Date.now(),\n\t\t};\n\t}\n}\n","/**\n * Vercel AI SDK Event Adapter\n *\n * Converts ATP streaming events to Vercel AI SDK UIMessageStream format.\n * Use this to forward ATP events to the chat UI.\n */\n\nimport { type ATPEvent, ATPEventType } from '@mondaydotcomorg/atp-protocol';\n\n/**\n * UIMessageStreamWriter interface (subset of Vercel AI SDK's UIMessageStreamWriter)\n */\nexport interface UIMessageStreamWriter {\n\twrite(event: UIStreamEvent): void;\n}\n\n/**\n * UI Stream event types that Vercel AI SDK understands\n */\nexport type UIStreamEvent =\n\t| { type: 'text-start'; id: string }\n\t| { type: 'text-delta'; id: string; delta: string }\n\t| { type: 'text-end'; id: string }\n\t| { type: 'reasoning-start'; id: string }\n\t| { type: 'reasoning-delta'; id: string; delta: string }\n\t| { type: 'reasoning-end'; id: string }\n\t| { type: 'source-url'; url: string; title: string; sourceId: string; providerMetadata?: unknown }\n\t| { type: string; data?: unknown; transient?: boolean };\n\n/**\n * Options for creating the Vercel event handler\n */\nexport interface CreateVercelEventHandlerOptions {\n\t/**\n\t * Prefix for generated text run IDs\n\t * @default 'atp-text'\n\t */\n\ttextRunIdPrefix?: string;\n}\n\n/**\n * Creates an event handler that forwards ATP events to a Vercel AI SDK UIMessageStreamWriter.\n *\n * Handles nested tool calls properly:\n * - First tool_start triggers agentStepStart\n * - Inner tool_start/end emit agentStep updates (not start/end)\n * - Last tool_end triggers agentStepEnd\n *\n * @param dataStream - The Vercel AI SDK UIMessageStreamWriter to write events to\n * @param options - Optional configuration\n * @returns An event handler function to pass to ATP's executeStream\n *\n * @example\n * ```typescript\n * const handler = createVercelEventHandler(dataStream);\n *\n * const result = await client.executeStream(code, {}, handler);\n * ```\n */\nexport function createVercelEventHandler(\n\tdataStream: UIMessageStreamWriter,\n\toptions: CreateVercelEventHandlerOptions = {}\n): (event: ATPEvent) => void {\n\tconst { textRunIdPrefix = 'atp-text' } = options;\n\n\tlet currentTextRunId: string | undefined;\n\tlet textRunCounter = 0;\n\n\t// Track reasoning state for proper start/delta/end events\n\tlet reasoningRunId: string | undefined;\n\tlet reasoningCounter = 0;\n\n\t// Track nested tool calls with a stack\n\tconst toolStack: Array<{ toolName: string; apiGroup: string }> = [];\n\n\treturn (event: ATPEvent) => {\n\t\tswitch (event.type) {\n\t\t\tcase ATPEventType.THINKING: {\n\t\t\t\tconst data = event.data as { content: string; step?: string };\n\n\t\t\t\t// Start new reasoning block if needed\n\t\t\t\tif (!reasoningRunId) {\n\t\t\t\t\treasoningRunId = `atp-reasoning-${++reasoningCounter}`;\n\t\t\t\t\tdataStream.write({\n\t\t\t\t\t\ttype: 'reasoning-start',\n\t\t\t\t\t\tid: reasoningRunId,\n\t\t\t\t\t} as UIStreamEvent);\n\t\t\t\t}\n\n\t\t\t\t// Write reasoning delta\n\t\t\t\tdataStream.write({\n\t\t\t\t\ttype: 'reasoning-delta',\n\t\t\t\t\tid: reasoningRunId,\n\t\t\t\t\tdelta: data.content + '\\n',\n\t\t\t\t} as UIStreamEvent);\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase ATPEventType.TOOL_START: {\n\t\t\t\tconst data = event.data as { toolName: string; apiGroup: string; input: unknown };\n\t\t\t\tconst isFirstTool = toolStack.length === 0;\n\n\t\t\t\ttoolStack.push({ toolName: data.toolName, apiGroup: data.apiGroup });\n\n\t\t\t\tif (isFirstTool) {\n\t\t\t\t\t// First tool - emit agentStepStart\n\t\t\t\t\tdataStream.write({\n\t\t\t\t\t\ttype: 'data-agentStepStart',\n\t\t\t\t\t\tdata: `Executing ${data.apiGroup}.${data.toolName}...`,\n\t\t\t\t\t\ttransient: true,\n\t\t\t\t\t});\n\t\t\t\t} else {\n\t\t\t\t\t// Nested tool - emit progress update instead\n\t\t\t\t\tdataStream.write({\n\t\t\t\t\t\ttype: 'data-agentStep',\n\t\t\t\t\t\tdata: `→ ${data.apiGroup}.${data.toolName}`,\n\t\t\t\t\t\ttransient: true,\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase ATPEventType.TOOL_END: {\n\t\t\t\tconst data = event.data as {\n\t\t\t\t\ttoolName: string;\n\t\t\t\t\tapiGroup: string;\n\t\t\t\t\toutput: unknown;\n\t\t\t\t\tduration: number;\n\t\t\t\t\tsuccess: boolean;\n\t\t\t\t\terror?: string;\n\t\t\t\t};\n\n\t\t\t\ttoolStack.pop();\n\t\t\t\tconst isLastTool = toolStack.length === 0;\n\n\t\t\t\t// Close reasoning block when outer tool ends\n\t\t\t\tif (isLastTool && reasoningRunId) {\n\t\t\t\t\tdataStream.write({\n\t\t\t\t\t\ttype: 'reasoning-end',\n\t\t\t\t\t\tid: reasoningRunId,\n\t\t\t\t\t} as UIStreamEvent);\n\t\t\t\t\treasoningRunId = undefined;\n\t\t\t\t}\n\n\t\t\t\tif (isLastTool) {\n\t\t\t\t\t// Last tool - emit agentStepEnd\n\t\t\t\t\tdataStream.write({\n\t\t\t\t\t\ttype: 'data-agentStepEnd',\n\t\t\t\t\t\tdata: data.success\n\t\t\t\t\t\t\t? `Completed ${data.apiGroup}.${data.toolName} (${data.duration}ms)`\n\t\t\t\t\t\t\t: `Failed ${data.apiGroup}.${data.toolName}: ${data.error}`,\n\t\t\t\t\t\ttransient: true,\n\t\t\t\t\t});\n\t\t\t\t} else {\n\t\t\t\t\t// Nested tool finished - emit progress update\n\t\t\t\t\tdataStream.write({\n\t\t\t\t\t\ttype: 'data-agentStep',\n\t\t\t\t\t\tdata: data.success\n\t\t\t\t\t\t\t? `✓ ${data.apiGroup}.${data.toolName} (${data.duration}ms)`\n\t\t\t\t\t\t\t: `✗ ${data.apiGroup}.${data.toolName}: ${data.error}`,\n\t\t\t\t\t\ttransient: true,\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase ATPEventType.TEXT: {\n\t\t\t\tconst data = event.data as { text: string };\n\t\t\t\tconst runId = event.runId || `${textRunIdPrefix}-${++textRunCounter}`;\n\n\t\t\t\t// If we're starting a new text run, close the previous one\n\t\t\t\tif (currentTextRunId && currentTextRunId !== runId) {\n\t\t\t\t\tdataStream.write({ type: 'text-end', id: currentTextRunId });\n\t\t\t\t\tcurrentTextRunId = undefined;\n\t\t\t\t}\n\n\t\t\t\t// Start new text run if needed\n\t\t\t\tif (!currentTextRunId || currentTextRunId !== runId) {\n\t\t\t\t\tcurrentTextRunId = runId;\n\t\t\t\t\tdataStream.write({ type: 'text-start', id: currentTextRunId });\n\t\t\t\t}\n\n\t\t\t\t// Write text delta\n\t\t\t\tdataStream.write({\n\t\t\t\t\ttype: 'text-delta',\n\t\t\t\t\tid: currentTextRunId,\n\t\t\t\t\tdelta: data.text,\n\t\t\t\t});\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase ATPEventType.TEXT_END:\n\t\t\t\tif (currentTextRunId) {\n\t\t\t\t\tdataStream.write({ type: 'text-end', id: currentTextRunId });\n\t\t\t\t\tcurrentTextRunId = undefined;\n\t\t\t\t}\n\t\t\t\tbreak;\n\n\t\t\tcase ATPEventType.SOURCE: {\n\t\t\t\tconst data = event.data as {\n\t\t\t\t\turl: string;\n\t\t\t\t\ttitle: string;\n\t\t\t\t\tsummary?: string;\n\t\t\t\t\tcreatedAt?: string;\n\t\t\t\t};\n\t\t\t\tdataStream.write({\n\t\t\t\t\ttype: 'source-url',\n\t\t\t\t\turl: data.url,\n\t\t\t\t\ttitle: data.title,\n\t\t\t\t\tsourceId: `${data.title}-${event.timestamp}`,\n\t\t\t\t\tproviderMetadata: {\n\t\t\t\t\t\taiChat: {\n\t\t\t\t\t\t\tsummary: data.summary,\n\t\t\t\t\t\t\tcreatedAt: data.createdAt,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t});\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase ATPEventType.PROGRESS: {\n\t\t\t\tconst data = event.data as { message: string; fraction: number };\n\t\t\t\tdataStream.write({\n\t\t\t\t\ttype: 'data-agentStep',\n\t\t\t\t\tdata: `${data.message} (${Math.round(data.fraction * 100)}%)`,\n\t\t\t\t\ttransient: true,\n\t\t\t\t});\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase ATPEventType.ERROR: {\n\t\t\t\tconst data = event.data as { message: string; code?: string };\n\t\t\t\tdataStream.write({\n\t\t\t\t\ttype: 'data-error',\n\t\t\t\t\tdata: data.message,\n\t\t\t\t});\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tdefault:\n\t\t\t\t// Forward unknown events as custom data events\n\t\t\t\tdataStream.write({\n\t\t\t\t\ttype: `data-${event.type}`,\n\t\t\t\t\tdata: event.data,\n\t\t\t\t\ttransient: true,\n\t\t\t\t});\n\t\t\t\tbreak;\n\t\t}\n\t};\n}\n\n/**\n * Creates an event handler that collects events into an array.\n * Useful for testing or post-processing events.\n *\n * @returns Object with handler function and collected events array\n */\nexport function createEventCollector(): {\n\thandler: (event: ATPEvent) => void;\n\tevents: ATPEvent[];\n\tclear: () => void;\n} {\n\tconst events: ATPEvent[] = [];\n\n\treturn {\n\t\thandler: (event: ATPEvent) => {\n\t\t\tevents.push(event);\n\t\t},\n\t\tevents,\n\t\tclear: () => {\n\t\t\tevents.length = 0;\n\t\t},\n\t};\n}\n\n","import { z } from 'zod';\nimport { VercelAIATPClient } from './client.js';\nimport type { CreateATPToolsOptions, ATPToolsResult, StreamingToolsOptions } from './types.js';\nimport {\n\tToolNames,\n\texecuteCodeInputSchema,\n\texploreApiInputSchema,\n\tsearchApiInputSchema,\n\tfetchAllApisInputSchema,\n} from '@mondaydotcomorg/atp-client';\nimport { ExecutionStatus } from '@mondaydotcomorg/atp-protocol';\nimport { tool } from 'ai';\nimport { createVercelEventHandler, type UIMessageStreamWriter } from './event-adapter.js';\n\nconst TOOL_SCHEMAS = {\n\t[ToolNames.EXECUTE_CODE]: executeCodeInputSchema.pick({ code: true }),\n\t[ToolNames.EXPLORE_API]: exploreApiInputSchema,\n\t[ToolNames.SEARCH_API]: searchApiInputSchema,\n\t[ToolNames.FETCH_ALL_APIS]: fetchAllApisInputSchema,\n} as const;\n\nexport async function createATPTools(options: CreateATPToolsOptions): Promise<ATPToolsResult> {\n\tconst { defaultExecutionConfig, ...clientOptions } = options;\n\n\tconst client = new VercelAIATPClient(clientOptions);\n\tawait client.connect();\n\n\tconst underlyingClient = client.getUnderlyingClient();\n\tconst vercelTools: Record<string, any> = {};\n\n\tvercelTools.atp_execute_code = tool({\n\t\tdescription:\n\t\t\t'Execute TypeScript code in ATP sandbox with access to runtime APIs (atp.llm.*, atp.embedding.*, atp.approval.*)',\n\t\tparameters: TOOL_SCHEMAS[ToolNames.EXECUTE_CODE],\n\t\texecute: async ({ code }: { code: string }) => {\n\t\t\ttry {\n\t\t\t\tconst result = await client.execute(code, defaultExecutionConfig);\n\n\t\t\t\tif (result.status === ExecutionStatus.COMPLETED) {\n\t\t\t\t\treturn {\n\t\t\t\t\t\tsuccess: true,\n\t\t\t\t\t\tresult: result.result,\n\t\t\t\t\t\tstats: result.stats,\n\t\t\t\t\t};\n\t\t\t\t} else if (result.status === ExecutionStatus.FAILED) {\n\t\t\t\t\treturn {\n\t\t\t\t\t\tsuccess: false,\n\t\t\t\t\t\terror: result.error,\n\t\t\t\t\t\tstats: result.stats,\n\t\t\t\t\t};\n\t\t\t\t} else {\n\t\t\t\t\treturn {\n\t\t\t\t\t\tsuccess: false,\n\t\t\t\t\t\terror: 'Execution in unexpected state: ' + result.status,\n\t\t\t\t\t};\n\t\t\t\t}\n\t\t\t} catch (error: any) {\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: false,\n\t\t\t\t\terror: error.message || 'Unknown error',\n\t\t\t\t};\n\t\t\t}\n\t\t},\n\t});\n\n\tvercelTools.atp_explore_api = tool({\n\t\tdescription:\n\t\t\t'Explore APIs using filesystem-like navigation. Navigate through directories to discover available functions.',\n\t\tparameters: TOOL_SCHEMAS[ToolNames.EXPLORE_API],\n\t\texecute: async ({ path }: { path: string }) => {\n\t\t\ttry {\n\t\t\t\tconst result = await underlyingClient.exploreAPI(path);\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: true,\n\t\t\t\t\tresult,\n\t\t\t\t};\n\t\t\t} catch (error: any) {\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: false,\n\t\t\t\t\terror: error.message,\n\t\t\t\t};\n\t\t\t}\n\t\t},\n\t});\n\n\tvercelTools.atp_search_api = tool({\n\t\tdescription:\n\t\t\t'Search for APIs by keyword. Provide search term as string like \"add\", \"math\", \"user\", etc.',\n\t\tparameters: TOOL_SCHEMAS[ToolNames.SEARCH_API],\n\t\texecute: async ({ query }: { query: string }) => {\n\t\t\ttry {\n\t\t\t\tconst results = await underlyingClient.searchAPI(query);\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: true,\n\t\t\t\t\tresults: results.map((r: any) => ({\n\t\t\t\t\t\tapiGroup: r.apiGroup,\n\t\t\t\t\t\tfunctionName: r.functionName,\n\t\t\t\t\t\tdescription: r.description,\n\t\t\t\t\t\tsignature: r.signature,\n\t\t\t\t\t})),\n\t\t\t\t\tcount: results.length,\n\t\t\t\t};\n\t\t\t} catch (error: any) {\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: false,\n\t\t\t\t\terror: error.message,\n\t\t\t\t};\n\t\t\t}\n\t\t},\n\t});\n\n\tvercelTools.atp_get_type_definitions = tool({\n\t\tdescription:\n\t\t\t'Get TypeScript type definitions for ATP runtime APIs to understand available functions',\n\t\tparameters: z.object({}),\n\t\texecute: async () => {\n\t\t\ttry {\n\t\t\t\tconst types = client.getTypeDefinitions();\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: true,\n\t\t\t\t\ttypes,\n\t\t\t\t};\n\t\t\t} catch (error: any) {\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: false,\n\t\t\t\t\terror: error.message,\n\t\t\t\t};\n\t\t\t}\n\t\t},\n\t});\n\n\treturn {\n\t\tclient,\n\t\ttools: vercelTools,\n\t};\n}\n\n/**\n * Creates ATP tools with streaming event support.\n * Events from tool execution will be forwarded to the provided dataStream.\n *\n * @param options - Tool creation options including dataStream for event forwarding\n * @returns Promise resolving to client and tools with streaming support\n *\n * @example\n * ```typescript\n * // In your Vercel AI SDK route handler\n * const dataStream = createUIMessageStream({...});\n *\n * const { tools } = await createATPStreamingTools({\n * serverUrl: 'http://localhost:3333',\n * model: openai('gpt-4'),\n * dataStream,\n * });\n *\n * // Use tools in streamText\n * const result = streamText({\n * model: openai('gpt-4'),\n * messages,\n * tools,\n * });\n * ```\n */\nexport async function createATPStreamingTools(\n\toptions: StreamingToolsOptions\n): Promise<ATPToolsResult> {\n\tconst { dataStream, defaultExecutionConfig, ...clientOptions } = options;\n\n\tconst client = new VercelAIATPClient(clientOptions);\n\tawait client.connect();\n\n\tconst eventHandler = createVercelEventHandler(dataStream);\n\tconst underlyingClient = client.getUnderlyingClient();\n\n\tconst vercelTools: Record<string, any> = {};\n\n\tvercelTools.atp_execute_code = tool({\n\t\tdescription:\n\t\t\t'Execute TypeScript code in ATP sandbox with streaming events for thinking, tool execution, and text output',\n\t\tparameters: TOOL_SCHEMAS[ToolNames.EXECUTE_CODE],\n\t\texecute: async ({ code }: { code: string }) => {\n\t\t\ttry {\n\t\t\t\tconst result = await underlyingClient.executeStream(\n\t\t\t\t\tcode,\n\t\t\t\t\tdefaultExecutionConfig,\n\t\t\t\t\teventHandler\n\t\t\t\t);\n\n\t\t\t\tif (result.status === ExecutionStatus.COMPLETED) {\n\t\t\t\t\treturn {\n\t\t\t\t\t\tsuccess: true,\n\t\t\t\t\t\tresult: result.result,\n\t\t\t\t\t\tstats: result.stats,\n\t\t\t\t\t};\n\t\t\t\t} else if (result.status === ExecutionStatus.FAILED) {\n\t\t\t\t\treturn {\n\t\t\t\t\t\tsuccess: false,\n\t\t\t\t\t\terror: result.error,\n\t\t\t\t\t\tstats: result.stats,\n\t\t\t\t\t};\n\t\t\t\t} else {\n\t\t\t\t\treturn {\n\t\t\t\t\t\tsuccess: false,\n\t\t\t\t\t\terror: 'Execution in unexpected state: ' + result.status,\n\t\t\t\t\t};\n\t\t\t\t}\n\t\t\t} catch (error: any) {\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: false,\n\t\t\t\t\terror: error.message || 'Unknown error',\n\t\t\t\t};\n\t\t\t}\n\t\t},\n\t});\n\n\tvercelTools.atp_explore_api = tool({\n\t\tdescription:\n\t\t\t'Explore APIs using filesystem-like navigation. Navigate through directories to discover available functions.',\n\t\tparameters: TOOL_SCHEMAS[ToolNames.EXPLORE_API],\n\t\texecute: async ({ path }: { path: string }) => {\n\t\t\ttry {\n\t\t\t\tconst result = await underlyingClient.exploreAPI(path);\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: true,\n\t\t\t\t\tresult,\n\t\t\t\t};\n\t\t\t} catch (error: any) {\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: false,\n\t\t\t\t\terror: error.message,\n\t\t\t\t};\n\t\t\t}\n\t\t},\n\t});\n\n\tvercelTools.atp_search_api = tool({\n\t\tdescription:\n\t\t\t'Search for APIs by keyword. Provide search term as string like \"add\", \"math\", \"user\", etc.',\n\t\tparameters: TOOL_SCHEMAS[ToolNames.SEARCH_API],\n\t\texecute: async ({ query }: { query: string }) => {\n\t\t\ttry {\n\t\t\t\tconst results = await underlyingClient.searchAPI(query);\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: true,\n\t\t\t\t\tresults: results.map((r: any) => ({\n\t\t\t\t\t\tapiGroup: r.apiGroup,\n\t\t\t\t\t\tfunctionName: r.functionName,\n\t\t\t\t\t\tdescription: r.description,\n\t\t\t\t\t\tsignature: r.signature,\n\t\t\t\t\t})),\n\t\t\t\t\tcount: results.length,\n\t\t\t\t};\n\t\t\t} catch (error: any) {\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: false,\n\t\t\t\t\terror: error.message,\n\t\t\t\t};\n\t\t\t}\n\t\t},\n\t});\n\n\tvercelTools.atp_get_type_definitions = tool({\n\t\tdescription:\n\t\t\t'Get TypeScript type definitions for ATP runtime APIs to understand available functions',\n\t\tparameters: z.object({}),\n\t\texecute: async () => {\n\t\t\ttry {\n\t\t\t\tconst types = client.getTypeDefinitions();\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: true,\n\t\t\t\t\ttypes,\n\t\t\t\t};\n\t\t\t} catch (error: any) {\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: false,\n\t\t\t\t\terror: error.message,\n\t\t\t\t};\n\t\t\t}\n\t\t},\n\t});\n\n\treturn {\n\t\tclient,\n\t\ttools: vercelTools,\n\t};\n}\n"]}
|
package/dist/tools.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../src/tools.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,qBAAqB,EAAE,cAAc,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;
|
|
1
|
+
{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../src/tools.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,qBAAqB,EAAE,cAAc,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AAmB/F,wBAAsB,cAAc,CAAC,OAAO,EAAE,qBAAqB,GAAG,OAAO,CAAC,cAAc,CAAC,CAkH5F;AAED;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,wBAAsB,uBAAuB,CAC5C,OAAO,EAAE,qBAAqB,GAC5B,OAAO,CAAC,cAAc,CAAC,CAwHzB"}
|