From 5636d16dec9a7986424030ad3816d7cfdfbbaada Mon Sep 17 00:00:00 2001 From: Byron Hsu Date: Mon, 2 Feb 2026 02:38:08 -0800 Subject: [PATCH] [CI] Add logs to debug `TestOpenAIServer.test_completion_stream` (#17471) --- test/registered/openai_server/basic/test_openai_server.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/test/registered/openai_server/basic/test_openai_server.py b/test/registered/openai_server/basic/test_openai_server.py index 33032f134..7d348e579 100644 --- a/test/registered/openai_server/basic/test_openai_server.py +++ b/test/registered/openai_server/basic/test_openai_server.py @@ -114,6 +114,9 @@ class TestOpenAIServer(CustomTestCase): def run_completion_stream( self, echo, logprobs, use_list_input, parallel_sample_num, token_input ): + print( + f"run_completion_stream: {echo=}, {logprobs=}, {use_list_input=}, {parallel_sample_num=}, {token_input=}" + ) client = openai.Client(api_key=self.api_key, base_url=self.base_url) prompt = "The capital of France is" if token_input: @@ -145,6 +148,7 @@ class TestOpenAIServer(CustomTestCase): is_firsts = {} for response in generator: + print(f"{response=}") usage = response.usage if usage is not None: assert usage.prompt_tokens > 0, f"usage.prompt_tokens was zero"