Observability for OpenAI with Highlight

Step 1

Install the Traceloop SDK and initialize it. It will automatically log all calls to OpenAI, with prompts and completions as separate spans.

import openai
from traceloop.sdk import Traceloop
from traceloop.sdk.decorators import workflow

Traceloop.init(app_name="joke_generation_service")

@workflow(name="joke_creation")
def create_joke():
    completion = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[{"role": "user", "content": "Tell me a joke about opentelemetry"}],
    )

    return completion.choices[0].message.content

import * as traceloop from "@traceloop/node-server-sdk";
import OpenAI from "openai";

Traceloop.init({ app_name="joke_generation_service" })
const openai = new OpenAI();

class MyLLM {
  @traceloop.workflow("joke_creation")
  async create_joke():
      completion = await openai.chat.completions.create({
          model: "gpt-3.5-turbo",
          messages: [{"role": "user", "content": "Tell me a joke about opentelemetry"}],
      })

      return completion.choices[0].message.content
}