import requests
import uuid
from datetime import datetime, timezone
# Configuration
API_KEY = "your_api_key"
BASE_URL = "https://api.agentops.ai"
# Create session
session_id = str(uuid.uuid4())
response = requests.post(
f"{BASE_URL}/v2/create_session",
headers={"X-Agentops-Api-Key": API_KEY},
json={
"session": {
"id": session_id,
"init_timestamp": datetime.now(timezone.utc).isoformat(),
"tags": ["example"]
}
}
)
jwt_token = response.json()["jwt"]
# Track LLM call
requests.post(
f"{BASE_URL}/v2/create_events",
headers={"Authorization": f"Bearer {jwt_token}"},
json={
"events": [{
"type": "llm",
"init_timestamp": datetime.now(timezone.utc).isoformat(),
"model": "gpt-4",
"prompt": "Hello, world!",
"completion": "Hi there!",
"prompt_tokens": 3,
"completion_tokens": 2
}]
}
)
# End session
requests.post(
f"{BASE_URL}/v2/update_session",
headers={"Authorization": f"Bearer {jwt_token}"},
json={
"session": {
"id": session_id,
"end_timestamp": datetime.now(timezone.utc).isoformat(),
"end_state": "completed"
}
}
)