# -*- coding: utf-8 -*-
"""
快速测试 LLM 连接是否正常
"""

import os
from dotenv import load_dotenv
from openai import OpenAI

load_dotenv()

client = OpenAI(
    base_url=os.getenv("OPENAI_BASE_URL", "http://127.0.0.1:9999/v1"),
    api_key=os.getenv("OPENAI_API_KEY", ""),
)

response = client.chat.completions.create(
    model=os.getenv("OPENAI_MODEL", "gpt-3.5-turbo"),
    messages=[{"role": "user", "content": "Hello"}],
)

print(response.choices[0].message.content)