chore: apply requested changes

This commit is contained in:
Ralph Khreish
2025-10-03 16:14:08 +02:00
parent f3deacb74e
commit 33386cdce8

View File

@@ -176,12 +176,19 @@ export class BaseAIProvider {
`${this.name} generateText completed successfully for model: ${params.modelId}`
);
const inputTokens =
result.usage?.inputTokens ?? result.usage?.promptTokens ?? 0;
const outputTokens =
result.usage?.outputTokens ?? result.usage?.completionTokens ?? 0;
const totalTokens =
result.usage?.totalTokens ?? inputTokens + outputTokens;
return {
text: result.text,
usage: {
inputTokens: result.usage?.inputTokens || 0,
outputTokens: result.usage?.outputTokens || 0,
totalTokens: result.usage?.totalTokens || 0
inputTokens,
outputTokens,
totalTokens
}
};
} catch (error) {
@@ -296,12 +303,19 @@ export class BaseAIProvider {
`${this.name} generateObject completed successfully for model: ${params.modelId}`
);
const inputTokens =
result.usage?.inputTokens ?? result.usage?.promptTokens ?? 0;
const outputTokens =
result.usage?.outputTokens ?? result.usage?.completionTokens ?? 0;
const totalTokens =
result.usage?.totalTokens ?? inputTokens + outputTokens;
return {
object: result.object,
usage: {
inputTokens: result.usage?.inputTokens || 0,
outputTokens: result.usage?.outputTokens || 0,
totalTokens: result.usage?.totalTokens || 0
inputTokens,
outputTokens,
totalTokens
}
};
} catch (error) {