OpenAI

pip install netra-sdk openai

npm install netra-sdk openai

from netra import Netra
from openai import OpenAI

# Initialize Netra with OpenAI instrumentation
Netra.init(
    app_name="my-ai-app",
    headers=f"x-api-key={os.environ.get("NETRA_API_KEY")}",
    trace_content=True
)

# Use OpenAI client as usual - all calls are automatically traced
client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))

completion = client.chat.completions.create(
    model="gpt-4",
    messages=[
        {"role": "user", "content": "What is observability?"}
    ]
)

print(completion.choices[0].message.content)

import { Netra } from "netra-sdk";
import OpenAI from "openai";

async function main() {
  // Initialize Netra with OpenAI instrumentation (must await)
  await Netra.init({
    appName: "my-ai-app",
    headers: `x-api-key=${process.env.NETRA_API_KEY}`,
    traceContent: true
  });

  // Use OpenAI client as usual - all calls are automatically traced
  const openai = new OpenAI({
    apiKey: process.env.OPENAI_API_KEY
  });

  const completion = await openai.chat.completions.create({
    model: "gpt-4",
    messages: [
      { role: "user", content: "What is observability?" }
    ]
  });

  console.log(completion.choices[0].message.content);
}

main();

stream = client.chat.completions.create(
    model="gpt-4",
    messages=[{"role": "user", "content": "Tell me a story"}],
    stream=True
)

for chunk in stream:
    if chunk.choices[0].delta.content:
        print(chunk.choices[0].delta.content, end="")

const stream = await openai.chat.completions.create({
  model: "gpt-4",
  messages: [{ role: "user", content: "Tell me a story" }],
  stream: true
});

for await (const chunk of stream) {
  process.stdout.write(chunk.choices[0]?.delta?.content || "");
}

embedding = client.embeddings.create(
    model="text-embedding-ada-002",
    input="The quick brown fox jumps over the lazy dog"
)

print(embedding.data[0].embedding)

const embedding = await openai.embeddings.create({
  model: "text-embedding-ada-002",
  input: "The quick brown fox jumps over the lazy dog"
});

console.log(embedding.data[0].embedding);

from netra import Netra
from netra.instrumentation.instruments import InstrumentSet

# Only enable OpenAI instrumentation
Netra.init(
    headers=f"x-api-key={os.environ.get('NETRA_API_KEY')}",
    instruments={InstrumentSet.OPENAI}
)

# Or block specific instrumentations
Netra.init(
    headers=f"x-api-key={os.environ.get('NETRA_API_KEY')}",
    block_instruments={InstrumentSet.HTTPX}
)

import { Netra, NetraInstruments } from "netra-sdk";

// Only enable OpenAI instrumentation
await Netra.init({
  headers: `x-api-key=${process.env.NETRA_API_KEY}`,
  instruments: new Set([NetraInstruments.OPENAI])
});

// Or block specific instrumentations
await Netra.init({
  headers: `x-api-key=${process.env.NETRA_API_KEY}`,
  blockInstruments: new Set([NetraInstruments.HTTP])
});