DK

!pip install -qq opentelemetry-sdk opentelemetry-exporter-otlp openai openinference-instrumentation-openai arize-otel

import os
from getpass import getpass

SPACE_ID = globals().get("SPACE_ID") or getpass(
    ":key: Enter your Arize Space ID: "
)
API_KEY = globals().get("API_KEY") or getpass(":key: Enter your Arize API Key: ")
OPENAI_API_KEY = globals().get("OPENAI_API_KEY") or getpass(
    ":key: Enter your OpenAI API key: "
)
os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY

# Import open-telemetry dependencies
from arize.otel import register

# Setup OTEL via our convenience function
tracer_provider = register(
    space_id=SPACE_ID,  # in app space settings page
    api_key=API_KEY,  # in app space settings page
    project_name="tracing-haiku-tutorial",  # name this to whatever you would like
)
# Import the automatic instrumentor from OpenInference
from openinference.instrumentation.openai import OpenAIInstrumentor

# Finish automatic instrumentation
OpenAIInstrumentor().instrument(tracer_provider=tracer_provider)

import openai

# Run OpenAI completion
openai_client = openai.OpenAI()
response = openai_client.chat.completions.create(
    model="gpt-3.5-turbo",
    messages=[{"role": "user", "content": "Write a haiku."}],
    max_tokens=20,
)
print(response.choices[0].message.content)