llm-advanced-tools 0.1.1 → 0.1.2
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.
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"vercel-ai.d.ts","sourceRoot":"","sources":["../../src/adapters/vercel-ai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgB,aAAa,
|
|
1
|
+
{"version":3,"file":"vercel-ai.d.ts","sourceRoot":"","sources":["../../src/adapters/vercel-ai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgB,aAAa,EAAQ,MAAM,IAAI,CAAC;AAEvD,OAAO,EACL,eAAe,EACf,oBAAoB,EACpB,cAAc,EACd,WAAW,EACX,YAAY,EACZ,QAAQ,EACR,UAAU,EAEX,MAAM,UAAU,CAAC;AAElB;;;GAGG;AACH,qBAAa,eAAgB,YAAW,eAAe;IACrD,QAAQ,CAAC,IAAI,eAAe;IAC5B,QAAQ,CAAC,YAAY,EAAE,oBAAoB,CAOzC;IAEF,OAAO,CAAC,KAAK,CAAgB;IAC7B,OAAO,CAAC,YAAY,CAAS;IAE7B;;;OAGG;gBACS,KAAK,EAAE,aAAa,EAAE,YAAY,SAAW;IAKzD;;OAEG;IACH,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;IAyBrD;;OAEG;IACH,OAAO,CAAC,eAAe;IAsBvB;;OAEG;IACH,OAAO,CAAC,eAAe;IAsDvB;;OAEG;IACG,IAAI,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IAwFvD;;OAEG;IACH,cAAc,CAAC,SAAS,EAAE,GAAG,EAAE,GAAG,QAAQ,EAAE;IAQ5C;;OAEG;IACH,iBAAiB,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,GAAG;IAK7C;;OAEG;IACH,OAAO,CAAC,eAAe;CAiBxB"}
|
|
@@ -135,31 +135,40 @@ class VercelAIAdapter {
|
|
|
135
135
|
// Convert schema to Zod
|
|
136
136
|
const parameters = this.jsonSchemaToZod(toolDef.inputSchema);
|
|
137
137
|
// Create the tool using AI SDK's tool() helper
|
|
138
|
-
|
|
138
|
+
// Store reference to toolDef in closure to avoid issues
|
|
139
|
+
const handler = toolDef.handler;
|
|
140
|
+
const name = toolDef.name;
|
|
141
|
+
tools[name] = {
|
|
139
142
|
description,
|
|
140
143
|
parameters,
|
|
141
144
|
execute: async (args) => {
|
|
142
145
|
try {
|
|
143
|
-
const result = await
|
|
146
|
+
const result = await handler(args);
|
|
144
147
|
return result;
|
|
145
148
|
}
|
|
146
149
|
catch (error) {
|
|
147
|
-
throw new Error(`Tool ${
|
|
150
|
+
throw new Error(`Tool ${name} failed: ${error.message}`);
|
|
148
151
|
}
|
|
149
152
|
}
|
|
150
|
-
}
|
|
153
|
+
};
|
|
151
154
|
}
|
|
152
155
|
// Generate text with tools
|
|
153
|
-
const
|
|
156
|
+
const generateOptions = {
|
|
154
157
|
model: this.model,
|
|
155
158
|
messages: messages.map(m => ({
|
|
156
159
|
role: m.role,
|
|
157
160
|
content: m.content
|
|
158
161
|
})),
|
|
159
|
-
tools: Object.keys(tools).length > 0 ? tools : undefined
|
|
160
|
-
|
|
161
|
-
|
|
162
|
-
|
|
162
|
+
tools: Object.keys(tools).length > 0 ? tools : undefined
|
|
163
|
+
};
|
|
164
|
+
// Add optional parameters if provided
|
|
165
|
+
if (request.temperature !== undefined) {
|
|
166
|
+
generateOptions.temperature = request.temperature;
|
|
167
|
+
}
|
|
168
|
+
if (request.maxTokens !== undefined) {
|
|
169
|
+
generateOptions.maxCompletionTokens = request.maxTokens;
|
|
170
|
+
}
|
|
171
|
+
const result = await (0, ai_1.generateText)(generateOptions);
|
|
163
172
|
// Parse tool calls from result
|
|
164
173
|
const toolCalls = [];
|
|
165
174
|
if (result.toolCalls && result.toolCalls.length > 0) {
|
|
@@ -179,9 +188,9 @@ class VercelAIAdapter {
|
|
|
179
188
|
},
|
|
180
189
|
usage: result.usage
|
|
181
190
|
? {
|
|
182
|
-
inputTokens: result.usage.promptTokens,
|
|
183
|
-
outputTokens: result.usage.completionTokens,
|
|
184
|
-
totalTokens: result.usage.totalTokens
|
|
191
|
+
inputTokens: result.usage.inputTokens || result.usage.promptTokens || 0,
|
|
192
|
+
outputTokens: result.usage.outputTokens || result.usage.completionTokens || 0,
|
|
193
|
+
totalTokens: result.usage.totalTokens || 0
|
|
185
194
|
}
|
|
186
195
|
: undefined,
|
|
187
196
|
stopReason: this.mapFinishReason(result.finishReason),
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"vercel-ai.js","sourceRoot":"","sources":["../../src/adapters/vercel-ai.ts"],"names":[],"mappings":";;;AAAA,
|
|
1
|
+
{"version":3,"file":"vercel-ai.js","sourceRoot":"","sources":["../../src/adapters/vercel-ai.ts"],"names":[],"mappings":";;;AAAA,2BAAuD;AACvD,6BAAwB;AAYxB;;;GAGG;AACH,MAAa,eAAe;IAc1B;;;OAGG;IACH,YAAY,KAAoB,EAAE,YAAY,GAAG,QAAQ;QAjBhD,SAAI,GAAG,WAAW,CAAC;QACnB,iBAAY,GAAyB;YAC5C,wBAAwB,EAAE,KAAK;YAC/B,2BAA2B,EAAE,KAAK;YAClC,sBAAsB,EAAE,KAAK;YAC7B,iBAAiB,EAAE,IAAI;YACvB,yBAAyB,EAAE,IAAI;YAC/B,SAAS,EAAE,MAAM,CAAC,uCAAuC;SAC1D,CAAC;QAUA,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,IAAoB;QAC7B,wDAAwD;QACxD,IAAI,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC;QAElC,gDAAgD;QAChD,IAAI,OAAO,UAAU,KAAK,QAAQ,IAAI,OAAO,IAAI,UAAU,EAAE,CAAC;YAC5D,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC;QAChC,CAAC;QAED,kDAAkD;QAClD,IAAI,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;QACnC,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxD,WAAW,IAAI,oBAAoB,CAAC;YACpC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE;gBACxC,WAAW,IAAI,KAAK,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;YAC1D,CAAC,CAAC,CAAC;QACL,CAAC;QAED,yBAAyB;QACzB,OAAO;YACL,WAAW;YACX,UAAU;SACX,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,QAAmB;QAIzC,MAAM,SAAS,GAAsE,EAAE,CAAC;QAExF,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACxB,iDAAiD;gBACjD,8CAA8C;gBAC9C,SAAS;YACX,CAAC;YAED,SAAS,CAAC,IAAI,CAAC;gBACb,IAAI,EAAE,GAAG,CAAC,IAAuC;gBACjD,OAAO,EAAE,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC;aACrF,CAAC,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,MAAW;QACjC,0CAA0C;QAC1C,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,OAAO,IAAI,MAAM,EAAE,CAAC;YAC9D,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,uCAAuC;QACvC,wEAAwE;QACxE,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,MAAM,KAAK,GAA8B,EAAE,CAAC;YAC5C,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC;YAEvC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC,EAAE,CAAC;gBACnE,MAAM,IAAI,GAAQ,KAAK,CAAC;gBACxB,IAAI,OAAkB,CAAC;gBAEvB,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;oBAClB,KAAK,QAAQ;wBACX,OAAO,GAAG,OAAC,CAAC,MAAM,EAAE,CAAC;wBACrB,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;4BACd,OAAO,GAAG,OAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBAC9B,CAAC;wBACD,MAAM;oBACR,KAAK,QAAQ;wBACX,OAAO,GAAG,OAAC,CAAC,MAAM,EAAE,CAAC;wBACrB,MAAM;oBACR,KAAK,SAAS;wBACZ,OAAO,GAAG,OAAC,CAAC,OAAO,EAAE,CAAC;wBACtB,MAAM;oBACR,KAAK,OAAO;wBACV,OAAO,GAAG,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC,CAAC;wBAC3B,MAAM;oBACR;wBACE,OAAO,GAAG,OAAC,CAAC,GAAG,EAAE,CAAC;gBACtB,CAAC;gBAED,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;oBACrB,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;gBAC/C,CAAC;gBAED,+CAA+C;gBAC/C,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;oBAC5B,OAAO,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;gBAC/B,CAAC;gBAED,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC;YACvB,CAAC;YAED,OAAO,OAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;QAED,OAAO,OAAC,CAAC,GAAG,EAAE,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,OAAoB;QAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAExD,6DAA6D;QAC7D,MAAM,KAAK,GAAwB,EAAE,CAAC;QACtC,MAAM,eAAe,GAAG,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC;QAE1E,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;YACtC,8BAA8B;YAC9B,IAAI,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;YACtC,IAAI,OAAO,CAAC,aAAa,IAAI,OAAO,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9D,WAAW,IAAI,oBAAoB,CAAC;gBACpC,OAAO,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE;oBAC3C,WAAW,IAAI,KAAK,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC1D,CAAC,CAAC,CAAC;YACL,CAAC;YAED,wBAAwB;YACxB,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAE7D,+CAA+C;YAC/C,wDAAwD;YACxD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;YAChC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;YAC1B,KAAK,CAAC,IAAI,CAAC,GAAG;gBACZ,WAAW;gBACX,UAAU;gBACV,OAAO,EAAE,KAAK,EAAE,IAAS,EAAE,EAAE;oBAC3B,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;wBACnC,OAAO,MAAM,CAAC;oBAChB,CAAC;oBAAC,OAAO,KAAU,EAAE,CAAC;wBACpB,MAAM,IAAI,KAAK,CAAC,QAAQ,IAAI,YAAY,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;oBAC3D,CAAC;gBACH,CAAC;aACK,CAAC;QACX,CAAC;QAED,2BAA2B;QAC3B,MAAM,eAAe,GAAQ;YAC3B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBAC3B,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,OAAO,EAAE,CAAC,CAAC,OAAO;aACnB,CAAC,CAAC;YACH,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;SACzD,CAAC;QAEF,sCAAsC;QACtC,IAAI,OAAO,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;YACtC,eAAe,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QACpD,CAAC;QACD,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACpC,eAAe,CAAC,mBAAmB,GAAG,OAAO,CAAC,SAAS,CAAC;QAC1D,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAA,iBAAY,EAAC,eAAe,CAAC,CAAC;QAEnD,+BAA+B;QAC/B,MAAM,SAAS,GAAe,EAAE,CAAC;QACjC,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpD,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,EAAO,EAAE,KAAa,EAAE,EAAE;gBAClD,SAAS,CAAC,IAAI,CAAC;oBACb,EAAE,EAAE,QAAQ,KAAK,EAAE;oBACnB,IAAI,EAAE,EAAE,CAAC,QAAQ;oBACjB,KAAK,EAAE,EAAE,CAAC,IAAI;iBACf,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;QAED,OAAO;YACL,OAAO,EAAE;gBACP,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,MAAM,CAAC,IAAI,IAAI,EAAE;gBAC1B,SAAS;aACV;YACD,KAAK,EAAE,MAAM,CAAC,KAAK;gBACjB,CAAC,CAAC;oBACE,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW,IAAK,MAAM,CAAC,KAAa,CAAC,YAAY,IAAI,CAAC;oBAChF,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY,IAAK,MAAM,CAAC,KAAa,CAAC,gBAAgB,IAAI,CAAC;oBACtF,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC;iBAC3C;gBACH,CAAC,CAAC,SAAS;YACb,UAAU,EAAE,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,YAAY,CAAC;YACrD,SAAS;SACV,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,SAAgB;QAC7B,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;YACnC,EAAE,EAAE,QAAQ,KAAK,EAAE;YACnB,IAAI,EAAE,EAAE,CAAC,QAAQ;YACjB,KAAK,EAAE,EAAE,CAAC,IAAI;SACf,CAAC,CAAC,CAAC;IACN,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,OAAqB;QACrC,oEAAoE;QACpE,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACK,eAAe,CACrB,MAA0B;QAE1B,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,MAAM;gBACT,OAAO,UAAU,CAAC;YACpB,KAAK,QAAQ;gBACX,OAAO,YAAY,CAAC;YACtB,KAAK,YAAY;gBACf,OAAO,UAAU,CAAC;YACpB,KAAK,gBAAgB,CAAC;YACtB,KAAK,OAAO;gBACV,OAAO,eAAe,CAAC;YACzB;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC;CACF;AAvQD,0CAuQC"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "llm-advanced-tools",
|
|
3
|
-
"version": "0.1.
|
|
3
|
+
"version": "0.1.2",
|
|
4
4
|
"description": "Provider-agnostic advanced tool use library for LLMs",
|
|
5
5
|
"main": "dist/index.js",
|
|
6
6
|
"types": "dist/index.d.ts",
|
|
@@ -28,14 +28,14 @@
|
|
|
28
28
|
"author": "",
|
|
29
29
|
"license": "MIT",
|
|
30
30
|
"dependencies": {
|
|
31
|
-
"openai": "^4.0.0",
|
|
32
31
|
"natural": "^6.0.0",
|
|
32
|
+
"openai": "^4.0.0",
|
|
33
33
|
"zod": "^3.22.0"
|
|
34
34
|
},
|
|
35
35
|
"peerDependencies": {
|
|
36
|
-
"ai": "^
|
|
36
|
+
"@ai-sdk/anthropic": "^1.0.0",
|
|
37
37
|
"@ai-sdk/openai": "^1.0.0",
|
|
38
|
-
"
|
|
38
|
+
"ai": "^4.0.0 || ^5.0.0"
|
|
39
39
|
},
|
|
40
40
|
"peerDependenciesMeta": {
|
|
41
41
|
"ai": {
|
|
@@ -57,7 +57,7 @@
|
|
|
57
57
|
"@types/node": "^20.0.0",
|
|
58
58
|
"@typescript-eslint/eslint-plugin": "^6.0.0",
|
|
59
59
|
"@typescript-eslint/parser": "^6.0.0",
|
|
60
|
-
"ai": "^
|
|
60
|
+
"ai": "^5.0.116",
|
|
61
61
|
"eslint": "^8.0.0",
|
|
62
62
|
"jest": "^29.0.0",
|
|
63
63
|
"prettier": "^3.0.0",
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { generateText, LanguageModel,
|
|
1
|
+
import { generateText, LanguageModel, tool } from 'ai';
|
|
2
2
|
import { z } from 'zod';
|
|
3
3
|
import {
|
|
4
4
|
ProviderAdapter,
|
|
@@ -155,7 +155,7 @@ export class VercelAIAdapter implements ProviderAdapter {
|
|
|
155
155
|
const messages = this.convertMessages(request.messages);
|
|
156
156
|
|
|
157
157
|
// Convert our tools to AI SDK format using the tool() helper
|
|
158
|
-
const tools: Record<string,
|
|
158
|
+
const tools: Record<string, any> = {};
|
|
159
159
|
const toolDefinitions = request.tools?.filter(t => !t.deferLoading) || [];
|
|
160
160
|
|
|
161
161
|
for (const toolDef of toolDefinitions) {
|
|
@@ -172,31 +172,42 @@ export class VercelAIAdapter implements ProviderAdapter {
|
|
|
172
172
|
const parameters = this.jsonSchemaToZod(toolDef.inputSchema);
|
|
173
173
|
|
|
174
174
|
// Create the tool using AI SDK's tool() helper
|
|
175
|
-
|
|
175
|
+
// Store reference to toolDef in closure to avoid issues
|
|
176
|
+
const handler = toolDef.handler;
|
|
177
|
+
const name = toolDef.name;
|
|
178
|
+
tools[name] = {
|
|
176
179
|
description,
|
|
177
180
|
parameters,
|
|
178
181
|
execute: async (args: any) => {
|
|
179
182
|
try {
|
|
180
|
-
const result = await
|
|
183
|
+
const result = await handler(args);
|
|
181
184
|
return result;
|
|
182
185
|
} catch (error: any) {
|
|
183
|
-
throw new Error(`Tool ${
|
|
186
|
+
throw new Error(`Tool ${name} failed: ${error.message}`);
|
|
184
187
|
}
|
|
185
188
|
}
|
|
186
|
-
}
|
|
189
|
+
} as any;
|
|
187
190
|
}
|
|
188
191
|
|
|
189
192
|
// Generate text with tools
|
|
190
|
-
const
|
|
193
|
+
const generateOptions: any = {
|
|
191
194
|
model: this.model,
|
|
192
195
|
messages: messages.map(m => ({
|
|
193
196
|
role: m.role,
|
|
194
197
|
content: m.content
|
|
195
198
|
})),
|
|
196
|
-
tools: Object.keys(tools).length > 0 ? tools : undefined
|
|
197
|
-
|
|
198
|
-
|
|
199
|
-
|
|
199
|
+
tools: Object.keys(tools).length > 0 ? tools : undefined
|
|
200
|
+
};
|
|
201
|
+
|
|
202
|
+
// Add optional parameters if provided
|
|
203
|
+
if (request.temperature !== undefined) {
|
|
204
|
+
generateOptions.temperature = request.temperature;
|
|
205
|
+
}
|
|
206
|
+
if (request.maxTokens !== undefined) {
|
|
207
|
+
generateOptions.maxCompletionTokens = request.maxTokens;
|
|
208
|
+
}
|
|
209
|
+
|
|
210
|
+
const result = await generateText(generateOptions);
|
|
200
211
|
|
|
201
212
|
// Parse tool calls from result
|
|
202
213
|
const toolCalls: ToolCall[] = [];
|
|
@@ -218,9 +229,9 @@ export class VercelAIAdapter implements ProviderAdapter {
|
|
|
218
229
|
},
|
|
219
230
|
usage: result.usage
|
|
220
231
|
? {
|
|
221
|
-
inputTokens: result.usage.promptTokens,
|
|
222
|
-
outputTokens: result.usage.completionTokens,
|
|
223
|
-
totalTokens: result.usage.totalTokens
|
|
232
|
+
inputTokens: result.usage.inputTokens || (result.usage as any).promptTokens || 0,
|
|
233
|
+
outputTokens: result.usage.outputTokens || (result.usage as any).completionTokens || 0,
|
|
234
|
+
totalTokens: result.usage.totalTokens || 0
|
|
224
235
|
}
|
|
225
236
|
: undefined,
|
|
226
237
|
stopReason: this.mapFinishReason(result.finishReason),
|