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