Agent Development

import { createGuardedAgent } from '@waymakerai/aicofounder-agent-sdk';

const agent = createGuardedAgent({
  // Required: which model to use
  model: 'claude-sonnet-4-20250514',

  // Optional: system instructions for the agent
  instructions: 'You are a helpful customer service agent for Acme Corp.',

  // Configure guards (or pass true for sensible defaults)
  guards: {
    // PII Protection
    pii: {
      mode: 'redact',             // 'detect' | 'redact' | 'block'
      onDetection: 'redact',      // What to do when PII is found
    },

    // Prompt Injection
    injection: {
      sensitivity: 'medium',      // 'low' | 'medium' | 'high'
      onDetection: 'block',       // 'block' | 'warn'
    },

    // Compliance (industry-specific rules)
    compliance: {
      frameworks: ['hipaa', 'gdpr'],
    },

    // Content Filtering
    contentFilter: true,           // Enable toxicity detection

    // Cost Tracking
    cost: {
      budgetPeriod: 'day',
      warningThreshold: 0.8,      // Warn at 80% of budget
    },

    // Audit Logging
    audit: {
      destination: 'file',        // 'console' | 'file' | 'custom'
      filePath: './audit.log',
      tamperProof: true,           // SHA-256 hash chain
      events: ['request', 'response', 'violation', 'cost', 'error'],
    },

    // Rate Limiting
    rateLimit: {
      maxRequests: 100,
      windowMs: 60_000,            // 1 minute window
    },
  },
});

// Run the agent
const result = await agent.run('I need to update my billing address to 123 Main St');

console.log(result.output);         // The AI response (PII redacted if found)
console.log(result.blocked);        // false (unless a guard blocked it)
console.log(result.violations);     // Any guard violations found
console.log(result.cost);           // Cost of this request
console.log(result.tokensUsed);     // { input: 150, output: 200 }
console.log(result.guardsApplied);  // ['rate-limit', 'injection', 'pii', 'compliance', ...]

// Quick setup with all defaults:
// - PII: redact mode
// - Injection: medium sensitivity, block on detection
// - Cost: daily budget tracking with 80% warning
// - Content filter: enabled
// - Audit: console logging
// - Rate limit: 100 requests per minute
const agent = createGuardedAgent({
  model: 'claude-sonnet-4-20250514',
  guards: true,
});

import {
  GuardPipeline,
  PIIInterceptor,
  InjectionInterceptor,
  CostInterceptor,
  AuditInterceptor,
} from '@waymakerai/aicofounder-agent-sdk';

// Build a custom pipeline
const pipeline = new GuardPipeline();

// Add interceptors in your preferred order
pipeline.use(new InjectionInterceptor({ sensitivity: 'high', onDetection: 'block' }));
pipeline.use(new PIIInterceptor({ mode: 'redact', onDetection: 'redact' }));
pipeline.use(new CostInterceptor({ budgetPeriod: 'day', warningThreshold: 0.8 }));
pipeline.use(new AuditInterceptor({
  destination: 'file',
  filePath: './custom-audit.log',
  tamperProof: true,
}));

// Process input
const inputResult = await pipeline.processInput(userMessage, { model: 'claude-sonnet-4-20250514' });

if (inputResult.blocked) {
  console.log('Blocked:', inputResult.reason);
} else {
  const safeInput = inputResult.transformed || userMessage;
  // Send safeInput to your LLM
  const aiResponse = await callYourLLM(safeInput);

  // Process output
  const outputResult = await pipeline.processOutput(aiResponse, { model: 'claude-sonnet-4-20250514' });
  const finalOutput = outputResult.transformed || aiResponse;
}

// Get pipeline stats
console.log(pipeline.stats);
console.log(pipeline.getViolations());
console.log(pipeline.getInterceptorNames());

import { guardTool } from '@waymakerai/aicofounder-agent-sdk';

// Define a tool
const databaseTool = {
  name: 'query_database',
  description: 'Query the customer database',
  parameters: {
    sql: { type: 'string', description: 'SQL query to execute' },
  },
  handler: async ({ sql }) => {
    return await db.query(sql);
  },
};

// Wrap it with guards
const guardedTool = guardTool(databaseTool, {
  pii: { mode: 'redact', onDetection: 'redact' },
  injection: { sensitivity: 'high', onDetection: 'block' },
  audit: { destination: 'file', filePath: './tool-audit.log' },
});

// The guarded tool will:
// 1. Check the SQL input for injection patterns
// 2. Check input/output for PII and redact
// 3. Log the tool call to the audit trail

import {
  createGuardedAgent,
  generateCostReport,
  formatCostReport,
  generateComplianceReport,
  formatComplianceReport,
} from '@waymakerai/aicofounder-agent-sdk';

const agent = createGuardedAgent({
  model: 'claude-sonnet-4-20250514',
  guards: true,
});

// After running the agent multiple times...
await agent.run('First message');
await agent.run('Second message');

// Get the guard report
const report = agent.getGuardReport();
console.log(report.totalRequests);           // 2
console.log(report.totalCost);               // $0.003
console.log(report.ppiDetections);           // 0
console.log(report.injectionAttempts);       // 0
console.log(report.complianceViolations);    // 0
console.log(report.contentFiltered);         // 0
console.log(report.rateLimitHits);           // 0
console.log(report.auditEvents);             // 4 (2 requests + 2 responses)

// Generate formatted reports
const costData = generateCostReport(agent);
console.log(formatCostReport(costData));
// Outputs: per-model token usage, spending, and budget remaining

const complianceData = generateComplianceReport(agent);
console.log(formatComplianceReport(complianceData));
// Outputs: violations by framework, severity breakdown