meta-scaler/procurement
0
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 