promptguard-sdk 1.2.0 → 1.6.0

package/dist/index.d.ts

@@ -2,156 +2,33 @@
  * PromptGuard Node.js SDK
  *
  * Drop-in security for AI applications.
- * Just change your base URL and add an API key.
  *
+ * **Option 1 - Auto-instrumentation (recommended)**
  * @example
- * import { PromptGuard } from 'promptguard';
+ * ```ts
+ * import { init } from 'promptguard-sdk';
+ * init({ apiKey: 'pg_xxx' });
+ * // All LLM calls are now secured automatically.
+ * ```
  *
+ * **Option 2 - Proxy mode**
+ * @example
+ * ```ts
+ * import { PromptGuard } from 'promptguard-sdk';
  * const pg = new PromptGuard({ apiKey: 'pg_xxx' });
+ * const response = await pg.chat.completions.create({ ... });
+ * ```
  *
- * const response = await pg.chat.completions.create({
- *   model: 'gpt-4',
- *   messages: [{ role: 'user', content: 'Hello!' }]
- * });
+ * **Option 3 - Framework integrations**
+ * @example
+ * ```ts
+ * import { PromptGuardCallbackHandler } from 'promptguard-sdk/integrations/langchain';
+ * import { promptGuardMiddleware } from 'promptguard-sdk/integrations/vercel-ai';
+ * ```
  */
-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[];
-}
-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>;
-}
-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[];
-}
-declare class Agent {
-    private client;
-    constructor(client: PromptGuard);
-    validateTool(agentId: string, toolName: string, args: Record<string, unknown>, sessionId?: string): Promise<ToolValidationResult>;
-    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;
-    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;
+export { type InitOptions, init, shutdown } from "./auto";
+export { default, PromptGuard, PromptGuardError } from "./client";
+export { GuardApiError, GuardClient, type GuardClientConfig, type GuardContext, GuardDecision, type GuardMessage, type GuardRequestBody, type GuardResponseBody, PromptGuardBlockedError, type ThreatDetail, } from "./guard";
+export { PromptGuardCallbackHandler } from "./integrations/langchain";
+export { promptGuardMiddleware } from "./integrations/vercel-ai";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +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,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,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,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,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,KAAK,EAAE,KAAK,CAAA;IACZ,OAAO,EAAE,OAAO,CAAA;gBAEJ,MAAM,EAAE,iBAAiB;IA4B/B,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,CAAC,CAAC;CA4B3E;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"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AAGH,OAAO,EAAE,KAAK,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAA;AAEzD,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAA;AAEjE,OAAO,EACL,aAAa,EACb,WAAW,EACX,KAAK,iBAAiB,EACtB,KAAK,YAAY,EACjB,aAAa,EACb,KAAK,YAAY,EACjB,KAAK,gBAAgB,EACrB,KAAK,iBAAiB,EACtB,uBAAuB,EACvB,KAAK,YAAY,GAClB,MAAM,SAAS,CAAA;AAGhB,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAA;AACrE,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAA"}

package/dist/index.js

@@ -3,161 +3,53 @@
  * PromptGuard Node.js SDK
  *
  * Drop-in security for AI applications.
- * Just change your base URL and add an API key.
  *
+ * **Option 1 - Auto-instrumentation (recommended)**
  * @example
- * import { PromptGuard } from 'promptguard';
+ * ```ts
+ * import { init } from 'promptguard-sdk';
+ * init({ apiKey: 'pg_xxx' });
+ * // All LLM calls are now secured automatically.
+ * ```
  *
+ * **Option 2 - Proxy mode**
+ * @example
+ * ```ts
+ * import { PromptGuard } from 'promptguard-sdk';
  * const pg = new PromptGuard({ apiKey: 'pg_xxx' });
+ * const response = await pg.chat.completions.create({ ... });
+ * ```
  *
- * const response = await pg.chat.completions.create({
- *   model: 'gpt-4',
- *   messages: [{ role: 'user', content: 'Hello!' }]
- * });
+ * **Option 3 - Framework integrations**
+ * @example
+ * ```ts
+ * import { PromptGuardCallbackHandler } from 'promptguard-sdk/integrations/langchain';
+ * import { promptGuardMiddleware } from 'promptguard-sdk/integrations/vercel-ai';
+ * ```
  */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 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 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 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.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": "1.2.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;
+exports.promptGuardMiddleware = exports.PromptGuardCallbackHandler = exports.PromptGuardBlockedError = exports.GuardDecision = exports.GuardClient = exports.GuardApiError = exports.PromptGuardError = exports.PromptGuard = exports.default = exports.shutdown = exports.init = void 0;
+// Auto-instrumentation
+var auto_1 = require("./auto");
+Object.defineProperty(exports, "init", { enumerable: true, get: function () { return auto_1.init; } });
+Object.defineProperty(exports, "shutdown", { enumerable: true, get: function () { return auto_1.shutdown; } });
+// Proxy client
+var client_1 = require("./client");
+Object.defineProperty(exports, "default", { enumerable: true, get: function () { return __importDefault(client_1).default; } });
+Object.defineProperty(exports, "PromptGuard", { enumerable: true, get: function () { return client_1.PromptGuard; } });
+Object.defineProperty(exports, "PromptGuardError", { enumerable: true, get: function () { return client_1.PromptGuardError; } });
+// Guard client (standalone scanning)
+var guard_1 = require("./guard");
+Object.defineProperty(exports, "GuardApiError", { enumerable: true, get: function () { return guard_1.GuardApiError; } });
+Object.defineProperty(exports, "GuardClient", { enumerable: true, get: function () { return guard_1.GuardClient; } });
+Object.defineProperty(exports, "GuardDecision", { enumerable: true, get: function () { return guard_1.GuardDecision; } });
+Object.defineProperty(exports, "PromptGuardBlockedError", { enumerable: true, get: function () { return guard_1.PromptGuardBlockedError; } });
+// Framework integrations (re-exported for convenience)
+var langchain_1 = require("./integrations/langchain");
+Object.defineProperty(exports, "PromptGuardCallbackHandler", { enumerable: true, get: function () { return langchain_1.PromptGuardCallbackHandler; } });
+var vercel_ai_1 = require("./integrations/vercel-ai");
+Object.defineProperty(exports, "promptGuardMiddleware", { enumerable: true, get: function () { return vercel_ai_1.promptGuardMiddleware; } });
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;GAeG;;;AAuDH,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;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;AAWD,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,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;IAYtB,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,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,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAEzD,CAAA;YACD,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"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;;;;;;AAEH,uBAAuB;AACvB,+BAAyD;AAA9B,4FAAA,IAAI,OAAA;AAAE,gGAAA,QAAQ,OAAA;AACzC,eAAe;AACf,mCAAiE;AAAxD,kHAAA,OAAO,OAAA;AAAE,qGAAA,WAAW,OAAA;AAAE,0GAAA,gBAAgB,OAAA;AAC/C,qCAAqC;AACrC,iCAWgB;AAVd,sGAAA,aAAa,OAAA;AACb,oGAAA,WAAW,OAAA;AAGX,sGAAA,aAAa,OAAA;AAIb,gHAAA,uBAAuB,OAAA;AAIzB,uDAAuD;AACvD,sDAAqE;AAA5D,uHAAA,0BAA0B,OAAA;AACnC,sDAAgE;AAAvD,kHAAA,qBAAqB,OAAA"}

package/dist/integrations/langchain.d.ts

@@ -0,0 +1,51 @@
+/**
+ * LangChain.js integration - PromptGuardCallbackHandler.
+ *
+ * Implements the LangChain BaseCallbackHandler interface to scan prompts
+ * before LLM calls and responses after, with rich context about chains,
+ * tools, and agent steps.
+ *
+ * @example
+ * ```ts
+ * import { PromptGuardCallbackHandler } from 'promptguard-sdk/integrations/langchain';
+ *
+ * const handler = new PromptGuardCallbackHandler({ apiKey: 'pg_xxx' });
+ *
+ * const llm = new ChatOpenAI({ callbacks: [handler] });
+ * // or
+ * await chain.invoke({ input: '...' }, { callbacks: [handler] });
+ * ```
+ */
+import { type GuardClientConfig } from "../guard";
+export interface PromptGuardCallbackOptions extends GuardClientConfig {
+    mode?: "enforce" | "monitor";
+    scanResponses?: boolean;
+    failOpen?: boolean;
+}
+export declare class PromptGuardCallbackHandler {
+    readonly name = "promptguard";
+    private readonly guard;
+    private readonly mode;
+    private readonly scanResponses;
+    private readonly failOpen;
+    /** Chain context per run ID for richer threat detection. */
+    private chainContext;
+    constructor(options: PromptGuardCallbackOptions);
+    handleLLMStart(serialized: Record<string, unknown>, prompts: string[], runId: string, parentRunId?: string, _extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>): Promise<void>;
+    handleChatModelStart(serialized: Record<string, unknown>, messages: unknown[][], runId: string, parentRunId?: string, _extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>): Promise<void>;
+    handleLLMEnd(output: unknown, runId: string): Promise<void>;
+    handleLLMError(_error: Error, runId: string): void;
+    handleChainStart(serialized: Record<string, unknown>, _inputs: Record<string, unknown>, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleChainEnd(_outputs: Record<string, unknown>, runId: string): void;
+    handleChainError(_error: Error, runId: string): void;
+    handleToolStart(serialized: Record<string, unknown>, input: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>): Promise<void>;
+    handleToolEnd(output: unknown, runId: string): Promise<void>;
+    handleToolError(_error: Error, runId: string): void;
+    private safeScan;
+    private handleDecision;
+    private buildContext;
+    private extractModel;
+    private mapRole;
+    private extractLLMResponse;
+}
+//# sourceMappingURL=langchain.d.ts.map

package/dist/integrations/langchain.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"langchain.d.ts","sourceRoot":"","sources":["../../src/integrations/langchain.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;GAiBG;AAEH,OAAO,EAEL,KAAK,iBAAiB,EAKvB,MAAM,UAAU,CAAA;AAOjB,MAAM,WAAW,0BAA2B,SAAQ,iBAAiB;IACnE,IAAI,CAAC,EAAE,SAAS,GAAG,SAAS,CAAA;IAC5B,aAAa,CAAC,EAAE,OAAO,CAAA;IACvB,QAAQ,CAAC,EAAE,OAAO,CAAA;CACnB;AAMD,qBAAa,0BAA0B;IACrC,QAAQ,CAAC,IAAI,iBAAgB;IAE7B,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAa;IACnC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAuB;IAC5C,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAS;IACvC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAElC,4DAA4D;IAC5D,OAAO,CAAC,YAAY,CAA6C;gBAErD,OAAO,EAAE,0BAA0B;IAWzC,cAAc,CAClB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACnC,OAAO,EAAE,MAAM,EAAE,EACjB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACtC,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACjC,OAAO,CAAC,IAAI,CAAC;IAYV,oBAAoB,CACxB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACnC,QAAQ,EAAE,OAAO,EAAE,EAAE,EACrB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACtC,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACjC,OAAO,CAAC,IAAI,CAAC;IAmBV,YAAY,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAgBjE,cAAc,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAM5C,gBAAgB,CACpB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACnC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAChC,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,IAAI,CAAC,EAAE,MAAM,EAAE,GACd,OAAO,CAAC,IAAI,CAAC;IAShB,cAAc,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAItE,gBAAgB,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAM9C,eAAe,CACnB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACnC,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACjC,OAAO,CAAC,IAAI,CAAC;IAoBV,aAAa,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAelE,eAAe,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;YAMrC,QAAQ;IActB,OAAO,CAAC,cAAc;IAetB,OAAO,CAAC,YAAY;IAwBpB,OAAO,CAAC,YAAY;IAUpB,OAAO,CAAC,OAAO;IASf,OAAO,CAAC,kBAAkB;CAmB3B"}

package/dist/integrations/langchain.js

@@ -0,0 +1,193 @@
+"use strict";
+/**
+ * LangChain.js integration - PromptGuardCallbackHandler.
+ *
+ * Implements the LangChain BaseCallbackHandler interface to scan prompts
+ * before LLM calls and responses after, with rich context about chains,
+ * tools, and agent steps.
+ *
+ * @example
+ * ```ts
+ * import { PromptGuardCallbackHandler } from 'promptguard-sdk/integrations/langchain';
+ *
+ * const handler = new PromptGuardCallbackHandler({ apiKey: 'pg_xxx' });
+ *
+ * const llm = new ChatOpenAI({ callbacks: [handler] });
+ * // or
+ * await chain.invoke({ input: '...' }, { callbacks: [handler] });
+ * ```
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PromptGuardCallbackHandler = void 0;
+const guard_1 = require("../guard");
+const resolve_1 = require("../resolve");
+// ---------------------------------------------------------------------------
+// Handler
+// ---------------------------------------------------------------------------
+class PromptGuardCallbackHandler {
+    constructor(options) {
+        this.name = "promptguard";
+        /** Chain context per run ID for richer threat detection. */
+        this.chainContext = new Map();
+        const { apiKey, baseUrl } = (0, resolve_1.resolveCredentials)(options.apiKey, options.baseUrl);
+        this.guard = new guard_1.GuardClient({ apiKey, baseUrl, timeout: options.timeout });
+        this.mode = options.mode ?? "enforce";
+        this.scanResponses = options.scanResponses ?? true;
+        this.failOpen = options.failOpen ?? true;
+    }
+    // -- LLM callbacks -------------------------------------------------------
+    async handleLLMStart(serialized, prompts, runId, parentRunId, _extraParams, tags, metadata) {
+        const messages = prompts.map((p) => ({
+            role: "user",
+            content: p,
+        }));
+        const model = this.extractModel(serialized);
+        const context = this.buildContext(runId, parentRunId, "llm", tags, metadata);
+        const decision = await this.safeScan(messages, "input", model, context);
+        this.handleDecision(decision, runId);
+    }
+    async handleChatModelStart(serialized, messages, runId, parentRunId, _extraParams, tags, metadata) {
+        const guardMessages = [];
+        for (const messageList of messages) {
+            for (const raw of messageList) {
+                const msg = raw;
+                const role = this.mapRole(String(msg?.type ?? msg?.role ?? "user"));
+                const content = String(msg?.content ?? msg?.text ?? "");
+                guardMessages.push({ role, content });
+            }
+        }
+        const model = this.extractModel(serialized);
+        const context = this.buildContext(runId, parentRunId, "chat_model", tags, metadata);
+        const decision = await this.safeScan(guardMessages, "input", model, context);
+        this.handleDecision(decision, runId);
+    }
+    async handleLLMEnd(output, runId) {
+        if (!this.scanResponses)
+            return;
+        const text = this.extractLLMResponse(output);
+        if (!text)
+            return;
+        const context = this.buildContext(runId, undefined, "llm_response");
+        const decision = await this.safeScan([{ role: "assistant", content: text }], "output", undefined, context);
+        this.handleDecision(decision, runId);
+    }
+    handleLLMError(_error, runId) {
+        this.chainContext.delete(runId);
+    }
+    // -- Chain callbacks -----------------------------------------------------
+    async handleChainStart(serialized, _inputs, runId, parentRunId, tags) {
+        const chainName = serialized?.id?.slice(-1)[0] ?? "unknown";
+        this.chainContext.set(runId, {
+            chain_name: chainName,
+            parent_run_id: parentRunId,
+            tags,
+        });
+    }
+    handleChainEnd(_outputs, runId) {
+        this.chainContext.delete(runId);
+    }
+    handleChainError(_error, runId) {
+        this.chainContext.delete(runId);
+    }
+    // -- Tool callbacks ------------------------------------------------------
+    async handleToolStart(serialized, input, runId, parentRunId, tags, metadata) {
+        const toolName = serialized?.name ??
+            serialized?.id?.slice(-1)[0] ??
+            "unknown";
+        const context = this.buildContext(runId, parentRunId, "tool", tags, {
+            ...(metadata ?? {}),
+            tool_name: toolName,
+        });
+        const decision = await this.safeScan([{ role: "user", content: input }], "input", "tool", context);
+        this.handleDecision(decision, runId);
+    }
+    async handleToolEnd(output, runId) {
+        if (!this.scanResponses)
+            return;
+        const text = output ? String(output) : "";
+        if (!text)
+            return;
+        const context = this.buildContext(runId, undefined, "tool_response");
+        const decision = await this.safeScan([{ role: "assistant", content: text }], "output", undefined, context);
+        this.handleDecision(decision, runId);
+    }
+    handleToolError(_error, runId) {
+        this.chainContext.delete(runId);
+    }
+    // -- Internal helpers ----------------------------------------------------
+    async safeScan(messages, direction, model, context) {
+        try {
+            return await this.guard.scan(messages, direction, model, context);
+        }
+        catch {
+            if (!this.failOpen)
+                throw new Error("Guard API unavailable");
+            return null;
+        }
+    }
+    handleDecision(decision, runId) {
+        if (!decision)
+            return;
+        if (decision.blocked) {
+            if (this.mode === "enforce")
+                throw new guard_1.PromptGuardBlockedError(decision);
+            console.warn(`[promptguard][monitor] would block: ${decision.threatType} (event=${decision.eventId}, run=${runId})`);
+        }
+        if (decision.redacted) {
+            console.info(`[promptguard] redacted content (event=${decision.eventId}, run=${runId})`);
+        }
+    }
+    buildContext(runId, parentRunId, component = "unknown", tags, metadata) {
+        const chainInfo = this.chainContext.get(runId) ??
+            (parentRunId ? this.chainContext.get(parentRunId) : undefined) ??
+            {};
+        return {
+            framework: "langchain",
+            chain_name: chainInfo.chain_name,
+            session_id: runId,
+            metadata: {
+                component,
+                tags: tags ?? chainInfo.tags,
+                ...(metadata ?? {}),
+            },
+        };
+    }
+    extractModel(serialized) {
+        const kwargs = serialized?.kwargs;
+        return (kwargs?.model_name ??
+            kwargs?.model ??
+            (serialized?.id ?? []).slice(-1)[0] ??
+            "unknown");
+    }
+    mapRole(type) {
+        const map = {
+            human: "user",
+            ai: "assistant",
+            system: "system",
+        };
+        return map[type] ?? type;
+    }
+    extractLLMResponse(output) {
+        try {
+            const typed = output;
+            if (typed?.generations) {
+                const texts = [];
+                for (const genList of typed.generations) {
+                    for (const gen of genList) {
+                        if (gen?.text)
+                            texts.push(gen.text);
+                        else if (gen?.message?.content)
+                            texts.push(String(gen.message.content));
+                    }
+                }
+                return texts.length ? texts.join("\n") : null;
+            }
+        }
+        catch {
+            // ignore
+        }
+        return null;
+    }
+}
+exports.PromptGuardCallbackHandler = PromptGuardCallbackHandler;
+//# sourceMappingURL=langchain.js.map