promptguard-sdk 1.2.0 → 1.5.1

package/dist/integrations/langchain.js

@@ -0,0 +1,212 @@
+"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");
+// ---------------------------------------------------------------------------
+// Handler
+// ---------------------------------------------------------------------------
+class PromptGuardCallbackHandler {
+    constructor(options) {
+        this.name = "promptguard";
+        /** Chain context per run ID for richer threat detection. */
+        this.chainContext = new Map();
+        const apiKey = options.apiKey ?? process.env.PROMPTGUARD_API_KEY ?? "";
+        if (!apiKey) {
+            throw new Error("PromptGuard API key required. Pass apiKey or set PROMPTGUARD_API_KEY.");
+        }
+        this.guard = new guard_1.GuardClient({
+            apiKey,
+            baseUrl: options.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);
+    }
+    // -- Retriever (no-op) ---------------------------------------------------
+    handleRetrieverStart(..._args) {
+        /* no-op */
+    }
+    handleRetrieverEnd(..._args) {
+        /* no-op */
+    }
+    handleRetrieverError(..._args) {
+        /* no-op */
+    }
+    handleText(..._args) {
+        /* no-op */
+    }
+    // -- 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

package/dist/integrations/langchain.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"langchain.js","sourceRoot":"","sources":["../../src/integrations/langchain.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;;GAiBG;;;AAEH,oCAOiB;AAYjB,8EAA8E;AAC9E,UAAU;AACV,8EAA8E;AAE9E,MAAa,0BAA0B;IAWrC,YAAY,OAAmC;QAVtC,SAAI,GAAG,aAAa,CAAA;QAO7B,4DAA4D;QACpD,iBAAY,GAAG,IAAI,GAAG,EAAmC,CAAA;QAG/D,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,EAAE,CAAA;QACtE,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,uEAAuE,CAAC,CAAA;QAC1F,CAAC;QAED,IAAI,CAAC,KAAK,GAAG,IAAI,mBAAW,CAAC;YAC3B,MAAM;YACN,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,OAAO,EAAE,OAAO,CAAC,OAAO;SACzB,CAAC,CAAA;QAEF,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,SAAS,CAAA;QACrC,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,IAAI,CAAA;QAClD,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAA;IAC1C,CAAC;IAED,2EAA2E;IAE3E,KAAK,CAAC,cAAc,CAClB,UAAmC,EACnC,OAAiB,EACjB,KAAa,EACb,WAAoB,EACpB,YAAsC,EACtC,IAAe,EACf,QAAkC;QAElC,MAAM,QAAQ,GAAmB,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACnD,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,CAAC;SACX,CAAC,CAAC,CAAA;QACH,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,CAAA;QAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAA;QAE5E,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,CAAA;QACvE,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAA;IACtC,CAAC;IAED,KAAK,CAAC,oBAAoB,CACxB,UAAmC,EACnC,QAAqB,EACrB,KAAa,EACb,WAAoB,EACpB,YAAsC,EACtC,IAAe,EACf,QAAkC;QAElC,MAAM,aAAa,GAAmB,EAAE,CAAA;QAExC,KAAK,MAAM,WAAW,IAAI,QAAQ,EAAE,CAAC;YACnC,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;gBAC9B,MAAM,GAAG,GAAG,GAA8B,CAAA;gBAC1C,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,IAAI,GAAG,EAAE,IAAI,IAAI,MAAM,CAAC,CAAC,CAAA;gBACnE,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,EAAE,OAAO,IAAI,GAAG,EAAE,IAAI,IAAI,EAAE,CAAC,CAAA;gBACvD,aAAa,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;YACvC,CAAC;QACH,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,CAAA;QAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAA;QAEnF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,CAAA;QAC5E,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAA;IACtC,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAe,EAAE,KAAa;QAC/C,IAAI,CAAC,IAAI,CAAC,aAAa;YAAE,OAAM;QAE/B,MAAM,IAAI,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAA;QAC5C,IAAI,CAAC,IAAI;YAAE,OAAM;QAEjB,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,SAAS,EAAE,cAAc,CAAC,CAAA;QACnE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAClC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EACtC,QAAQ,EACR,SAAS,EACT,OAAO,CACR,CAAA;QACD,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAA;IACtC,CAAC;IAED,cAAc,CAAC,MAAa,EAAE,KAAa;QACzC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IACjC,CAAC;IAED,2EAA2E;IAE3E,KAAK,CAAC,gBAAgB,CACpB,UAAmC,EACnC,OAAgC,EAChC,KAAa,EACb,WAAoB,EACpB,IAAe;QAEf,MAAM,SAAS,GAAI,UAAU,EAAE,EAA2B,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,SAAS,CAAA;QACrF,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE;YAC3B,UAAU,EAAE,SAAS;YACrB,aAAa,EAAE,WAAW;YAC1B,IAAI;SACL,CAAC,CAAA;IACJ,CAAC;IAED,cAAc,CAAC,QAAiC,EAAE,KAAa;QAC7D,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IACjC,CAAC;IAED,gBAAgB,CAAC,MAAa,EAAE,KAAa;QAC3C,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IACjC,CAAC;IAED,2EAA2E;IAE3E,KAAK,CAAC,eAAe,CACnB,UAAmC,EACnC,KAAa,EACb,KAAa,EACb,WAAoB,EACpB,IAAe,EACf,QAAkC;QAElC,MAAM,QAAQ,GACX,UAAU,EAAE,IAAe;YAC3B,UAAU,EAAE,EAA2B,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACtD,SAAS,CAAA;QAEX,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,IAAI,EAAE;YAClE,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC;YACnB,SAAS,EAAE,QAAQ;SACpB,CAAC,CAAA;QAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAClC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAClC,OAAO,EACP,MAAM,EACN,OAAO,CACR,CAAA;QACD,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAA;IACtC,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAe,EAAE,KAAa;QAChD,IAAI,CAAC,IAAI,CAAC,aAAa;YAAE,OAAM;QAC/B,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAA;QACzC,IAAI,CAAC,IAAI;YAAE,OAAM;QAEjB,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,SAAS,EAAE,eAAe,CAAC,CAAA;QACpE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAClC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EACtC,QAAQ,EACR,SAAS,EACT,OAAO,CACR,CAAA;QACD,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAA;IACtC,CAAC;IAED,eAAe,CAAC,MAAa,EAAE,KAAa;QAC1C,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IACjC,CAAC;IAED,2EAA2E;IAE3E,oBAAoB,CAAC,GAAG,KAAgB;QACtC,WAAW;IACb,CAAC;IACD,kBAAkB,CAAC,GAAG,KAAgB;QACpC,WAAW;IACb,CAAC;IACD,oBAAoB,CAAC,GAAG,KAAgB;QACtC,WAAW;IACb,CAAC;IACD,UAAU,CAAC,GAAG,KAAgB;QAC5B,WAAW;IACb,CAAC;IAED,2EAA2E;IAEnE,KAAK,CAAC,QAAQ,CACpB,QAAwB,EACxB,SAA6B,EAC7B,KAAc,EACd,OAAsB;QAEtB,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,CAAC,CAAA;QACnE,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,CAAC,IAAI,CAAC,QAAQ;gBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAA;YAC5D,OAAO,IAAI,CAAA;QACb,CAAC;IACH,CAAC;IAEO,cAAc,CAAC,QAA8B,EAAE,KAAa;QAClE,IAAI,CAAC,QAAQ;YAAE,OAAM;QAErB,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACrB,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS;gBAAE,MAAM,IAAI,+BAAuB,CAAC,QAAQ,CAAC,CAAA;YACxE,OAAO,CAAC,IAAI,CACV,uCAAuC,QAAQ,CAAC,UAAU,WAAW,QAAQ,CAAC,OAAO,SAAS,KAAK,GAAG,CACvG,CAAA;QACH,CAAC;QAED,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACtB,OAAO,CAAC,IAAI,CAAC,yCAAyC,QAAQ,CAAC,OAAO,SAAS,KAAK,GAAG,CAAC,CAAA;QAC1F,CAAC;IACH,CAAC;IAEO,YAAY,CAClB,KAAa,EACb,WAAoB,EACpB,SAAS,GAAG,SAAS,EACrB,IAAe,EACf,QAAkC;QAElC,MAAM,SAAS,GACb,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC;YAC5B,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAC9D,EAAE,CAAA;QAEJ,OAAO;YACL,SAAS,EAAE,WAAW;YACtB,UAAU,EAAE,SAAS,CAAC,UAAgC;YACtD,UAAU,EAAE,KAAK;YACjB,QAAQ,EAAE;gBACR,SAAS;gBACT,IAAI,EAAE,IAAI,IAAK,SAAS,CAAC,IAA6B;gBACtD,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC;aACpB;SACF,CAAA;IACH,CAAC;IAEO,YAAY,CAAC,UAAmC;QACtD,MAAM,MAAM,GAAG,UAAU,EAAE,MAA6C,CAAA;QACxE,OAAO,CACJ,MAAM,EAAE,UAAqB;YAC7B,MAAM,EAAE,KAAgB;YACzB,CAAE,UAAU,EAAE,EAAe,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACjD,SAAS,CACV,CAAA;IACH,CAAC;IAEO,OAAO,CAAC,IAAY;QAC1B,MAAM,GAAG,GAA2B;YAClC,KAAK,EAAE,MAAM;YACb,EAAE,EAAE,WAAW;YACf,MAAM,EAAE,QAAQ;SACjB,CAAA;QACD,OAAO,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,CAAA;IAC1B,CAAC;IAEO,kBAAkB,CAAC,MAAe;QACxC,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAiC,CAAA;YAC/C,IAAI,KAAK,EAAE,WAAW,EAAE,CAAC;gBACvB,MAAM,KAAK,GAAa,EAAE,CAAA;gBAC1B,KAAK,MAAM,OAAO,IAAI,KAAK,CAAC,WAAwB,EAAE,CAAC;oBACrD,KAAK,MAAM,GAAG,IAAI,OAAoC,EAAE,CAAC;wBACvD,IAAI,GAAG,EAAE,IAAI;4BAAE,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,IAAc,CAAC,CAAA;6BACxC,IAAK,GAAG,EAAE,OAAmC,EAAE,OAAO;4BACzD,KAAK,CAAC,IAAI,CAAC,MAAM,CAAE,GAAG,CAAC,OAAmC,CAAC,OAAO,CAAC,CAAC,CAAA;oBACxE,CAAC;gBACH,CAAC;gBACD,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;YAC/C,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,SAAS;QACX,CAAC;QACD,OAAO,IAAI,CAAA;IACb,CAAC;CACF;AAtRD,gEAsRC"}

package/dist/integrations/vercel-ai.d.ts

@@ -0,0 +1,49 @@
+/**
+ * Vercel AI SDK integration — middleware for `ai` package.
+ *
+ * Works with the Vercel AI SDK's `wrapLanguageModel` API to intercept
+ * all model calls with PromptGuard security scanning.
+ *
+ * @example
+ * ```ts
+ * import { openai } from '@ai-sdk/openai';
+ * import { wrapLanguageModel, generateText } from 'ai';
+ * import { promptGuardMiddleware } from 'promptguard-sdk/integrations/vercel-ai';
+ *
+ * const model = wrapLanguageModel({
+ *   model: openai('gpt-4o'),
+ *   middleware: promptGuardMiddleware({ apiKey: 'pg_xxx' }),
+ * });
+ *
+ * const { text } = await generateText({ model, prompt: 'Hello!' });
+ * ```
+ */
+import { type GuardClientConfig } from "../guard";
+export interface PromptGuardMiddlewareOptions extends GuardClientConfig {
+    mode?: "enforce" | "monitor";
+    scanResponses?: boolean;
+    failOpen?: boolean;
+}
+/**
+ * Create a Vercel AI SDK middleware that scans inputs/outputs via
+ * the PromptGuard Guard API.
+ *
+ * Returns an object conforming to the Vercel AI SDK `LanguageModelMiddleware`
+ * interface (`transformParams` + `wrapGenerate`).
+ */
+export declare function promptGuardMiddleware(options: PromptGuardMiddlewareOptions): {
+    /**
+     * transformParams — scans the input messages before they reach the model.
+     */
+    transformParams: ({ params }: {
+        params: Record<string, unknown>;
+    }) => Promise<Record<string, unknown>>;
+    /**
+     * wrapGenerate — optionally scans the model's response.
+     */
+    wrapGenerate: (({ doGenerate, params, }: {
+        doGenerate: () => Promise<unknown>;
+        params: Record<string, unknown>;
+    }) => Promise<unknown>) | undefined;
+};
+//# sourceMappingURL=vercel-ai.d.ts.map

package/dist/integrations/vercel-ai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vercel-ai.d.ts","sourceRoot":"","sources":["../../src/integrations/vercel-ai.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;GAmBG;AAEH,OAAO,EAEL,KAAK,iBAAiB,EAGvB,MAAM,UAAU,CAAA;AAMjB,MAAM,WAAW,4BAA6B,SAAQ,iBAAiB;IACrE,IAAI,CAAC,EAAE,SAAS,GAAG,SAAS,CAAA;IAC5B,aAAa,CAAC,EAAE,OAAO,CAAA;IACvB,QAAQ,CAAC,EAAE,OAAO,CAAA;CACnB;AAMD;;;;;;GAMG;AACH,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,4BAA4B;IAiBvE;;OAEG;kCACiC;QAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;KAAE;IAuCvE;;OAEG;6CAKI;QACD,UAAU,EAAE,MAAM,OAAO,CAAC,OAAO,CAAC,CAAA;QAClC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;KAChC;EAiCR"}

package/dist/integrations/vercel-ai.js

@@ -0,0 +1,170 @@
+"use strict";
+/**
+ * Vercel AI SDK integration — middleware for `ai` package.
+ *
+ * Works with the Vercel AI SDK's `wrapLanguageModel` API to intercept
+ * all model calls with PromptGuard security scanning.
+ *
+ * @example
+ * ```ts
+ * import { openai } from '@ai-sdk/openai';
+ * import { wrapLanguageModel, generateText } from 'ai';
+ * import { promptGuardMiddleware } from 'promptguard-sdk/integrations/vercel-ai';
+ *
+ * const model = wrapLanguageModel({
+ *   model: openai('gpt-4o'),
+ *   middleware: promptGuardMiddleware({ apiKey: 'pg_xxx' }),
+ * });
+ *
+ * const { text } = await generateText({ model, prompt: 'Hello!' });
+ * ```
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.promptGuardMiddleware = promptGuardMiddleware;
+const guard_1 = require("../guard");
+// ---------------------------------------------------------------------------
+// Middleware factory
+// ---------------------------------------------------------------------------
+/**
+ * Create a Vercel AI SDK middleware that scans inputs/outputs via
+ * the PromptGuard Guard API.
+ *
+ * Returns an object conforming to the Vercel AI SDK `LanguageModelMiddleware`
+ * interface (`transformParams` + `wrapGenerate`).
+ */
+function promptGuardMiddleware(options) {
+    const apiKey = options.apiKey ?? process.env.PROMPTGUARD_API_KEY ?? "";
+    if (!apiKey) {
+        throw new Error("PromptGuard API key required. Pass apiKey or set PROMPTGUARD_API_KEY.");
+    }
+    const guard = new guard_1.GuardClient({
+        apiKey,
+        baseUrl: options.baseUrl,
+        timeout: options.timeout,
+    });
+    const mode = options.mode ?? "enforce";
+    const scanResponses = options.scanResponses ?? false;
+    const failOpen = options.failOpen ?? true;
+    return {
+        /**
+         * transformParams — scans the input messages before they reach the model.
+         */
+        transformParams: async ({ params }) => {
+            const prompt = params?.prompt;
+            if (!prompt)
+                return params;
+            const guardMessages = vercelPromptToGuardMessages(prompt);
+            if (!guardMessages.length)
+                return params;
+            try {
+                const decision = await guard.scan(guardMessages, "input", params?.modelId ?? undefined, {
+                    framework: "vercel-ai-sdk",
+                });
+                if (decision.blocked) {
+                    if (mode === "enforce")
+                        throw new guard_1.PromptGuardBlockedError(decision);
+                    console.warn(`[promptguard][monitor] would block: ${decision.threatType} (event=${decision.eventId})`);
+                }
+                if (decision.redacted && decision.redactedMessages && mode === "enforce") {
+                    return {
+                        ...params,
+                        prompt: applyRedactionToPrompt(prompt, decision.redactedMessages),
+                    };
+                }
+            }
+            catch (err) {
+                if (err instanceof guard_1.PromptGuardBlockedError)
+                    throw err;
+                if (!failOpen)
+                    throw err;
+                // fail-open: continue with original params
+            }
+            return params;
+        },
+        /**
+         * wrapGenerate — optionally scans the model's response.
+         */
+        wrapGenerate: scanResponses
+            ? async ({ doGenerate, params, }) => {
+                const result = await doGenerate();
+                const typed = result;
+                const toolCalls = typed?.toolCalls;
+                const text = typed?.text ?? toolCalls?.[0]?.args ?? null;
+                if (typeof text !== "string" || !text)
+                    return result;
+                try {
+                    const respDecision = await guard.scan([{ role: "assistant", content: text }], "output", params?.modelId ?? undefined, { framework: "vercel-ai-sdk" });
+                    if (respDecision.blocked) {
+                        if (mode === "enforce") {
+                            throw new guard_1.PromptGuardBlockedError(respDecision);
+                        }
+                        console.warn(`[promptguard][monitor] would block response: ${respDecision.threatType}`);
+                    }
+                }
+                catch (err) {
+                    if (err instanceof guard_1.PromptGuardBlockedError)
+                        throw err;
+                    if (!failOpen)
+                        throw err;
+                }
+                return result;
+            }
+            : undefined,
+    };
+}
+// ---------------------------------------------------------------------------
+// Helpers
+// ---------------------------------------------------------------------------
+/**
+ * Convert the Vercel AI SDK prompt structure to GuardMessages.
+ *
+ * The Vercel AI SDK prompt can be:
+ * - A string (simple prompt)
+ * - An array of message objects with `role` and `content`
+ */
+function vercelPromptToGuardMessages(prompt) {
+    if (typeof prompt === "string") {
+        return [{ role: "user", content: prompt }];
+    }
+    if (!Array.isArray(prompt))
+        return [];
+    const result = [];
+    for (const msg of prompt) {
+        if (!msg)
+            continue;
+        const role = String(msg.role ?? "user");
+        if (typeof msg.content === "string") {
+            result.push({ role, content: msg.content });
+        }
+        else if (Array.isArray(msg.content)) {
+            const textParts = [];
+            for (const part of msg.content) {
+                if (typeof part === "string")
+                    textParts.push(part);
+                else if (part?.type === "text")
+                    textParts.push(part.text ?? "");
+            }
+            if (textParts.length) {
+                result.push({ role, content: textParts.join("\n") });
+            }
+        }
+    }
+    return result;
+}
+/**
+ * Apply redacted content back into the Vercel AI SDK prompt structure.
+ */
+function applyRedactionToPrompt(prompt, redacted) {
+    if (typeof prompt === "string" && redacted[0]) {
+        return redacted[0].content;
+    }
+    if (!Array.isArray(prompt))
+        return prompt;
+    return prompt.map((msg, i) => {
+        if (i < redacted.length && msg && typeof msg === "object") {
+            return { ...msg, content: redacted[i].content };
+        }
+        return msg;
+    });
+}
+//# sourceMappingURL=vercel-ai.js.map

package/dist/integrations/vercel-ai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vercel-ai.js","sourceRoot":"","sources":["../../src/integrations/vercel-ai.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;;;;GAmBG;;AA8BH,sDAsGC;AAlID,oCAKiB;AAYjB,8EAA8E;AAC9E,qBAAqB;AACrB,8EAA8E;AAE9E;;;;;;GAMG;AACH,SAAgB,qBAAqB,CAAC,OAAqC;IACzE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,EAAE,CAAA;IACtE,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,uEAAuE,CAAC,CAAA;IAC1F,CAAC;IAED,MAAM,KAAK,GAAG,IAAI,mBAAW,CAAC;QAC5B,MAAM;QACN,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,OAAO,EAAE,OAAO,CAAC,OAAO;KACzB,CAAC,CAAA;IAEF,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,SAAS,CAAA;IACtC,MAAM,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,KAAK,CAAA;IACpD,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAA;IAEzC,OAAO;QACL;;WAEG;QACH,eAAe,EAAE,KAAK,EAAE,EAAE,MAAM,EAAuC,EAAE,EAAE;YACzE,MAAM,MAAM,GAAG,MAAM,EAAE,MAAM,CAAA;YAC7B,IAAI,CAAC,MAAM;gBAAE,OAAO,MAAM,CAAA;YAE1B,MAAM,aAAa,GAAG,2BAA2B,CAAC,MAAM,CAAC,CAAA;YACzD,IAAI,CAAC,aAAa,CAAC,MAAM;gBAAE,OAAO,MAAM,CAAA;YAExC,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAC/B,aAAa,EACb,OAAO,EACN,MAAM,EAAE,OAAkB,IAAI,SAAS,EACxC;oBACE,SAAS,EAAE,eAAe;iBAC3B,CACF,CAAA;gBAED,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;oBACrB,IAAI,IAAI,KAAK,SAAS;wBAAE,MAAM,IAAI,+BAAuB,CAAC,QAAQ,CAAC,CAAA;oBACnE,OAAO,CAAC,IAAI,CACV,uCAAuC,QAAQ,CAAC,UAAU,WAAW,QAAQ,CAAC,OAAO,GAAG,CACzF,CAAA;gBACH,CAAC;gBAED,IAAI,QAAQ,CAAC,QAAQ,IAAI,QAAQ,CAAC,gBAAgB,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;oBACzE,OAAO;wBACL,GAAG,MAAM;wBACT,MAAM,EAAE,sBAAsB,CAAC,MAAM,EAAE,QAAQ,CAAC,gBAAgB,CAAC;qBAClE,CAAA;gBACH,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,GAAG,YAAY,+BAAuB;oBAAE,MAAM,GAAG,CAAA;gBACrD,IAAI,CAAC,QAAQ;oBAAE,MAAM,GAAG,CAAA;gBACxB,2CAA2C;YAC7C,CAAC;YAED,OAAO,MAAM,CAAA;QACf,CAAC;QAED;;WAEG;QACH,YAAY,EAAE,aAAa;YACzB,CAAC,CAAC,KAAK,EAAE,EACL,UAAU,EACV,MAAM,GAIP,EAAE,EAAE;gBACH,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAA;gBACjC,MAAM,KAAK,GAAG,MAAiC,CAAA;gBAE/C,MAAM,SAAS,GAAG,KAAK,EAAE,SAAkD,CAAA;gBAC3E,MAAM,IAAI,GAAG,KAAK,EAAE,IAAI,IAAI,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,IAAI,CAAA;gBACxD,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,CAAC,IAAI;oBAAE,OAAO,MAAM,CAAA;gBAEpD,IAAI,CAAC;oBACH,MAAM,YAAY,GAAG,MAAM,KAAK,CAAC,IAAI,CACnC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EACtC,QAAQ,EACP,MAAM,EAAE,OAAkB,IAAI,SAAS,EACxC,EAAE,SAAS,EAAE,eAAe,EAAE,CAC/B,CAAA;oBAED,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;wBACzB,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;4BACvB,MAAM,IAAI,+BAAuB,CAAC,YAAY,CAAC,CAAA;wBACjD,CAAC;wBACD,OAAO,CAAC,IAAI,CACV,gDAAgD,YAAY,CAAC,UAAU,EAAE,CAC1E,CAAA;oBACH,CAAC;gBACH,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,IAAI,GAAG,YAAY,+BAAuB;wBAAE,MAAM,GAAG,CAAA;oBACrD,IAAI,CAAC,QAAQ;wBAAE,MAAM,GAAG,CAAA;gBAC1B,CAAC;gBAED,OAAO,MAAM,CAAA;YACf,CAAC;YACH,CAAC,CAAC,SAAS;KACd,CAAA;AACH,CAAC;AAED,8EAA8E;AAC9E,UAAU;AACV,8EAA8E;AAE9E;;;;;;GAMG;AACH,SAAS,2BAA2B,CAAC,MAAe;IAClD,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAA;IAC5C,CAAC;IAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;QAAE,OAAO,EAAE,CAAA;IAErC,MAAM,MAAM,GAAmB,EAAE,CAAA;IACjC,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QACzB,IAAI,CAAC,GAAG;YAAE,SAAQ;QAClB,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,MAAM,CAAC,CAAA;QAEvC,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACpC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAA;QAC7C,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YACtC,MAAM,SAAS,GAAa,EAAE,CAAA;YAC9B,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBAC/B,IAAI,OAAO,IAAI,KAAK,QAAQ;oBAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;qBAC7C,IAAI,IAAI,EAAE,IAAI,KAAK,MAAM;oBAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAA;YACjE,CAAC;YACD,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;gBACrB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;YACtD,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAA;AACf,CAAC;AAED;;GAEG;AACH,SAAS,sBAAsB,CAAC,MAAe,EAAE,QAAwB;IACvE,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9C,OAAO,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAA;IAC5B,CAAC;IAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;QAAE,OAAO,MAAM,CAAA;IAEzC,OAAQ,MAAoB,CAAC,GAAG,CAAC,CAAC,GAAY,EAAE,CAAS,EAAE,EAAE;QAC3D,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;YAC1D,OAAO,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAA;QACjD,CAAC;QACD,OAAO,GAAG,CAAA;IACZ,CAAC,CAAC,CAAA;AACJ,CAAC"}

package/dist/patches/anthropic.d.ts

@@ -0,0 +1,13 @@
+/**
+ * Anthropic SDK patch — wraps `Messages.prototype.create` from
+ * `@anthropic-ai/sdk`.
+ *
+ * Covers direct Anthropic usage and frameworks that use it (LangChain.js
+ * ChatAnthropic, etc.).
+ */
+import { type GuardMessage } from "../guard";
+export declare function messagesToGuardFormat(messages: unknown[], system?: unknown): GuardMessage[];
+export declare function extractResponseContent(response: unknown): string | null;
+export declare function apply(): boolean;
+export declare function revert(): void;
+//# sourceMappingURL=anthropic.d.ts.map

package/dist/patches/anthropic.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/patches/anthropic.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAiB,KAAK,YAAY,EAA2B,MAAM,UAAU,CAAA;AASpF,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,MAAM,CAAC,EAAE,OAAO,GAAG,YAAY,EAAE,CA2C3F;AAED,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,OAAO,GAAG,MAAM,GAAG,IAAI,CAcvE;AAMD,wBAAgB,KAAK,IAAI,OAAO,CAiG/B;AAED,wBAAgB,MAAM,IAAI,IAAI,CAa7B"}

package/dist/patches/anthropic.js

@@ -0,0 +1,188 @@
+"use strict";
+/**
+ * Anthropic SDK patch — wraps `Messages.prototype.create` from
+ * `@anthropic-ai/sdk`.
+ *
+ * Covers direct Anthropic usage and frameworks that use it (LangChain.js
+ * ChatAnthropic, etc.).
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.messagesToGuardFormat = messagesToGuardFormat;
+exports.extractResponseContent = extractResponseContent;
+exports.apply = apply;
+exports.revert = revert;
+const guard_1 = require("../guard");
+let originalCreate = null;
+let patched = false;
+// ---------------------------------------------------------------------------
+// Message conversion
+// ---------------------------------------------------------------------------
+function messagesToGuardFormat(messages, system) {
+    const result = [];
+    // Anthropic separates the system prompt from the messages array.
+    if (system) {
+        if (typeof system === "string") {
+            result.push({ role: "system", content: system });
+        }
+        else if (Array.isArray(system)) {
+            const parts = [];
+            for (const block of system) {
+                if (typeof block === "object" && block?.type === "text") {
+                    parts.push(block.text ?? "");
+                }
+            }
+            if (parts.length)
+                result.push({ role: "system", content: parts.join("\n") });
+        }
+    }
+    if (!messages)
+        return result;
+    for (const raw of messages) {
+        if (!raw)
+            continue;
+        const msg = raw;
+        const role = String(msg.role ?? "user");
+        let content;
+        if (typeof msg.content === "string") {
+            content = msg.content;
+        }
+        else if (Array.isArray(msg.content)) {
+            const textParts = [];
+            for (const block of msg.content) {
+                if (typeof block === "string")
+                    textParts.push(block);
+                else if (block?.type === "text")
+                    textParts.push(block.text ?? "");
+            }
+            content = textParts.join("\n");
+        }
+        else {
+            content = String(msg.content ?? "");
+        }
+        result.push({ role, content });
+    }
+    return result;
+}
+function extractResponseContent(response) {
+    try {
+        const r = response;
+        if (r?.content && Array.isArray(r.content)) {
+            const parts = [];
+            for (const block of r.content) {
+                if (block?.type === "text")
+                    parts.push(block.text ?? "");
+            }
+            return parts.length ? parts.join("\n") : null;
+        }
+    }
+    catch {
+        // ignore
+    }
+    return null;
+}
+// ---------------------------------------------------------------------------
+// Apply / revert
+// ---------------------------------------------------------------------------
+function apply() {
+    if (patched)
+        return true;
+    let Messages;
+    try {
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const mod = require("@anthropic-ai/sdk/resources/messages");
+        Messages = mod.Messages;
+        if (!Messages?.prototype?.create)
+            return false;
+    }
+    catch {
+        return false;
+    }
+    originalCreate = Messages.prototype.create;
+    Messages.prototype.create = async function patchedCreate(...args) {
+        const { getGuardClient, getMode, isFailOpen, shouldScanResponses } = require("../auto");
+        const guard = getGuardClient();
+        if (!guard)
+            return originalCreate?.apply(this, args);
+        const params = (args[0] ?? {});
+        const messages = params.messages;
+        const system = params.system;
+        const model = params.model;
+        // -- Pre-call scan ---------------------------------------------------
+        if (messages) {
+            const guardMessages = messagesToGuardFormat(messages, system);
+            let decision = null;
+            try {
+                decision = await guard.scan(guardMessages, "input", model ? String(model) : undefined, {
+                    framework: "anthropic",
+                });
+            }
+            catch (err) {
+                if (err instanceof guard_1.GuardApiError && !isFailOpen())
+                    throw err;
+            }
+            if (decision?.blocked) {
+                if (getMode() === "enforce")
+                    throw new guard_1.PromptGuardBlockedError(decision);
+                console.warn(`[promptguard][monitor] would block: ${decision.threatType} (event=${decision.eventId})`);
+            }
+            if (decision?.redacted && decision.redactedMessages) {
+                if (getMode() === "enforce") {
+                    const redacted = decision.redactedMessages;
+                    const hasSystem = system != null;
+                    const offset = hasSystem ? 1 : 0;
+                    const newParams = { ...params };
+                    if (hasSystem && redacted[0]) {
+                        newParams.system = redacted[0].content;
+                    }
+                    newParams.messages = messages.map((msg, i) => {
+                        const idx = i + offset;
+                        if (idx < redacted.length) {
+                            const typedMsg = msg;
+                            return { ...typedMsg, content: redacted[idx].content };
+                        }
+                        return msg;
+                    });
+                    args[0] = newParams;
+                }
+            }
+        }
+        // -- Original call ---------------------------------------------------
+        const response = await originalCreate?.apply(this, args);
+        // -- Post-call scan --------------------------------------------------
+        if (shouldScanResponses() && response && guard) {
+            try {
+                const text = extractResponseContent(response);
+                if (text) {
+                    const respDecision = await guard.scan([{ role: "assistant", content: text }], "output", model ? String(model) : undefined);
+                    if (respDecision.blocked && getMode() === "enforce") {
+                        throw new guard_1.PromptGuardBlockedError(respDecision);
+                    }
+                }
+            }
+            catch (err) {
+                if (err instanceof guard_1.PromptGuardBlockedError)
+                    throw err;
+                if (err instanceof guard_1.GuardApiError && !isFailOpen())
+                    throw err;
+            }
+        }
+        return response;
+    };
+    patched = true;
+    return true;
+}
+function revert() {
+    if (!patched || !originalCreate)
+        return;
+    try {
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const mod = require("@anthropic-ai/sdk/resources/messages");
+        mod.Messages.prototype.create = originalCreate;
+    }
+    catch {
+        // ignore
+    }
+    originalCreate = null;
+    patched = false;
+}
+//# sourceMappingURL=anthropic.js.map