from arize.otel import register
from openinference.instrumentation.openai import OpenAIInstrumentor
import openai
# Initialize Arize and register the tracer provider
tracer_provider = register(
space_id="your-space-id",
api_key="your-arize-api-key",
project_name="your-project-name",
)
# Instrument OpenAI SDK
OpenAIInstrumentor().instrument(tracer_provider=tracer_provider)
# Configure OpenAI client for OpenRouter
client = openai.OpenAI(
base_url="https://openrouter.ai/api/v1",
api_key="your_openrouter_api_key",
default_headers={
"HTTP-Referer": "<YOUR_SITE_URL>", # Optional: Your site URL
"X-OpenRouter-Title": "<YOUR_SITE_NAME>", # Optional: Your site name
}
)
# Make a traced chat completion request
response = client.chat.completions.create(
model="meta-llama/llama-3.1-8b-instruct:free",
messages=[
{"role": "user", "content": "Write a haiku about observability."}
],
)
# Print the assistant's reply
print(response.choices[0].message.content)