feat: add forward timeout (#17831)

Co-authored-by: qiuxuan.lzw <qiuxuan.lzw@alibaba-inc.com>
This commit is contained in:
StonyPort
2026-01-30 08:52:29 +08:00
committed by GitHub
co-authored by qiuxuan.lzw
parent 6a6b36367e
commit 2b3408ff14
4 changed files with 63 additions and 1 deletions
+36
View File
@@ -283,5 +283,41 @@ class TestAbortWithQueueTimeout(CustomTestCase):
self.assertEqual(error_count, 1)
class TestAbortWithForwardTimeout(CustomTestCase):
@classmethod
def setUpClass(cls):
cls.model = DEFAULT_MODEL_NAME_FOR_TEST
cls.base_url = DEFAULT_URL_FOR_TEST
with envs.SGLANG_FORWARD_TIMEOUT_MS.override(
1
), envs.SGLANG_ENABLE_HEALTH_ENDPOINT_GENERATION.override(False):
cls.process = popen_launch_server(
cls.model,
cls.base_url,
timeout=DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
other_args=["--skip-server-warmup"],
)
@classmethod
def tearDownClass(cls):
kill_process_tree(cls.process.pid)
def test_forward_timeout(self):
response = requests.post(
self.base_url + "/generate",
json={
"text": "Today is ",
"sampling_params": {
"temperature": 0,
"max_new_tokens": 512,
"ignore_eos": True,
},
},
)
result = response.json()
self.assertEqual(result["object"], "error")
self.assertEqual(result["code"], 503)
if __name__ == "__main__":
unittest.main()