from netra import Netra, ConversationType
import anthropic
import os
Netra.init(headers=f"x-api-key={os.getenv('NETRA_API_KEY')}")
# Set session and user context
Netra.set_session_id("session-123")
Netra.set_user_id("user-456")
client = anthropic.Anthropic(api_key=os.getenv("ANTHROPIC_API_KEY"))
# Track conversation
Netra.add_conversation(
content="What is machine learning?",
conversation_type=ConversationType.INPUT
)
response = client.messages.create(
model="claude-3-5-sonnet-20241022",
max_tokens=1024,
messages=[{"role": "user", "content": "What is machine learning?"}]
)
Netra.add_conversation(
content=response.content[0].text,
conversation_type=ConversationType.OUTPUT
)