Vertex AI

pip install netra-sdk google-cloud-aiplatform

npm install netra-sdk @google-cloud/vertexai

from netra import Netra
from vertexai.preview.generative_models import GenerativeModel
import vertexai
import os

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

# Use Vertex AI client as usual - all calls are automatically traced
vertexai.init(project="your-project-id", location="us-central1")

model = GenerativeModel("gemini-pro")

response = model.generate_content("What is observability?")
print(response.text)

import { Netra } from "netra-sdk";
import { VertexAI } from "@google-cloud/vertexai";

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

  // Use Vertex AI client as usual - all calls are automatically traced
  const vertexAI = new VertexAI({
    project: "your-project-id",
    location: "us-central1"
  });

  const model = vertexAI.getGenerativeModel({ model: "gemini-pro" });

  const result = await model.generateContent("What is observability?");
  const response = await result.response;

  console.log(response.text());
}

main();

model = GenerativeModel("gemini-pro")

responses = model.generate_content("Tell me a story", stream=True)

for response in responses:
    print(response.text, end="")

const model = vertexAI.getGenerativeModel({ model: "gemini-pro" });

const result = await model.generateContentStream("Tell me a story");

for await (const chunk of result.stream) {
  const chunkText = chunk.text();
  process.stdout.write(chunkText);
}

model = GenerativeModel("gemini-pro")

chat = model.start_chat()

response1 = chat.send_message("Hello")
print(response1.text)

response2 = chat.send_message("What is machine learning?")
print(response2.text)

const model = vertexAI.getGenerativeModel({ model: "gemini-pro" });

const chat = model.startChat();

const result1 = await chat.sendMessage("Hello");
console.log(result1.response.text());

const result2 = await chat.sendMessage("What is machine learning?");
console.log(result2.response.text());

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

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

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

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