diff --git a/src/core/task/Task.ts b/src/core/task/Task.ts index 23651498e1..493263b9d4 100644 --- a/src/core/task/Task.ts +++ b/src/core/task/Task.ts @@ -3338,12 +3338,10 @@ export class Task extends EventEmitter implements TaskLike { const { profileThresholds = {} } = state ?? {} const { contextTokens } = this.getTokenUsage() - // Get the actual model info from the API handler which includes 1M context updates - const actualModel = this.api.getModel() - const modelInfo = actualModel.info + const modelInfo = this.api.getModel().info const maxTokens = getModelMaxOutputTokens({ - modelId: actualModel.id, + modelId: this.api.getModel().id, model: modelInfo, settings: this.apiConfiguration, }) @@ -3485,12 +3483,10 @@ export class Task extends EventEmitter implements TaskLike { const { contextTokens } = this.getTokenUsage() if (contextTokens) { - // Get the actual model info from the API handler which includes 1M context updates - const actualModel = this.api.getModel() - const modelInfo = actualModel.info + const modelInfo = this.api.getModel().info const maxTokens = getModelMaxOutputTokens({ - modelId: actualModel.id, + modelId: this.api.getModel().id, model: modelInfo, settings: this.apiConfiguration, })