From 88d693cc005e733ea9a5e87299805a5768bb5a14 Mon Sep 17 00:00:00 2001
From: Anton Kuzmenko <1917237+default-anton@users.noreply.github.com>
Date: Sun, 28 Dec 2025 16:33:25 -0800
Subject: [PATCH] test: add thinkingBudgetTokens parameter to thinking mode
 tests

---
 packages/ai/test/stream.test.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/packages/ai/test/stream.test.ts b/packages/ai/test/stream.test.ts
index 9a05b48d..6389f49b 100644
--- a/packages/ai/test/stream.test.ts
+++ b/packages/ai/test/stream.test.ts
@@ -572,11 +572,11 @@ describe("Generate E2E Tests", () => {
 		});
 
 		it.skip("should handle thinking mode", { retry: 3 }, async () => {
-			await handleThinking(llm, { thinkingEnabled: true });
+			await handleThinking(llm, { thinkingEnabled: true, thinkingBudgetTokens: 2048 });
 		});
 
 		it("should handle multi-turn with thinking and tools", { retry: 3 }, async () => {
-			await multiTurn(llm, { thinkingEnabled: true });
+			await multiTurn(llm, { thinkingEnabled: true, thinkingBudgetTokens: 2048 });
 		});
 	});