refactor: rename chat response token IDs (#35225)

2026-08-17 23:07:43 -07:00
parent 0077f84d37
commit e6df23f3c2
5 changed files with 11 additions and 10 deletions
@@ -213,7 +213,7 @@ class TestSamplingMask(SamplingMaskTestMixin, CustomTestCase):
        self.assertEqual(response.status_code, 200, response.text)

        choice = response.json()["choices"][0]
-        output_ids = choice["token_ids"]
+        output_ids = choice["response_token_ids"]
        meta_info = choice["meta_info"]
        sampling_masks = meta_info["output_token_sampling_mask"]
        sampling_logprobs = meta_info["output_token_sampling_logprobs"]