Files
ai-agents/agents/test_llm2.py
T

39 lines
1.2 KiB
Python

#!/usr/bin/env python3
import json
from urllib import request
API_KEY = "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"
# Try different model names
models = [
"claude-sonnet-4-5-20250514",
"claude-3-5-sonnet-20241022",
"claude-3-haiku-20240307",
"claude-sonnet-4-5-20250514",
]
for model in models:
try:
body = json.dumps({
"model": model,
"max_tokens": 20,
"messages": [{"role": "user", "content": "Say hi"}],
}).encode()
req = request.Request(
"https://api.anthropic.com/v1/messages",
data=body,
headers={
"x-api-key": API_KEY,
"anthropic-version": "2023-06-01",
"content-type": "application/json",
},
method="POST",
)
with request.urlopen(req, timeout=30) as resp:
result = json.loads(resp.read().decode())
text = result.get("content", [{}])[0].get("text", "")
print(f" {model}: OK - {text}")
break
except Exception as e:
print(f" {model}: {e}")