#!/usr/bin/env python3
import os
import anthropic

# 加载环境变量
with open("/home/atai/telegram-bot/.env") as f:
    for line in f:
        line = line.strip()
        if line and not line.startswith("#") and "=" in line:
            key, value = line.split("=", 1)
            os.environ[key] = value

# 初始化客户端
try:
    client = anthropic.Anthropic(
        api_key=os.environ.get('ANTHROPIC_AUTH_TOKEN'),
        base_url=os.environ.get('ANTHROPIC_BASE_URL', 'https://api.anthropic.com')
    )
    
    # 尝试不同的模型名称
    models = [
        "claude-3-5-sonnet-latest",
        "claude-3-5-sonnet-20240620",
        "claude-3-sonnet-20240229",
        "claude-3-opus-20240229"
    ]
    
    for model in models:
        try:
            print(f"\n测试模型: {model}")
            response = client.messages.create(
                model=model,
                max_tokens=50,
                messages=[{"role": "user", "content": "Hi"}]
            )
            print(f"✅ {model} 成功!")
            print(f"响应: {response.content[0].text[:100]}")
            break
        except Exception as e:
            print(f"❌ {model} 失败: {str(e)[:100]}")
    
except Exception as e:
    print(f"\n❌ 错误: {e}")