import os
from openai import OpenAI, APIConnectionError, APIStatusError
with OpenAI(
base_url=os.environ["API_BASE_URL"].rstrip("/"),
api_key=os.environ["API_KEY"],
timeout=60.0,
max_retries=0,
) as client:
try:
result = client.chat.completions.create(
model=os.environ["API_MODEL"],
messages=[{"role": "user", "content": "Say hello in one short sentence."}],
)
print(result.choices[0].message.content or "")
except APIStatusError as error:
raise SystemExit(f"Request failed (HTTP {error.status_code}). Check the request logs.")
except APIConnectionError:
raise SystemExit("Connection failed. Check the API address and network.")