File size: 530 Bytes
981b783 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
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()
|