PYTHON - ANTHROPIC - API
Simon-Pierre Boucher
2024-09-14
In [1]:
import os
import requests
import json
def call_claude_api(
model="claude-3-5-sonnet-20240620",
messages=None,
max_tokens=1024,
anthropic_version="2023-06-01"
):
# Get the API key from environment variables
api_key = os.getenv("ANTHROPIC_API_KEY")
if not api_key:
raise ValueError("API key not found in environment variables. Please set ANTHROPIC_API_KEY.")
url = "https://api.anthropic.com/v1/messages"
headers = {
"x-api-key": api_key,
"anthropic-version": anthropic_version,
"content-type": "application/json"
}
body = {
"model": model,
"max_tokens": max_tokens,
"messages": messages
}
response = requests.post(url, headers=headers, data=json.dumps(body))
if response.status_code != 200:
raise ValueError(f"API request failed: {response.text}")
result = response.json()
return result
In [2]:
# Define the messages
messages = [
{"role": "user", "content": "give me a list of gift idea for my girlfriend"}
]
In [3]:
# Call the function
response = call_claude_api(
model="claude-3-5-sonnet-20240620",
messages=messages,
max_tokens=1024
)
# Display the assistant's response
print(response['content'][0]['text'])