Fix Req array token-id concatenation (#26182)

Co-authored-by: Lianmin Zheng <lianminzheng@gmail.com>
This commit is contained in:
Mohammad Miadh Angkad
2026-06-06 19:59:51 -07:00
committed by GitHub
co-authored by Lianmin Zheng
parent 066b4a2180
commit 52f221cce0
8 changed files with 71 additions and 50 deletions
@@ -7,6 +7,7 @@ register_cpu_ci(est_time=7, suite="base-b-test-cpu")
import json
import unittest
from array import array
from unittest.mock import MagicMock
import torch
@@ -25,8 +26,8 @@ from sglang.test.test_utils import CustomTestCase
# Helper: mock a Req object (used by ThinkingBudget and NGram processors)
def _make_req(origin_input_ids=None, output_ids=None):
req = MagicMock()
req.origin_input_ids = origin_input_ids or []
req.output_ids = output_ids or []
req.origin_input_ids = array("q", origin_input_ids or [])
req.output_ids = array("q", output_ids or [])
return req