Skip to content

Commit 411e483

Browse files
authored
chore (provider): refactor usage (language model v2) (#5653)
1 parent c77e1d9 commit 411e483

File tree

53 files changed

+447
-583
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

53 files changed

+447
-583
lines changed

‎.changeset/gentle-gorillas-mate.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
---
2+
'@ai-sdk/provider': major
3+
---
4+
5+
chore (provider): refactor usage (language model v2)

‎content/docs/03-ai-sdk-core/55-testing.mdx

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ const result = await generateText({
3535
model: new MockLanguageModelV1({
3636
doGenerate: async () => ({
3737
finishReason: 'stop',
38-
usage: { promptTokens: 10, completionTokens: 20 },
38+
usage: { inputTokens: 10, outputTokens: 20 },
3939
text: `Hello, world!`,
4040
}),
4141
}),
@@ -61,7 +61,7 @@ const result = streamText({
6161
type: 'finish',
6262
finishReason: 'stop',
6363
logprobs: undefined,
64-
usage: { completionTokens: 10, promptTokens: 3 },
64+
usage: { inputTokens: 3, outputTokens: 10 },
6565
},
6666
],
6767
}),
@@ -83,7 +83,7 @@ const result = await generateObject({
8383
defaultObjectGenerationMode: 'json',
8484
doGenerate: async () => ({
8585
finishReason: 'stop',
86-
usage: { promptTokens: 10, completionTokens: 20 },
86+
usage: { inputTokens: 10, outputTokens: 20 },
8787
text: `{"content":"Hello, world!"}`,
8888
}),
8989
}),
@@ -115,7 +115,7 @@ const result = streamObject({
115115
type: 'finish',
116116
finishReason: 'stop',
117117
logprobs: undefined,
118-
usage: { completionTokens: 10, promptTokens: 3 },
118+
usage: { inputTokens: 3, outputTokens: 10 },
119119
},
120120
],
121121
}),

‎examples/ai-core/src/generate-object/mock-error.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ async function main() {
1515
modelId: 'model-1',
1616
},
1717
finishReason: 'stop',
18-
usage: { promptTokens: 10, completionTokens: 20 },
18+
usage: { inputTokens: 10, outputTokens: 20 },
1919
text: `{"content":"Hello broken json`,
2020
}),
2121
}),

‎examples/ai-core/src/generate-object/mock-repair-add-close.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ async function main() {
77
const result = await generateObject({
88
model: new MockLanguageModelV2({
99
doGenerate: async () => ({
10-
usage: { promptTokens: 10, completionTokens: 20 },
10+
usage: { inputTokens: 10, outputTokens: 20 },
1111
finishReason: 'tool-calls',
1212
text: `{ "content": "provider metadata test"`,
1313
}),

‎examples/ai-core/src/generate-object/mock.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ async function main() {
1010
doGenerate: async () => ({
1111
text: `{"content":"Hello, world!"}`,
1212
finishReason: 'stop',
13-
usage: { promptTokens: 10, completionTokens: 20 },
13+
usage: { inputTokens: 10, outputTokens: 20 },
1414
}),
1515
}),
1616
schema: z.object({ content: z.string() }),

‎examples/ai-core/src/generate-text/mock-tool-call-repair-reask.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ async function main() {
88
const result = await generateText({
99
model: new MockLanguageModelV2({
1010
doGenerate: async () => ({
11-
usage: { promptTokens: 10, completionTokens: 20 },
11+
usage: { inputTokens: 10, outputTokens: 20 },
1212
finishReason: 'tool-calls',
1313
toolCalls: [
1414
{

‎examples/ai-core/src/generate-text/mock-tool-call-repair-structured-model.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ async function main() {
88
const result = await generateText({
99
model: new MockLanguageModelV2({
1010
doGenerate: async () => ({
11-
usage: { promptTokens: 10, completionTokens: 20 },
11+
usage: { inputTokens: 10, outputTokens: 20 },
1212
finishReason: 'tool-calls',
1313
toolCalls: [
1414
{

‎examples/ai-core/src/generate-text/mock.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ async function main() {
88
doGenerate: async () => ({
99
text: `Hello, world!`,
1010
finishReason: 'stop',
11-
usage: { promptTokens: 10, completionTokens: 20 },
11+
usage: { inputTokens: 10, outputTokens: 20 },
1212
}),
1313
}),
1414
prompt: 'Hello, test!',

‎examples/ai-core/src/stream-object/mock.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ async function main() {
1919
type: 'finish',
2020
finishReason: 'stop',
2121
logprobs: undefined,
22-
usage: { completionTokens: 10, promptTokens: 3 },
22+
usage: { inputTokens: 3, outputTokens: 10 },
2323
},
2424
]),
2525
}),

‎examples/ai-core/src/stream-text/mock.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ async function main() {
1414
type: 'finish',
1515
finishReason: 'stop',
1616
logprobs: undefined,
17-
usage: { completionTokens: 10, promptTokens: 3 },
17+
usage: { inputTokens: 3, outputTokens: 10 },
1818
},
1919
]),
2020
}),

0 commit comments

Comments
 (0)