ros/claude-task-master
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: apply requested changes

Browse Source
This commit is contained in:
Ralph Khreish
2025-10-03 16:14:08 +02:00
parent f3deacb74e
commit 33386cdce8
1 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/ai-providers/base-provider.js
View File

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