OpenAI SDK
from openai import OpenAI
client = OpenAI(
api_key="YOUR_API_KEY",
base_url="https://api.996tokens.com/v1"
)
resp = client.chat.completions.create(
model="yu-chat-auto",
messages=[{"role": "user", "content": "你好"}]
)
print(resp.choices[0].message.content)