import { ModelCost } from "@modelcost/sdk";
import OpenAI from "openai";

// Initialize
ModelCost.init({
  apiKey: "mc_your_api_key_here",
  orgId: "your-org-id",
  monthlyBudget: 500,
  budgetAction: "block",
});

// Wrap your AI client
const openai = ModelCost.wrap(new OpenAI());

// All calls are now tracked and budget-enforced
const response = await openai.chat.completions.create({
  model: "gpt-4o",
  messages: [{ role: "user", content: "Hello!" }],
});

// Check budget status
const budget = await ModelCost.checkBudget("organization", "your-org-id");
console.log(`Allowed: ${budget.allowed}`);

// Scan for PII
const pii = await ModelCost.scanPii("My SSN is 123-45-6789");
console.log(`PII detected: ${pii.detected}`);

// Shutdown gracefully
await ModelCost.shutdown();