CoolFace
Apppublic

meta-scaler/procurement

sourceHugging Faceupdated 6mo agoView on Hugging Face
0likes
test_api.py32 linesDownload Raw Back to root
1"""Quick API connection test."""2from pathlib import Path3from dotenv import load_dotenv4 5load_dotenv(Path(__file__).resolve().parent / ".env")6 7import os8 9print("API_BASE_URL:", os.environ.get("API_BASE_URL", "NOT SET"))10print("MODEL_NAME:", os.environ.get("MODEL_NAME", "NOT SET"))11token = os.environ.get("HF_TOKEN", "")12print("HF_TOKEN:", token[:10] + "..." if token else "NOT SET")13print()14 15print("Testing API connection...")16from openai import OpenAI17 18client = OpenAI(19    base_url=os.environ["API_BASE_URL"],20    api_key=os.environ["HF_TOKEN"],21    timeout=30.0,22)23try:24    resp = client.chat.completions.create(25        model=os.environ["MODEL_NAME"],26        messages=[{"role": "user", "content": "Say hello in one word."}],27        max_tokens=10,28    )29    print("SUCCESS:", resp.choices[0].message.content)30except Exception as e:31    print(f"FAILED: {type(e).__name__}: {e}")32