From 2b51d2f5526c879b47957a9dffb68ebeff1a8b8b Mon Sep 17 00:00:00 2001
From: leehack <leehack@gmail.com>
Date: Thu, 12 Nov 2026 12:59:08 -0500
Subject: [PATCH] Pin the lazy-grammar pause and the Seed-OSS call-name
 constraint in tests (#1950)

---
 .../thinking_budget_grammar_test.dart         | 45 +++++++++++++++++++
 .../handlers/seed_oss_handler_test.dart       |  4 +-
 2 files changed, 48 insertions(+), 1 deletion(-)

diff --git a/test/integration/thinking_budget_grammar_test.dart b/test/integration/thinking_budget_grammar_test.dart
index 984687865..f1d06b568 100644
--- a/test/integration/thinking_budget_grammar_test.dart
+++ b/test/integration/thinking_budget_grammar_test.dart
@@ -120,4 +120,49 @@ thought-text ::= [^d]*
       );
     }
   });
+
+  // A lazy grammar is dormant until its trigger, so a budget that leaves it
+  // applied through the thought would lock the reasoning onto its root.
+  test('a lazy grammar stays paused through a spent thinking budget even when '
+      'its root takes the thought', () async {
+    for (final root in ['think-body', 'seed-thought']) {
+      final grammar =
+          'root ::= $root "ok"\n'
+          '$root ::= [^<]* "</think>"\n';
+
+      for (final budget in [1, 3]) {
+        final chunks = await backend
+            .generate(
+              context,
+              prompt,
+              GenerationParams(
+                maxTokens: 48,
+                temp: 0,
+                seed: 1,
+                penalty: 1,
+                grammar: grammar,
+                grammarLazy: true,
+                grammarTriggers: const [
+                  GenerationGrammarTrigger.typed(
+                    type: GrammarTriggerType.word,
+                    value: ' a',
+                  ),
+                ],
+                thinkingBudget: ThinkingBudget(
+                  maxTokens: budget,
+                  startTag: '<think>',
+                  endTag: '</think>',
+                ),
+              ),
+            )
+            .toList();
+
+        expect(
+          utf8.decode(chunks.expand((chunk) => chunk).toList()),
+          isNot(endsWith('</think>ok')),
+          reason: '$root, budget $budget',
+        );
+      }
+    }
+  });
 }
diff --git a/test/unit/core/template/handlers/seed_oss_handler_test.dart b/test/unit/core/template/handlers/seed_oss_handler_test.dart
index 7de91be76..f6ae095a3 100644
--- a/test/unit/core/template/handlers/seed_oss_handler_test.dart
+++ b/test/unit/core/template/handlers/seed_oss_handler_test.dart
@@ -483,7 +483,9 @@ void main() {
     expect(
       grammar.accepts(
         '<seed:think>Plan ${budget.forcedMessage}${budget.endTag}'
-        '<seed:tool_call>\n<function=get_weather>\n</function>',
+        '<seed:tool_call>\n<function=get_time>\n'
+        '<parameter=location>\nParis\n</parameter>\n'
+        '</function>\n</seed:tool_call>',
       ),
       isFalse,
     );