23 lines
484 B
Python
23 lines
484 B
Python
# -*- 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)
|