From 656b6cb4a2c1441c36847b4654cb3c3970293625 Mon Sep 17 00:00:00 2001
From: Will Chen <willchen90@gmail.com>
Date: Wed, 17 Dec 2025 22:19:37 -0800
Subject: [PATCH] fix thinking budget e2e (#1981)

---
 e2e-tests/fixtures/engine/1.md                |  1 +
 ...nking_budget.spec.ts_thinking-budget-2.txt |  8 +++++++
 ...nking_budget.spec.ts_thinking-budget-4.txt | 16 +++++++++++++
 ...nking_budget.spec.ts_thinking-budget-6.txt | 24 +++++++++++++++++++
 4 files changed, 49 insertions(+)
 create mode 100644 e2e-tests/fixtures/engine/1.md

diff --git a/e2e-tests/fixtures/engine/1.md b/e2e-tests/fixtures/engine/1.md
new file mode 100644
index 0000000..d00491f
--- /dev/null
+++ b/e2e-tests/fixtures/engine/1.md
@@ -0,0 +1 @@
+1
diff --git a/e2e-tests/snapshots/thinking_budget.spec.ts_thinking-budget-2.txt b/e2e-tests/snapshots/thinking_budget.spec.ts_thinking-budget-2.txt
index 4334f60..f0db542 100644
--- a/e2e-tests/snapshots/thinking_budget.spec.ts_thinking-budget-2.txt
+++ b/e2e-tests/snapshots/thinking_budget.spec.ts_thinking-budget-2.txt
@@ -8,6 +8,14 @@
         "role": "system",
         "content": "[[SYSTEM_MESSAGE]]"
       },
+      {
+        "role": "user",
+        "content": "tc=1"
+      },
+      {
+        "role": "assistant",
+        "content": "1"
+      },
       {
         "role": "user",
         "content": "[dump] hi"
diff --git a/e2e-tests/snapshots/thinking_budget.spec.ts_thinking-budget-4.txt b/e2e-tests/snapshots/thinking_budget.spec.ts_thinking-budget-4.txt
index aa87e68..d2ee45a 100644
--- a/e2e-tests/snapshots/thinking_budget.spec.ts_thinking-budget-4.txt
+++ b/e2e-tests/snapshots/thinking_budget.spec.ts_thinking-budget-4.txt
@@ -8,6 +8,22 @@
         "role": "system",
         "content": "[[SYSTEM_MESSAGE]]"
       },
+      {
+        "role": "user",
+        "content": "tc=1"
+      },
+      {
+        "role": "assistant",
+        "content": "1"
+      },
+      {
+        "role": "user",
+        "content": "[dump] hi"
+      },
+      {
+        "role": "assistant",
+        "content": "[[dyad-dump-path=*]]"
+      },
       {
         "role": "user",
         "content": "[dump] hi"
diff --git a/e2e-tests/snapshots/thinking_budget.spec.ts_thinking-budget-6.txt b/e2e-tests/snapshots/thinking_budget.spec.ts_thinking-budget-6.txt
index 7118400..b28ba32 100644
--- a/e2e-tests/snapshots/thinking_budget.spec.ts_thinking-budget-6.txt
+++ b/e2e-tests/snapshots/thinking_budget.spec.ts_thinking-budget-6.txt
@@ -8,6 +8,30 @@
         "role": "system",
         "content": "[[SYSTEM_MESSAGE]]"
       },
+      {
+        "role": "user",
+        "content": "tc=1"
+      },
+      {
+        "role": "assistant",
+        "content": "1"
+      },
+      {
+        "role": "user",
+        "content": "[dump] hi"
+      },
+      {
+        "role": "assistant",
+        "content": "[[dyad-dump-path=*]]"
+      },
+      {
+        "role": "user",
+        "content": "[dump] hi"
+      },
+      {
+        "role": "assistant",
+        "content": "[[dyad-dump-path=*]]"
+      },
       {
         "role": "user",
         "content": "[dump] hi"