import os
import requests
from dotenv import load_dotenv

load_dotenv()

def test_glm_auth():
    api_key = os.getenv("GLM_API_KEY")
    base_url = os.getenv("GLM_BASE_URL", "https://open.bigmodel.cn/api/paas/v4/chat/completions")
    model = os.getenv("GLM_MODEL", "glm-4v-flash")
    
    print(f"Testing GLM Auth with:")
    print(f"URL: {base_url}")
    print(f"Model: {model}")
    print(f"Key: {api_key[:10]}...{api_key[-5:] if api_key else ''}")

    try:
        response = requests.post(
            base_url,
            headers={
                "Authorization": f"Bearer {api_key}",
                "Content-Type": "application/json"
            },
            json={
                "model": model,
                "messages": [{"role": "user", "content": "hi"}],
                "max_tokens": 10
            },
            timeout=10
        )
        print(f"Status Code: {response.status_code}")
        print(f"Response: {response.text}")
    except Exception as e:
        print(f"Error: {e}")

if __name__ == "__main__":
    test_glm_auth()