Tracing Anthropic Claude with Maxim

import os
import dotenv

dotenv.load_dotenv()
MODEL_NAME = "claude-3-5-sonnet-20241022"
ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY")

if not ANTHROPIC_API_KEY:
    raise RuntimeError("Missing ANTHROPIC_API_KEY environment variable")

from maxim import Maxim
logger = Maxim().logger()

user_input = "What was the capital of France in 1800s?"

response = client.messages.create(
    model=MODEL_NAME,
    max_tokens=1024,
    messages=[{"role": "user", "content": user_input}],
    extra_headers={"x-maxim-trace-id": str(uuid4())}
)

print(response)

user_input = "What was the capital of France in 1800s?"
final_response = ""
response_chunks = []

with client.messages.stream(
        max_tokens=1024,
        messages=[{"role": "user", "content": user_input}],
        model=MODEL_NAME,
    ) as stream:
    for text_chunk in stream.text_stream:
        # Collect streamed chunks
        response_chunks.append(text_chunk)
        # Print the streamed text chunk
        print(text_chunk, end="", flush=True)
    final_response = "".join(response_chunks)