npm install safety-agent

import { createClient } from "safety-agent";

const client = createClient();

// Guard: Detect threats
const guardResult = await client.guard({
  input: "user message to analyze"
});

// Redact: Remove PII
const redactResult = await client.redact({
  input: "My email is john@example.com",
  model: "openai/gpt-4o-mini"
});

// Scan: Analyze repository for AI agent attacks
const scanResult = await client.scan({
  repo: "https://github.com/user/repo"
});

const client = createClient({
  apiKey: "your-api-key" // Or set SUPERAGENT_API_KEY env var
});

const client = createClient({
  enableFallback: true,           // Enable fallback (default: true)
  fallbackTimeoutMs: 5000,        // Timeout before fallback (default: 5000ms)
  fallbackUrl: "https://..."      // Custom fallback URL (optional)
});

const result = await client.guard({
  input: "user message to analyze",
  model: "google/gemini-2.5-flash-lite",
  fallbackModel: "google/gemini-2.5-pro"
});

const result = await client.guard({
  input: "user message to analyze",
  model: "google/gemini-2.5-flash-lite",
  fallbackModel: "openai/gpt-4o-mini"
});

const result = await client.guard({
  input: "user message to analyze"
});

if (result.classification === "block") {
  console.log("Blocked:", result.violation_types);
  console.log("Reason:", result.reasoning);
}

// URL input
const result = await client.guard({
  input: "https://example.com/document.pdf"
});

// Image input (requires vision model)
const result = await client.guard({
  input: imageBlob,
  model: "openai/gpt-4o"
});

const result = await client.redact({
  input: "My email is john@example.com and SSN is 123-45-6789",
  model: "openai/gpt-4o-mini"
});

console.log(result.redacted);
// "My email is <EMAIL_REDACTED> and SSN is <SSN_REDACTED>"

const result = await client.redact({
  input: "My email is john@example.com",
  model: "openai/gpt-4o-mini",
  rewrite: true
});

console.log(result.redacted);
// "My email is on file"

const result = await client.redact({
  input: "Contact john@example.com or call 555-123-4567",
  model: "openai/gpt-4o-mini",
  entities: ["email addresses"] // Only redact emails
});

const response = await client.scan({
  repo: "https://github.com/user/repo"
});

console.log(response.result);  // Security report
console.log(`Cost: $${response.usage.cost.toFixed(4)}`);

# Required for scan()
export DAYTONA_API_KEY=your-daytona-api-key

# Required for the model (at least one)
export ANTHROPIC_API_KEY=your-anthropic-key
export OPENAI_API_KEY=your-openai-key

const response = await client.scan({
  repo: "https://github.com/user/repo",
  branch: "feature-branch",
  model: "anthropic/claude-sonnet-4-5"
});

console.log("Security Report:");
console.log(response.result);
console.log(`\nTokens: ${response.usage.inputTokens} in, ${response.usage.outputTokens} out`);
console.log(`Cost: $${response.usage.cost.toFixed(4)}`);
}