From 10f887d449369a6a6b8b6dc5f9643997a96c914e Mon Sep 17 00:00:00 2001 From: IlyaMescheryakov1402 Date: Wed, 12 Mar 2025 02:20:19 +0300 Subject: [PATCH] add some sugar --- examples/vllm/test_openai_api.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/vllm/test_openai_api.py b/examples/vllm/test_openai_api.py index a6b908b..c3b7d76 100644 --- a/examples/vllm/test_openai_api.py +++ b/examples/vllm/test_openai_api.py @@ -14,7 +14,7 @@ def main(model_name: str): max_tokens=1024, top_p=1.0 ) - print(f"ChatCompletion: \n\n {chat_response.choices[0].message.content}") + print(f"\n\nChatCompletion:\n\n {chat_response.choices[0].message.content}") comp_response = client.completions.create( model=model_name, @@ -22,10 +22,10 @@ def main(model_name: str): temperature=1.0, max_tokens=256 ) - print(f"\n\n Completion: \n\n {comp_response.choices[0].text}") + print(f"\n\nCompletion:\n\n {comp_response.choices[0].text}") fake_body = {"stream": False, "model": model_name, "prompt": "test"} - print(f"Models:\n") + print(f"\n\nModels:\n") print('\n\n'.join(map(str, client.models.list(extra_body=fake_body).data))) return None