| from openai import OpenAI | |
| import sys | |
| def main(): | |
| openai_api_key = "EMPTY" | |
| api_base = f"http://{sys.argv[1]}/v1" | |
| client = OpenAI( | |
| api_key=openai_api_key, | |
| base_url=api_base, | |
| ) | |
| try: | |
| completion = client.chat.completions.create( | |
| model='qwen', | |
| messages=[{'role': 'user', 'content': 'Hello!'}], | |
| timeout=15 | |
| ) | |
| print(completion.choices[0].message.content) | |
| except Exception: | |
| exit(1) | |
| if __name__ == '__main__': | |
| main() | |