import { PiiGuard } from "@agentium/core";

const guard = new PiiGuard({
  builtIn: ["email", "phone", "ssn", "creditCard"],
  action: "placeholder", // "redact" | "hash" | "placeholder"
  rehydrate: true,       // restore PII in final output
});

const agent = new Agent({
  name: "safe-bot",
  model: openai("gpt-4o"),
  loopHooks: {
    beforeLLMCall: guard.toBeforeLLMCallHook(),
    afterToolExec: guard.toAfterToolExecHook(),
  },
});

const agent = new Agent({
  name: "safe-bot",
  model: openai("gpt-4o"),
  guardrails: {
    input: [guard.toInputGuardrail()],
  },
});

const guard = new PiiGuard({
  patterns: [
    { name: "employee_id", regex: /EMP-\d{6}/g },
    { name: "project_code", regex: /PRJ-[A-Z]{3}-\d{4}/g },
  ],
  action: "placeholder",
});