40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
#!/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
|
|
|
|
print("环境变量检查:")
|
|
print(f"ANTHROPIC_AUTH_TOKEN: {os.environ.get('ANTHROPIC_AUTH_TOKEN', '未设置')[:30]}...")
|
|
print(f"ANTHROPIC_BASE_URL: {os.environ.get('ANTHROPIC_BASE_URL', '未设置')}")
|
|
|
|
# 初始化客户端
|
|
try:
|
|
client = anthropic.Anthropic(
|
|
api_key=os.environ.get('ANTHROPIC_AUTH_TOKEN'),
|
|
base_url=os.environ.get('ANTHROPIC_BASE_URL', 'https://api.anthropic.com')
|
|
)
|
|
print("\n✅ Claude客户端初始化成功")
|
|
|
|
# 测试API调用
|
|
print("\n测试API调用...")
|
|
response = client.messages.create(
|
|
model="claude-3-5-sonnet-20241022",
|
|
max_tokens=100,
|
|
messages=[
|
|
{"role": "user", "content": "你好，请用一句话介绍自己"}
|
|
]
|
|
)
|
|
|
|
print(f"\n✅ API调用成功!")
|
|
print(f"响应: {response.content[0].text}")
|
|
|
|
except Exception as e:
|
|
print(f"\n❌ 错误: {e}")
|