Watsonx AI

pip install netra-sdk ibm-watsonx-ai

npm install netra-sdk @ibm-cloud/watsonx-ai

from netra import Netra
from ibm_watsonx_ai.foundation_models import Model
from ibm_watsonx_ai.metanames import GenTextParamsMetaNames as GenParams
import os

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

# Use Watsonx client as usual - all calls are automatically traced
model = Model(
    model_id="ibm/granite-13b-chat-v2",
    credentials={
        "url": "https://us-south.ml.cloud.ibm.com",
        "apikey": os.environ.get("WATSONX_API_KEY")
    },
    project_id=os.environ.get("WATSONX_PROJECT_ID")
)

response = model.generate_text(prompt="What is observability?")
print(response)

import { Netra } from "netra-sdk";
import WatsonxAI from "@ibm-cloud/watsonx-ai";

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

// Use Watsonx client as usual - all calls are automatically traced
const watsonx = new WatsonxAI({
  version: "2024-05-31",
  serviceUrl: "https://us-south.ml.cloud.ibm.com",
  apikey: process.env.WATSONX_API_KEY
});

async function main() {
  const response = await watsonx.generateText({
    input: "What is observability?",
    modelId: "ibm/granite-13b-chat-v2",
    projectId: process.env.WATSONX_PROJECT_ID
  });

  console.log(response.results[0].generated_text);
}

main();

model = Model(
    model_id="ibm/granite-13b-chat-v2",
    credentials={
        "url": "https://us-south.ml.cloud.ibm.com",
        "apikey": os.environ.get("WATSONX_API_KEY")
    },
    project_id=os.environ.get("WATSONX_PROJECT_ID")
)

for chunk in model.generate_text_stream(prompt="Tell me a story"):
    print(chunk, end="")

const stream = await watsonx.generateTextStream({
  input: "Tell me a story",
  modelId: "ibm/granite-13b-chat-v2",
  projectId: process.env.WATSONX_PROJECT_ID
});

for await (const chunk of stream) {
  if (chunk.results && chunk.results[0]) {
    process.stdout.write(chunk.results[0].generated_text);
  }
}

generate_params = {
    GenParams.MAX_NEW_TOKENS: 100,
    GenParams.TEMPERATURE: 0.7,
    GenParams.TOP_P: 0.9
}

model = Model(
    model_id="ibm/granite-13b-chat-v2",
    credentials={
        "url": "https://us-south.ml.cloud.ibm.com",
        "apikey": os.environ.get("WATSONX_API_KEY")
    },
    project_id=os.environ.get("WATSONX_PROJECT_ID"),
    params=generate_params
)

response = model.generate_text(prompt="Explain AI")
print(response)

const response = await watsonx.generateText({
  input: "Explain AI",
  modelId: "ibm/granite-13b-chat-v2",
  projectId: process.env.WATSONX_PROJECT_ID,
  parameters: {
    max_new_tokens: 100,
    temperature: 0.7,
    top_p: 0.9
  }
});

console.log(response.results[0].generated_text);

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

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

# 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 Watsonx instrumentation
await Netra.init({
  headers: `x-api-key=${process.env.NETRA_API_KEY}`,
  instruments: new Set([NetraInstruments.WATSONX])
});

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