promptguard-sdk 0.1.0

package/README.md

@@ -0,0 +1,173 @@
+# PromptGuard Node.js SDK
+
+Drop-in security for AI applications. No code changes required.
+
+## Installation
+
+```bash
+npm install promptguard-sdk
+# or
+yarn add promptguard-sdk
+# or
+pnpm add promptguard-sdk
+```
+
+## Quick Start
+
+```typescript
+import { PromptGuard } from 'promptguard-sdk';
+
+// Initialize client
+const pg = new PromptGuard({ apiKey: 'pg_xxx' });
+
+// Use exactly like OpenAI client
+const response = await pg.chat.completions.create({
+  model: 'gpt-4',
+  messages: [{ role: 'user', content: 'Hello!' }]
+});
+
+console.log(response.choices[0].message.content);
+```
+
+## Drop-in Replacement
+
+If you're already using OpenAI's Node.js client, just change the import:
+
+```typescript
+// Before
+import OpenAI from 'openai';
+const client = new OpenAI();
+
+// After
+import { PromptGuard } from 'promptguard-sdk';
+const client = new PromptGuard({ apiKey: 'pg_xxx' });
+
+// Your existing code works unchanged!
+```
+
+## Features
+
+### Security Scanning
+
+```typescript
+// Scan content for threats
+const result = await pg.security.scan('Ignore previous instructions...');
+
+if (result.blocked) {
+  console.log(`Threat detected: ${result.reason}`);
+}
+```
+
+### PII Redaction
+
+```typescript
+// Redact PII before sending to LLM
+const result = await pg.security.redact(
+  'My email is john@example.com and SSN is 123-45-6789'
+);
+
+console.log(result.redactedContent);
+// Output: "My email is [EMAIL] and SSN is [SSN]"
+```
+
+### Secure Web Scraping
+
+```typescript
+// Scan scraped content for indirect prompt injection
+const result = await pg.scrape.scanUrl('https://example.com/article');
+
+if (result.isSafe) {
+  // Use content safely
+  console.log(result.content);
+} else {
+  console.log('Threats detected:', result.threats);
+}
+```
+
+### AI Agent Security
+
+```typescript
+// Validate tool calls before execution
+const validation = await pg.agent.validateTool({
+  agentId: 'agent-123',
+  toolName: 'execute_shell',
+  arguments: { command: 'ls -la' }
+});
+
+if (validation.action === 'block') {
+  console.log(`Blocked: ${validation.reason}`);
+}
+```
+
+## Configuration
+
+```typescript
+import { PromptGuard, Config } from '@promptguard/sdk';
+
+const config: Config = {
+  apiKey: 'pg_xxx',
+  baseUrl: 'https://api.promptguard.co/api/v1',
+  timeout: 30000,
+  enableCaching: true,
+  enableSecurityScan: true,
+};
+
+const pg = new PromptGuard(config);
+```
+
+## Environment Variables
+
+```bash
+export PROMPTGUARD_API_KEY="pg_xxx"
+export PROMPTGUARD_BASE_URL="https://api.promptguard.co/api/v1"
+```
+
+Then just:
+
+```typescript
+import { PromptGuard } from 'promptguard-sdk';
+
+const pg = new PromptGuard(); // Uses env vars automatically
+```
+
+## Error Handling
+
+```typescript
+import { PromptGuard, PromptGuardError } from '@promptguard/sdk';
+
+try {
+  const response = await pg.chat.completions.create({...});
+} catch (error) {
+  if (error instanceof PromptGuardError) {
+    if (error.code === 'BLOCKED') {
+      console.log(`Request blocked: ${error.message}`);
+    } else if (error.code === 'RATE_LIMITED') {
+      console.log('Rate limited, try again later');
+    }
+  }
+  throw error;
+}
+```
+
+## TypeScript Support
+
+Full TypeScript support with type definitions included.
+
+```typescript
+import type { 
+  ChatCompletionRequest, 
+  ChatCompletionResponse,
+  SecurityScanResult 
+} from '@promptguard/sdk';
+```
+
+## Links
+
+- [Documentation](https://docs.promptguard.co)
+- [GitHub](https://github.com/acebot712/promptguard)
+- [Issues](https://github.com/acebot712/promptguard/issues)
+
+## License
+
+MIT
+

package/dist/index.d.ts

@@ -0,0 +1,197 @@
+/**
+ * PromptGuard Node.js SDK
+ *
+ * Drop-in security for AI applications.
+ * Just change your base URL and add an API key.
+ *
+ * @example
+ * import { PromptGuard } from 'promptguard';
+ *
+ * const pg = new PromptGuard({ apiKey: 'pg_xxx' });
+ *
+ * const response = await pg.chat.completions.create({
+ *   model: 'gpt-4',
+ *   messages: [{ role: 'user', content: 'Hello!' }]
+ * });
+ */
+export interface PromptGuardConfig {
+    apiKey: string;
+    baseUrl?: string;
+    timeout?: number;
+    enableCaching?: boolean;
+    enableSecurityScan?: boolean;
+}
+export interface Message {
+    role: "system" | "user" | "assistant";
+    content: string;
+}
+export interface ChatCompletionRequest {
+    model: string;
+    messages: Message[];
+    temperature?: number;
+    maxTokens?: number;
+    stream?: boolean;
+    [key: string]: unknown;
+}
+export interface ChatCompletionResponse {
+    id: string;
+    object: string;
+    created: number;
+    model: string;
+    choices: Array<{
+        index: number;
+        message: Message;
+        finish_reason: string;
+    }>;
+    usage?: {
+        prompt_tokens: number;
+        completion_tokens: number;
+        total_tokens: number;
+    };
+}
+export interface SecurityScanResult {
+    blocked: boolean;
+    decision: "allow" | "block" | "redact";
+    reason?: string;
+    threatType?: string;
+    confidence?: number;
+}
+export interface RedactResult {
+    original: string;
+    redacted: string;
+    piiFound: string[];
+}
+export interface MemoryEntry {
+    id: string;
+    content: string;
+    memoryType: string;
+    createdAt: number;
+}
+export interface MemoryRetrieveResult {
+    memories: MemoryEntry[];
+    formattedContext: string;
+    tokensSaved: number;
+}
+declare class ChatCompletions {
+    private client;
+    constructor(client: PromptGuard);
+    create(params: ChatCompletionRequest): Promise<ChatCompletionResponse>;
+}
+declare class Chat {
+    completions: ChatCompletions;
+    constructor(client: PromptGuard);
+}
+declare class Security {
+    private client;
+    constructor(client: PromptGuard);
+    scan(content: string, type?: "prompt" | "response"): Promise<SecurityScanResult>;
+    redact(content: string, piiTypes?: string[]): Promise<RedactResult>;
+}
+declare class Memory {
+    private client;
+    constructor(client: PromptGuard);
+    store(agentId: string, content: string, options?: {
+        sessionId?: string;
+        metadata?: Record<string, unknown>;
+        importance?: number;
+    }): Promise<MemoryEntry>;
+    retrieve(agentId: string, query: string, options?: {
+        topK?: number;
+        minRelevance?: number;
+        sessionId?: string;
+    }): Promise<MemoryRetrieveResult>;
+    clear(agentId: string, sessionId?: string): Promise<{
+        entries_deleted: number;
+    }>;
+    stats(): Promise<Record<string, unknown>>;
+}
+export interface ScrapeResult {
+    url: string;
+    status: "safe" | "blocked";
+    content: string;
+    threats_detected: string[];
+    message?: string;
+}
+declare class Scrape {
+    private client;
+    constructor(client: PromptGuard);
+    url(url: string, options?: {
+        renderJs?: boolean;
+        extractText?: boolean;
+        timeout?: number;
+    }): Promise<ScrapeResult>;
+    batch(urls: string[], options?: Record<string, unknown>): Promise<{
+        job_id: string;
+    }>;
+}
+export interface ToolValidationResult {
+    allowed: boolean;
+    risk_score: number;
+    risk_level: string;
+    reason: string;
+    warnings: string[];
+    blocked_reasons: string[];
+}
+export interface BehaviorAnalysisResult {
+    is_normal: boolean;
+    anomaly_score: number;
+    detected_patterns: string[];
+    recommendations: string[];
+}
+declare class Agent {
+    private client;
+    constructor(client: PromptGuard);
+    validateTool(agentId: string, toolName: string, args: Record<string, unknown>, sessionId?: string): Promise<ToolValidationResult>;
+    analyzeBehavior(agentId: string, recentActions: Array<{
+        tool_name: string;
+        arguments: Record<string, unknown>;
+    }>, sessionId?: string): Promise<BehaviorAnalysisResult>;
+    stats(agentId: string): Promise<Record<string, unknown>>;
+}
+export interface RedTeamTestResult {
+    test_name: string;
+    prompt: string;
+    decision: string;
+    reason: string;
+    threat_type?: string;
+    confidence: number;
+    blocked: boolean;
+    details: Record<string, unknown>;
+}
+export interface RedTeamSummary {
+    total_tests: number;
+    blocked: number;
+    allowed: number;
+    block_rate: number;
+    results: RedTeamTestResult[];
+}
+declare class RedTeam {
+    private client;
+    private base;
+    constructor(client: PromptGuard);
+    listTests(): Promise<{
+        total: number;
+        tests: Array<Record<string, unknown>>;
+    }>;
+    runTest(testName: string, targetPreset?: string): Promise<RedTeamTestResult>;
+    runAll(targetPreset?: string): Promise<RedTeamSummary>;
+    runCustom(prompt: string, targetPreset?: string): Promise<RedTeamTestResult>;
+}
+export declare class PromptGuard {
+    private config;
+    chat: Chat;
+    security: Security;
+    memory: Memory;
+    scrape: Scrape;
+    agent: Agent;
+    redteam: RedTeam;
+    constructor(config: PromptGuardConfig);
+    request<T>(method: string, path: string, body?: unknown): Promise<T>;
+}
+export declare class PromptGuardError extends Error {
+    code: string;
+    statusCode: number;
+    constructor(message: string, code: string, statusCode: number);
+}
+export default PromptGuard;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAEH,MAAM,WAAW,iBAAiB;IAChC,MAAM,EAAE,MAAM,CAAA;IACd,OAAO,CAAC,EAAE,MAAM,CAAA;IAChB,OAAO,CAAC,EAAE,MAAM,CAAA;IAChB,aAAa,CAAC,EAAE,OAAO,CAAA;IACvB,kBAAkB,CAAC,EAAE,OAAO,CAAA;CAC7B;AAED,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAA;IACrC,OAAO,EAAE,MAAM,CAAA;CAChB;AAED,MAAM,WAAW,qBAAqB;IACpC,KAAK,EAAE,MAAM,CAAA;IACb,QAAQ,EAAE,OAAO,EAAE,CAAA;IACnB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,MAAM,CAAC,EAAE,OAAO,CAAA;IAChB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;CACvB;AAED,MAAM,WAAW,sBAAsB;IACrC,EAAE,EAAE,MAAM,CAAA;IACV,MAAM,EAAE,MAAM,CAAA;IACd,OAAO,EAAE,MAAM,CAAA;IACf,KAAK,EAAE,MAAM,CAAA;IACb,OAAO,EAAE,KAAK,CAAC;QACb,KAAK,EAAE,MAAM,CAAA;QACb,OAAO,EAAE,OAAO,CAAA;QAChB,aAAa,EAAE,MAAM,CAAA;KACtB,CAAC,CAAA;IACF,KAAK,CAAC,EAAE;QACN,aAAa,EAAE,MAAM,CAAA;QACrB,iBAAiB,EAAE,MAAM,CAAA;QACzB,YAAY,EAAE,MAAM,CAAA;KACrB,CAAA;CACF;AAED,MAAM,WAAW,kBAAkB;IACjC,OAAO,EAAE,OAAO,CAAA;IAChB,QAAQ,EAAE,OAAO,GAAG,OAAO,GAAG,QAAQ,CAAA;IACtC,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,UAAU,CAAC,EAAE,MAAM,CAAA;CACpB;AAED,MAAM,WAAW,YAAY;IAC3B,QAAQ,EAAE,MAAM,CAAA;IAChB,QAAQ,EAAE,MAAM,CAAA;IAChB,QAAQ,EAAE,MAAM,EAAE,CAAA;CACnB;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAA;IACV,OAAO,EAAE,MAAM,CAAA;IACf,UAAU,EAAE,MAAM,CAAA;IAClB,SAAS,EAAE,MAAM,CAAA;CAClB;AAED,MAAM,WAAW,oBAAoB;IACnC,QAAQ,EAAE,WAAW,EAAE,CAAA;IACvB,gBAAgB,EAAE,MAAM,CAAA;IACxB,WAAW,EAAE,MAAM,CAAA;CACpB;AAED,cAAM,eAAe;IACnB,OAAO,CAAC,MAAM,CAAa;gBAEf,MAAM,EAAE,WAAW;IAIzB,MAAM,CAAC,MAAM,EAAE,qBAAqB,GAAG,OAAO,CAAC,sBAAsB,CAAC;CAQ7E;AAED,cAAM,IAAI;IACR,WAAW,EAAE,eAAe,CAAA;gBAEhB,MAAM,EAAE,WAAW;CAGhC;AAED,cAAM,QAAQ;IACZ,OAAO,CAAC,MAAM,CAAa;gBAEf,MAAM,EAAE,WAAW;IAIzB,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,GAAE,QAAQ,GAAG,UAAqB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAO1F,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC;CAM1E;AAED,cAAM,MAAM;IACV,OAAO,CAAC,MAAM,CAAa;gBAEf,MAAM,EAAE,WAAW;IAIzB,KAAK,CACT,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE;QACR,SAAS,CAAC,EAAE,MAAM,CAAA;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;QAClC,UAAU,CAAC,EAAE,MAAM,CAAA;KACpB,GACA,OAAO,CAAC,WAAW,CAAC;IAUjB,QAAQ,CACZ,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE;QACR,IAAI,CAAC,EAAE,MAAM,CAAA;QACb,YAAY,CAAC,EAAE,MAAM,CAAA;QACrB,SAAS,CAAC,EAAE,MAAM,CAAA;KACnB,GACA,OAAO,CAAC,oBAAoB,CAAC;IAU1B,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,eAAe,EAAE,MAAM,CAAA;KAAE,CAAC;IAKhF,KAAK,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAGhD;AAED,MAAM,WAAW,YAAY;IAC3B,GAAG,EAAE,MAAM,CAAA;IACX,MAAM,EAAE,MAAM,GAAG,SAAS,CAAA;IAC1B,OAAO,EAAE,MAAM,CAAA;IACf,gBAAgB,EAAE,MAAM,EAAE,CAAA;IAC1B,OAAO,CAAC,EAAE,MAAM,CAAA;CACjB;AAED,cAAM,MAAM;IACV,OAAO,CAAC,MAAM,CAAa;gBAEf,MAAM,EAAE,WAAW;IAIzB,GAAG,CACP,GAAG,EAAE,MAAM,EACX,OAAO,CAAC,EAAE;QACR,QAAQ,CAAC,EAAE,OAAO,CAAA;QAClB,WAAW,CAAC,EAAE,OAAO,CAAA;QACrB,OAAO,CAAC,EAAE,MAAM,CAAA;KACjB,GACA,OAAO,CAAC,YAAY,CAAC;IASlB,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC;QAAE,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC;CAM5F;AAED,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,OAAO,CAAA;IAChB,UAAU,EAAE,MAAM,CAAA;IAClB,UAAU,EAAE,MAAM,CAAA;IAClB,MAAM,EAAE,MAAM,CAAA;IACd,QAAQ,EAAE,MAAM,EAAE,CAAA;IAClB,eAAe,EAAE,MAAM,EAAE,CAAA;CAC1B;AAED,MAAM,WAAW,sBAAsB;IACrC,SAAS,EAAE,OAAO,CAAA;IAClB,aAAa,EAAE,MAAM,CAAA;IACrB,iBAAiB,EAAE,MAAM,EAAE,CAAA;IAC3B,eAAe,EAAE,MAAM,EAAE,CAAA;CAC1B;AAED,cAAM,KAAK;IACT,OAAO,CAAC,MAAM,CAAa;gBAEf,MAAM,EAAE,WAAW;IAIzB,YAAY,CAChB,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,oBAAoB,CAAC;IAS1B,eAAe,CACnB,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,KAAK,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;KAAE,CAAC,EAC/E,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,sBAAsB,CAAC;IAQ5B,KAAK,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAG/D;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,MAAM,CAAA;IACjB,MAAM,EAAE,MAAM,CAAA;IACd,QAAQ,EAAE,MAAM,CAAA;IAChB,MAAM,EAAE,MAAM,CAAA;IACd,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,UAAU,EAAE,MAAM,CAAA;IAClB,OAAO,EAAE,OAAO,CAAA;IAChB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CACjC;AAED,MAAM,WAAW,cAAc;IAC7B,WAAW,EAAE,MAAM,CAAA;IACnB,OAAO,EAAE,MAAM,CAAA;IACf,OAAO,EAAE,MAAM,CAAA;IACf,UAAU,EAAE,MAAM,CAAA;IAClB,OAAO,EAAE,iBAAiB,EAAE,CAAA;CAC7B;AAED,cAAM,OAAO;IACX,OAAO,CAAC,MAAM,CAAa;IAC3B,OAAO,CAAC,IAAI,CAAsB;gBAEtB,MAAM,EAAE,WAAW;IAIzB,SAAS,IAAI,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAA;KAAE,CAAC;IAI9E,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,YAAY,SAAY,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAM/E,MAAM,CAAC,YAAY,SAAY,GAAG,OAAO,CAAC,cAAc,CAAC;IAMzD,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,YAAY,SAAY,GAAG,OAAO,CAAC,iBAAiB,CAAC;CAMtF;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,MAAM,CAA6B;IAG3C,IAAI,EAAE,IAAI,CAAA;IAGV,QAAQ,EAAE,QAAQ,CAAA;IAClB,MAAM,EAAE,MAAM,CAAA;IACd,MAAM,EAAE,MAAM,CAAA;IACd,KAAK,EAAE,KAAK,CAAA;IACZ,OAAO,EAAE,OAAO,CAAA;gBAEJ,MAAM,EAAE,iBAAiB;IA6B/B,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,CAAC,CAAC;CA0B3E;AAED,qBAAa,gBAAiB,SAAQ,KAAK;IACzC,IAAI,EAAE,MAAM,CAAA;IACZ,UAAU,EAAE,MAAM,CAAA;gBAEN,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM;CAM9D;AAED,eAAe,WAAW,CAAA"}

package/dist/index.js

@@ -0,0 +1,201 @@
+"use strict";
+/**
+ * PromptGuard Node.js SDK
+ *
+ * Drop-in security for AI applications.
+ * Just change your base URL and add an API key.
+ *
+ * @example
+ * import { PromptGuard } from 'promptguard';
+ *
+ * const pg = new PromptGuard({ apiKey: 'pg_xxx' });
+ *
+ * const response = await pg.chat.completions.create({
+ *   model: 'gpt-4',
+ *   messages: [{ role: 'user', content: 'Hello!' }]
+ * });
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PromptGuardError = exports.PromptGuard = void 0;
+class ChatCompletions {
+    constructor(client) {
+        this.client = client;
+    }
+    async create(params) {
+        const response = await this.client.request("POST", "/chat/completions", params);
+        return response;
+    }
+}
+class Chat {
+    constructor(client) {
+        this.completions = new ChatCompletions(client);
+    }
+}
+class Security {
+    constructor(client) {
+        this.client = client;
+    }
+    async scan(content, type = "prompt") {
+        return this.client.request("POST", "/security/scan", {
+            content,
+            type,
+        });
+    }
+    async redact(content, piiTypes) {
+        return this.client.request("POST", "/security/redact", {
+            content,
+            pii_types: piiTypes,
+        });
+    }
+}
+class Memory {
+    constructor(client) {
+        this.client = client;
+    }
+    async store(agentId, content, options) {
+        return this.client.request("POST", "/memory/store", {
+            agent_id: agentId,
+            content,
+            session_id: options?.sessionId,
+            metadata: options?.metadata,
+            importance: options?.importance ?? 0.5,
+        });
+    }
+    async retrieve(agentId, query, options) {
+        return this.client.request("POST", "/memory/retrieve", {
+            agent_id: agentId,
+            query,
+            top_k: options?.topK ?? 5,
+            min_relevance: options?.minRelevance ?? 0.5,
+            session_id: options?.sessionId,
+        });
+    }
+    async clear(agentId, sessionId) {
+        const url = sessionId ? `/memory/${agentId}?session_id=${sessionId}` : `/memory/${agentId}`;
+        return this.client.request("DELETE", url);
+    }
+    async stats() {
+        return this.client.request("GET", "/memory/stats");
+    }
+}
+class Scrape {
+    constructor(client) {
+        this.client = client;
+    }
+    async url(url, options) {
+        return this.client.request("POST", "/scrape", {
+            url,
+            render_js: options?.renderJs ?? false,
+            extract_text: options?.extractText ?? true,
+            timeout: options?.timeout ?? 30,
+        });
+    }
+    async batch(urls, options) {
+        return this.client.request("POST", "/scrape/batch", {
+            urls,
+            ...options,
+        });
+    }
+}
+class Agent {
+    constructor(client) {
+        this.client = client;
+    }
+    async validateTool(agentId, toolName, args, sessionId) {
+        return this.client.request("POST", "/agent/validate-tool", {
+            agent_id: agentId,
+            tool_name: toolName,
+            arguments: args,
+            session_id: sessionId,
+        });
+    }
+    async analyzeBehavior(agentId, recentActions, sessionId) {
+        return this.client.request("POST", "/agent/analyze-behavior", {
+            agent_id: agentId,
+            recent_actions: recentActions,
+            session_id: sessionId,
+        });
+    }
+    async stats(agentId) {
+        return this.client.request("GET", `/agent/${agentId}/stats`);
+    }
+}
+class RedTeam {
+    constructor(client) {
+        this.base = "/internal/redteam";
+        this.client = client;
+    }
+    async listTests() {
+        return this.client.request("GET", `${this.base}/tests`);
+    }
+    async runTest(testName, targetPreset = "default") {
+        return this.client.request("POST", `${this.base}/test/${testName}`, {
+            target_preset: targetPreset,
+        });
+    }
+    async runAll(targetPreset = "default") {
+        return this.client.request("POST", `${this.base}/test-all`, {
+            target_preset: targetPreset,
+        });
+    }
+    async runCustom(prompt, targetPreset = "default") {
+        return this.client.request("POST", `${this.base}/test-custom`, {
+            custom_prompt: prompt,
+            target_preset: targetPreset,
+        });
+    }
+}
+class PromptGuard {
+    constructor(config) {
+        const apiKey = config.apiKey || process.env.PROMPTGUARD_API_KEY;
+        if (!apiKey) {
+            throw new Error("API key required. Pass apiKey in config or set PROMPTGUARD_API_KEY environment variable.");
+        }
+        this.config = {
+            apiKey,
+            baseUrl: config.baseUrl || process.env.PROMPTGUARD_BASE_URL || "https://api.promptguard.co/api/v1",
+            timeout: config.timeout || 30000,
+            enableCaching: config.enableCaching ?? true,
+            enableSecurityScan: config.enableSecurityScan ?? true,
+        };
+        // OpenAI-compatible
+        this.chat = new Chat(this);
+        // PromptGuard-specific
+        this.security = new Security(this);
+        this.memory = new Memory(this);
+        this.scrape = new Scrape(this);
+        this.agent = new Agent(this);
+        this.redteam = new RedTeam(this);
+    }
+    async request(method, path, body) {
+        const url = `${this.config.baseUrl}${path}`;
+        const response = await fetch(url, {
+            method,
+            headers: {
+                Authorization: `Bearer ${this.config.apiKey}`,
+                "Content-Type": "application/json",
+                "X-PromptGuard-SDK": "node",
+                "X-PromptGuard-Version": "0.1.0",
+            },
+            body: body ? JSON.stringify(body) : undefined,
+            signal: AbortSignal.timeout(this.config.timeout),
+        });
+        if (!response.ok) {
+            const errorBody = await response.json().catch(() => ({}));
+            throw new PromptGuardError(errorBody.error?.message || "Request failed", errorBody.error?.code || "UNKNOWN", response.status);
+        }
+        return response.json();
+    }
+}
+exports.PromptGuard = PromptGuard;
+class PromptGuardError extends Error {
+    constructor(message, code, statusCode) {
+        super(`${code}: ${message}`);
+        this.name = "PromptGuardError";
+        this.code = code;
+        this.statusCode = statusCode;
+    }
+}
+exports.PromptGuardError = PromptGuardError;
+exports.default = PromptGuard;
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;GAeG;;;AAoEH,MAAM,eAAe;IAGnB,YAAY,MAAmB;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;IACtB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAA6B;QACxC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CACxC,MAAM,EACN,mBAAmB,EACnB,MAAM,CACP,CAAA;QACD,OAAO,QAAQ,CAAA;IACjB,CAAC;CACF;AAED,MAAM,IAAI;IAGR,YAAY,MAAmB;QAC7B,IAAI,CAAC,WAAW,GAAG,IAAI,eAAe,CAAC,MAAM,CAAC,CAAA;IAChD,CAAC;CACF;AAED,MAAM,QAAQ;IAGZ,YAAY,MAAmB;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;IACtB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAe,EAAE,OAA8B,QAAQ;QAChE,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAqB,MAAM,EAAE,gBAAgB,EAAE;YACvE,OAAO;YACP,IAAI;SACL,CAAC,CAAA;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAAe,EAAE,QAAmB;QAC/C,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAe,MAAM,EAAE,kBAAkB,EAAE;YACnE,OAAO;YACP,SAAS,EAAE,QAAQ;SACpB,CAAC,CAAA;IACJ,CAAC;CACF;AAED,MAAM,MAAM;IAGV,YAAY,MAAmB;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;IACtB,CAAC;IAED,KAAK,CAAC,KAAK,CACT,OAAe,EACf,OAAe,EACf,OAIC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAc,MAAM,EAAE,eAAe,EAAE;YAC/D,QAAQ,EAAE,OAAO;YACjB,OAAO;YACP,UAAU,EAAE,OAAO,EAAE,SAAS;YAC9B,QAAQ,EAAE,OAAO,EAAE,QAAQ;YAC3B,UAAU,EAAE,OAAO,EAAE,UAAU,IAAI,GAAG;SACvC,CAAC,CAAA;IACJ,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,OAAe,EACf,KAAa,EACb,OAIC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAuB,MAAM,EAAE,kBAAkB,EAAE;YAC3E,QAAQ,EAAE,OAAO;YACjB,KAAK;YACL,KAAK,EAAE,OAAO,EAAE,IAAI,IAAI,CAAC;YACzB,aAAa,EAAE,OAAO,EAAE,YAAY,IAAI,GAAG;YAC3C,UAAU,EAAE,OAAO,EAAE,SAAS;SAC/B,CAAC,CAAA;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,OAAe,EAAE,SAAkB;QAC7C,MAAM,GAAG,GAAG,SAAS,CAAC,CAAC,CAAC,WAAW,OAAO,eAAe,SAAS,EAAE,CAAC,CAAC,CAAC,WAAW,OAAO,EAAE,CAAA;QAC3F,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAA;IAC3C,CAAC;IAED,KAAK,CAAC,KAAK;QACT,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,eAAe,CAAC,CAAA;IACpD,CAAC;CACF;AAUD,MAAM,MAAM;IAGV,YAAY,MAAmB;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;IACtB,CAAC;IAED,KAAK,CAAC,GAAG,CACP,GAAW,EACX,OAIC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAe,MAAM,EAAE,SAAS,EAAE;YAC1D,GAAG;YACH,SAAS,EAAE,OAAO,EAAE,QAAQ,IAAI,KAAK;YACrC,YAAY,EAAE,OAAO,EAAE,WAAW,IAAI,IAAI;YAC1C,OAAO,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE;SAChC,CAAC,CAAA;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAc,EAAE,OAAiC;QAC3D,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,eAAe,EAAE;YAClD,IAAI;YACJ,GAAG,OAAO;SACX,CAAC,CAAA;IACJ,CAAC;CACF;AAkBD,MAAM,KAAK;IAGT,YAAY,MAAmB;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;IACtB,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,OAAe,EACf,QAAgB,EAChB,IAA6B,EAC7B,SAAkB;QAElB,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAuB,MAAM,EAAE,sBAAsB,EAAE;YAC/E,QAAQ,EAAE,OAAO;YACjB,SAAS,EAAE,QAAQ;YACnB,SAAS,EAAE,IAAI;YACf,UAAU,EAAE,SAAS;SACtB,CAAC,CAAA;IACJ,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,OAAe,EACf,aAA+E,EAC/E,SAAkB;QAElB,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAyB,MAAM,EAAE,yBAAyB,EAAE;YACpF,QAAQ,EAAE,OAAO;YACjB,cAAc,EAAE,aAAa;YAC7B,UAAU,EAAE,SAAS;SACtB,CAAC,CAAA;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,OAAe;QACzB,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,UAAU,OAAO,QAAQ,CAAC,CAAA;IAC9D,CAAC;CACF;AAqBD,MAAM,OAAO;IAIX,YAAY,MAAmB;QAFvB,SAAI,GAAG,mBAAmB,CAAA;QAGhC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;IACtB,CAAC;IAED,KAAK,CAAC,SAAS;QACb,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,IAAI,QAAQ,CAAC,CAAA;IACzD,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,QAAgB,EAAE,YAAY,GAAG,SAAS;QACtD,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAoB,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,SAAS,QAAQ,EAAE,EAAE;YACrF,aAAa,EAAE,YAAY;SAC5B,CAAC,CAAA;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,YAAY,GAAG,SAAS;QACnC,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAiB,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,WAAW,EAAE;YAC1E,aAAa,EAAE,YAAY;SAC5B,CAAC,CAAA;IACJ,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAc,EAAE,YAAY,GAAG,SAAS;QACtD,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAoB,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,cAAc,EAAE;YAChF,aAAa,EAAE,MAAM;YACrB,aAAa,EAAE,YAAY;SAC5B,CAAC,CAAA;IACJ,CAAC;CACF;AAED,MAAa,WAAW;IAatB,YAAY,MAAyB;QACnC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAA;QAE/D,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CACb,0FAA0F,CAC3F,CAAA;QACH,CAAC;QAED,IAAI,CAAC,MAAM,GAAG;YACZ,MAAM;YACN,OAAO,EACL,MAAM,CAAC,OAAO,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,mCAAmC;YAC3F,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,KAAK;YAChC,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,IAAI;YAC3C,kBAAkB,EAAE,MAAM,CAAC,kBAAkB,IAAI,IAAI;SACtD,CAAA;QAED,oBAAoB;QACpB,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,CAAA;QAE1B,uBAAuB;QACvB,IAAI,CAAC,QAAQ,GAAG,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAA;QAClC,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,CAAA;QAC9B,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,CAAA;QAC9B,IAAI,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC,IAAI,CAAC,CAAA;QAC5B,IAAI,CAAC,OAAO,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,CAAA;IAClC,CAAC;IAED,KAAK,CAAC,OAAO,CAAI,MAAc,EAAE,IAAY,EAAE,IAAc;QAC3D,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,IAAI,EAAE,CAAA;QAE3C,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,MAAM;YACN,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;gBAC7C,cAAc,EAAE,kBAAkB;gBAClC,mBAAmB,EAAE,MAAM;gBAC3B,uBAAuB,EAAE,OAAO;aACjC;YACD,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;YAC7C,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;SACjD,CAAC,CAAA;QAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAoD,CAAA;YAC5G,MAAM,IAAI,gBAAgB,CACxB,SAAS,CAAC,KAAK,EAAE,OAAO,IAAI,gBAAgB,EAC5C,SAAS,CAAC,KAAK,EAAE,IAAI,IAAI,SAAS,EAClC,QAAQ,CAAC,MAAM,CAChB,CAAA;QACH,CAAC;QAED,OAAO,QAAQ,CAAC,IAAI,EAAgB,CAAA;IACtC,CAAC;CACF;AApED,kCAoEC;AAED,MAAa,gBAAiB,SAAQ,KAAK;IAIzC,YAAY,OAAe,EAAE,IAAY,EAAE,UAAkB;QAC3D,KAAK,CAAC,GAAG,IAAI,KAAK,OAAO,EAAE,CAAC,CAAA;QAC5B,IAAI,CAAC,IAAI,GAAG,kBAAkB,CAAA;QAC9B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;QAChB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAA;IAC9B,CAAC;CACF;AAVD,4CAUC;AAED,kBAAe,WAAW,CAAA"}

package/package.json

@@ -0,0 +1,47 @@
+{
+  "name": "promptguard-sdk",
+  "version": "0.1.0",
+  "description": "Drop-in security for AI applications - AI Firewall SDK",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "scripts": {
+    "build": "tsc",
+    "test": "jest",
+    "lint": "eslint src/",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "ai",
+    "security",
+    "llm",
+    "prompt-injection",
+    "openai",
+    "anthropic",
+    "ai-firewall",
+    "pii-detection",
+    "llm-security"
+  ],
+  "author": "PromptGuard <support@promptguard.co>",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/acebot712/promptguard.git",
+    "directory": "packages/sdk-node"
+  },
+  "homepage": "https://promptguard.co",
+  "bugs": {
+    "url": "https://github.com/acebot712/promptguard/issues"
+  },
+  "engines": {
+    "node": ">=16.0.0"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "files": ["dist", "README.md", "LICENSE"],
+  "dependencies": {},
+  "devDependencies": {
+    "@types/node": "^20.0.0",
+    "typescript": "^5.0.0"
+  }
+}