[Refactor] StructuredOutputManager x Speculative Decoding Refactor (#48200)

2026-09-15 09:48:49 -04:00
parent 64856080b0
commit e6960af33b
12 changed files with 596 additions and 1192 deletions
@@ -185,8 +185,8 @@ class TestReasoningEndTokenIds:
    def test_find_reasoning_end_offset_returns_first_match(self):
        engine = _make_engine()
        assert engine.find_reasoning_end_offset([5, 201, 6, 201]) == 1
-        assert engine.find_reasoning_end_offset([5, 6]) is None
-        assert engine.find_reasoning_end_offset([]) is None
+        assert engine.find_reasoning_end_offset([5, 6]) == 2
+        assert engine.find_reasoning_end_offset([]) == 0
        # Rejected-draft placeholders never match.
        assert engine.find_reasoning_end_offset([-1, 201]) == 1