promptguard-sdk 1.2.0 → 1.5.1

package/dist/patches/anthropic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/patches/anthropic.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;AAWH,sDA2CC;AAED,wDAcC;AAMD,sBAiGC;AAED,wBAaC;AA1LD,oCAAoF;AAEpF,IAAI,cAAc,GAA6C,IAAI,CAAA;AACnE,IAAI,OAAO,GAAG,KAAK,CAAA;AAEnB,8EAA8E;AAC9E,qBAAqB;AACrB,8EAA8E;AAE9E,SAAgB,qBAAqB,CAAC,QAAmB,EAAE,MAAgB;IACzE,MAAM,MAAM,GAAmB,EAAE,CAAA;IAEjC,iEAAiE;IACjE,IAAI,MAAM,EAAE,CAAC;QACX,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC/B,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAA;QAClD,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACjC,MAAM,KAAK,GAAa,EAAE,CAAA;YAC1B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBAC3B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;oBACxD,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAA;gBAC9B,CAAC;YACH,CAAC;YACD,IAAI,KAAK,CAAC,MAAM;gBAAE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QAC9E,CAAC;IACH,CAAC;IAED,IAAI,CAAC,QAAQ;QAAE,OAAO,MAAM,CAAA;IAE5B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,IAAI,CAAC,GAAG;YAAE,SAAQ;QAClB,MAAM,GAAG,GAAG,GAA8B,CAAA;QAC1C,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,MAAM,CAAC,CAAA;QACvC,IAAI,OAAe,CAAA;QAEnB,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACpC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAA;QACvB,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YACtC,MAAM,SAAS,GAAa,EAAE,CAAA;YAC9B,KAAK,MAAM,KAAK,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBAChC,IAAI,OAAO,KAAK,KAAK,QAAQ;oBAAE,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;qBAC/C,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM;oBAAE,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAA;YACnE,CAAC;YACD,OAAO,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAChC,CAAC;aAAM,CAAC;YACN,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC,CAAA;QACrC,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;IAChC,CAAC;IAED,OAAO,MAAM,CAAA;AACf,CAAC;AAED,SAAgB,sBAAsB,CAAC,QAAiB;IACtD,IAAI,CAAC;QACH,MAAM,CAAC,GAAG,QAAmC,CAAA;QAC7C,IAAI,CAAC,EAAE,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3C,MAAM,KAAK,GAAa,EAAE,CAAA;YAC1B,KAAK,MAAM,KAAK,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;gBAC9B,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM;oBAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAA;YAC1D,CAAC;YACD,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;QAC/C,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;IACD,OAAO,IAAI,CAAA;AACb,CAAC;AAED,8EAA8E;AAC9E,iBAAiB;AACjB,8EAA8E;AAE9E,SAAgB,KAAK;IACnB,IAAI,OAAO;QAAE,OAAO,IAAI,CAAA;IAExB,IAAI,QAA4C,CAAA;IAChD,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,GAAG,GAAG,OAAO,CAAC,sCAAsC,CAAC,CAAA;QAC3D,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAA;QACvB,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,MAAM;YAAE,OAAO,KAAK,CAAA;IAChD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAA;IACd,CAAC;IAED,cAAc,GAAG,QAAQ,CAAC,SAAS,CAAC,MAA+B,CAAA;IAEnE,QAAQ,CAAC,SAAS,CAAC,MAAM,GAAG,KAAK,UAAU,aAAa,CAAgB,GAAG,IAAe;QACxF,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE,UAAU,EAAE,mBAAmB,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAA;QAEvF,MAAM,KAAK,GAAG,cAAc,EAAE,CAAA;QAC9B,IAAI,CAAC,KAAK;YAAE,OAAO,cAAc,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAA;QAEpD,MAAM,MAAM,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAA4B,CAAA;QACzD,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAiC,CAAA;QACzD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAA;QAC5B,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAA;QAE1B,uEAAuE;QACvE,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,aAAa,GAAG,qBAAqB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;YAC7D,IAAI,QAAQ,GAAG,IAAI,CAAA;YAEnB,IAAI,CAAC;gBACH,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE;oBACrF,SAAS,EAAE,WAAW;iBACvB,CAAC,CAAA;YACJ,CAAC;YAAC,OAAO,GAAY,EAAE,CAAC;gBACtB,IAAI,GAAG,YAAY,qBAAa,IAAI,CAAC,UAAU,EAAE;oBAAE,MAAM,GAAG,CAAA;YAC9D,CAAC;YAED,IAAI,QAAQ,EAAE,OAAO,EAAE,CAAC;gBACtB,IAAI,OAAO,EAAE,KAAK,SAAS;oBAAE,MAAM,IAAI,+BAAuB,CAAC,QAAQ,CAAC,CAAA;gBACxE,OAAO,CAAC,IAAI,CACV,uCAAuC,QAAQ,CAAC,UAAU,WAAW,QAAQ,CAAC,OAAO,GAAG,CACzF,CAAA;YACH,CAAC;YAED,IAAI,QAAQ,EAAE,QAAQ,IAAI,QAAQ,CAAC,gBAAgB,EAAE,CAAC;gBACpD,IAAI,OAAO,EAAE,KAAK,SAAS,EAAE,CAAC;oBAC5B,MAAM,QAAQ,GAAG,QAAQ,CAAC,gBAAgB,CAAA;oBAC1C,MAAM,SAAS,GAAG,MAAM,IAAI,IAAI,CAAA;oBAChC,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;oBAEhC,MAAM,SAAS,GAAG,EAAE,GAAG,MAAM,EAAE,CAAA;oBAC/B,IAAI,SAAS,IAAI,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;wBAC7B,SAAS,CAAC,MAAM,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAA;oBACxC,CAAC;oBACD,SAAS,CAAC,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAY,EAAE,CAAS,EAAE,EAAE;wBAC5D,MAAM,GAAG,GAAG,CAAC,GAAG,MAAM,CAAA;wBACtB,IAAI,GAAG,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC;4BAC1B,MAAM,QAAQ,GAAG,GAA8B,CAAA;4BAC/C,OAAO,EAAE,GAAG,QAAQ,EAAE,OAAO,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAA;wBACxD,CAAC;wBACD,OAAO,GAAG,CAAA;oBACZ,CAAC,CAAC,CAAA;oBACF,IAAI,CAAC,CAAC,CAAC,GAAG,SAAS,CAAA;gBACrB,CAAC;YACH,CAAC;QACH,CAAC;QAED,uEAAuE;QACvE,MAAM,QAAQ,GAAG,MAAM,cAAc,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAA;QAExD,uEAAuE;QACvE,IAAI,mBAAmB,EAAE,IAAI,QAAQ,IAAI,KAAK,EAAE,CAAC;YAC/C,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAA;gBAC7C,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM,YAAY,GAAG,MAAM,KAAK,CAAC,IAAI,CACnC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EACtC,QAAQ,EACR,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAClC,CAAA;oBACD,IAAI,YAAY,CAAC,OAAO,IAAI,OAAO,EAAE,KAAK,SAAS,EAAE,CAAC;wBACpD,MAAM,IAAI,+BAAuB,CAAC,YAAY,CAAC,CAAA;oBACjD,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,OAAO,GAAY,EAAE,CAAC;gBACtB,IAAI,GAAG,YAAY,+BAAuB;oBAAE,MAAM,GAAG,CAAA;gBACrD,IAAI,GAAG,YAAY,qBAAa,IAAI,CAAC,UAAU,EAAE;oBAAE,MAAM,GAAG,CAAA;YAC9D,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAA;IACjB,CAAC,CAAA;IAED,OAAO,GAAG,IAAI,CAAA;IACd,OAAO,IAAI,CAAA;AACb,CAAC;AAED,SAAgB,MAAM;IACpB,IAAI,CAAC,OAAO,IAAI,CAAC,cAAc;QAAE,OAAM;IAEvC,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,GAAG,GAAG,OAAO,CAAC,sCAAsC,CAAC,CAAA;QAC3D,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,GAAG,cAAc,CAAA;IAChD,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;IAED,cAAc,GAAG,IAAI,CAAA;IACrB,OAAO,GAAG,KAAK,CAAA;AACjB,CAAC"}

package/dist/patches/bedrock.d.ts

@@ -0,0 +1,14 @@
+/**
+ * AWS Bedrock Runtime patch — wraps the `send` method on
+ * `BedrockRuntimeClient` from `@aws-sdk/client-bedrock-runtime`.
+ *
+ * Unlike the other patches which target LLM-specific SDKs, this wraps
+ * the AWS SDK v3 command pattern: `client.send(new InvokeModelCommand(...))`
+ * and `client.send(new ConverseCommand(...))`.
+ *
+ * Covers all Bedrock-hosted models: Claude, Titan, Llama, Mistral, Cohere,
+ * and any model accessible via the Bedrock Runtime API.
+ */
+export declare function apply(): boolean;
+export declare function revert(): void;
+//# sourceMappingURL=bedrock.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.d.ts","sourceRoot":"","sources":["../../src/patches/bedrock.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAkJH,wBAAgB,KAAK,IAAI,OAAO,CAyF/B;AAED,wBAAgB,MAAM,IAAI,IAAI,CAe7B"}

package/dist/patches/bedrock.js

@@ -0,0 +1,246 @@
+"use strict";
+/**
+ * AWS Bedrock Runtime patch — wraps the `send` method on
+ * `BedrockRuntimeClient` from `@aws-sdk/client-bedrock-runtime`.
+ *
+ * Unlike the other patches which target LLM-specific SDKs, this wraps
+ * the AWS SDK v3 command pattern: `client.send(new InvokeModelCommand(...))`
+ * and `client.send(new ConverseCommand(...))`.
+ *
+ * Covers all Bedrock-hosted models: Claude, Titan, Llama, Mistral, Cohere,
+ * and any model accessible via the Bedrock Runtime API.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.apply = apply;
+exports.revert = revert;
+const guard_1 = require("../guard");
+let originalSend = null;
+let patched = false;
+const BEDROCK_COMMANDS = new Set(["InvokeModelCommand", "ConverseCommand", "ConverseStreamCommand"]);
+// ---------------------------------------------------------------------------
+// Message extraction (Bedrock-specific, handles multiple model formats)
+// ---------------------------------------------------------------------------
+function flattenContentBlocks(content) {
+    if (typeof content === "string")
+        return content;
+    if (Array.isArray(content)) {
+        const texts = [];
+        for (const block of content) {
+            if (typeof block === "object" && block !== null) {
+                if (block.type === "text" || "text" in block)
+                    texts.push(String(block.text ?? ""));
+            }
+        }
+        return texts.join("\n");
+    }
+    return String(content ?? "");
+}
+function extractSystem(system, result) {
+    if (!system)
+        return;
+    if (typeof system === "string") {
+        result.push({ role: "system", content: system });
+    }
+    else if (Array.isArray(system)) {
+        const texts = [];
+        for (const block of system) {
+            if (typeof block === "object" && block !== null && "text" in block) {
+                texts.push(String(block.text ?? ""));
+            }
+        }
+        if (texts.length)
+            result.push({ role: "system", content: texts.join("\n") });
+    }
+}
+function extractMessagesFromBody(raw, _modelId) {
+    let body = raw;
+    if (body instanceof Uint8Array || Buffer.isBuffer(body)) {
+        try {
+            body = JSON.parse(new TextDecoder().decode(body));
+        }
+        catch {
+            return [];
+        }
+    }
+    if (typeof body === "string") {
+        try {
+            body = JSON.parse(body);
+        }
+        catch {
+            return [{ role: "user", content: body }];
+        }
+    }
+    if (typeof body !== "object" || body === null)
+        return [];
+    const obj = body;
+    const result = [];
+    // Anthropic on Bedrock (InvokeModel)
+    if (obj.messages && Array.isArray(obj.messages)) {
+        extractSystem(obj.system, result);
+        for (const msg of obj.messages) {
+            if (typeof msg === "object" && msg !== null) {
+                result.push({
+                    role: String(msg.role ?? "user"),
+                    content: flattenContentBlocks(msg.content ?? ""),
+                });
+            }
+        }
+        return result;
+    }
+    // Converse API (capitalized keys)
+    if (obj.Messages && Array.isArray(obj.Messages)) {
+        extractSystem(obj.System ?? obj.system, result);
+        for (const msg of obj.Messages) {
+            if (typeof msg === "object" && msg !== null) {
+                result.push({
+                    role: String((msg.Role ?? msg.role ?? "user")).toLowerCase(),
+                    content: flattenContentBlocks(msg.Content ?? msg.content ?? ""),
+                });
+            }
+        }
+        return result;
+    }
+    // Amazon Titan
+    if (obj.inputText)
+        return [{ role: "user", content: String(obj.inputText) }];
+    // Llama / Mistral
+    if (obj.prompt)
+        return [{ role: "user", content: String(obj.prompt) }];
+    return [];
+}
+function extractBedrockResponseText(response, commandName) {
+    if (!response || typeof response !== "object")
+        return "";
+    const obj = response;
+    if (commandName === "ConverseCommand") {
+        const output = obj.output;
+        const message = output?.message;
+        const content = message?.content;
+        if (Array.isArray(content)) {
+            return content
+                .map((block) => {
+                if (typeof block === "object" && block !== null && "text" in block) {
+                    return String(block.text);
+                }
+                return "";
+            })
+                .join("");
+        }
+        return "";
+    }
+    const body = obj.body;
+    if (body instanceof Uint8Array || (typeof Buffer !== "undefined" && Buffer.isBuffer(body))) {
+        try {
+            const parsed = JSON.parse(new TextDecoder().decode(body));
+            if (typeof parsed.completion === "string")
+                return parsed.completion;
+            if (typeof parsed.generation === "string")
+                return parsed.generation;
+            const results = parsed.results;
+            if (Array.isArray(results) && results.length > 0) {
+                return String(results[0].outputText ?? results[0].text ?? "");
+            }
+        }
+        catch {
+            return "";
+        }
+    }
+    return "";
+}
+// ---------------------------------------------------------------------------
+// Apply / revert
+// ---------------------------------------------------------------------------
+function apply() {
+    if (patched)
+        return true;
+    let BedrockRuntimeClient;
+    try {
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const mod = require("@aws-sdk/client-bedrock-runtime");
+        BedrockRuntimeClient = mod.BedrockRuntimeClient;
+        if (!BedrockRuntimeClient?.prototype?.send)
+            return false;
+    }
+    catch {
+        return false;
+    }
+    originalSend = BedrockRuntimeClient.prototype.send;
+    BedrockRuntimeClient.prototype.send = async function patchedSend(...args) {
+        const { getGuardClient, getMode, isFailOpen, shouldScanResponses } = require("../auto");
+        const command = args[0];
+        const commandName = command?.constructor?.name ?? "";
+        if (!BEDROCK_COMMANDS.has(commandName)) {
+            return originalSend?.apply(this, args);
+        }
+        const guard = getGuardClient();
+        if (!guard)
+            return originalSend?.apply(this, args);
+        const input = (command.input ?? {});
+        const modelId = String(input.modelId ?? input.ModelId ?? "bedrock");
+        let guardMessages;
+        if (commandName === "InvokeModelCommand") {
+            guardMessages = extractMessagesFromBody(input.body, modelId);
+        }
+        else {
+            guardMessages = extractMessagesFromBody(input, modelId);
+        }
+        // -- Pre-call scan ---------------------------------------------------
+        if (guardMessages.length) {
+            let decision = null;
+            try {
+                decision = await guard.scan(guardMessages, "input", modelId, {
+                    framework: "aws-bedrock",
+                });
+            }
+            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})`);
+            }
+        }
+        // -- Original call ---------------------------------------------------
+        const response = await originalSend?.apply(this, args);
+        // -- Post-call scan --------------------------------------------------
+        if (shouldScanResponses() && response && guard) {
+            try {
+                const text = extractBedrockResponseText(response, commandName);
+                if (text) {
+                    const respDecision = await guard.scan([{ role: "assistant", content: text }], "output", modelId);
+                    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 || !originalSend)
+        return;
+    try {
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const mod = require("@aws-sdk/client-bedrock-runtime");
+        if (mod.BedrockRuntimeClient?.prototype) {
+            mod.BedrockRuntimeClient.prototype.send = originalSend;
+        }
+    }
+    catch {
+        // ignore
+    }
+    originalSend = null;
+    patched = false;
+}
+//# sourceMappingURL=bedrock.js.map

package/dist/patches/bedrock.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.js","sourceRoot":"","sources":["../../src/patches/bedrock.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;GAUG;;AAkJH,sBAyFC;AAED,wBAeC;AA1PD,oCAAoF;AAEpF,IAAI,YAAY,GAA6C,IAAI,CAAA;AACjE,IAAI,OAAO,GAAG,KAAK,CAAA;AAEnB,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,CAAC,oBAAoB,EAAE,iBAAiB,EAAE,uBAAuB,CAAC,CAAC,CAAA;AAEpG,8EAA8E;AAC9E,wEAAwE;AACxE,8EAA8E;AAE9E,SAAS,oBAAoB,CAAC,OAAgB;IAC5C,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,OAAO,CAAA;IAC/C,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,MAAM,KAAK,GAAa,EAAE,CAAA;QAC1B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;gBAChD,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,MAAM,IAAI,KAAK;oBAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,CAAA;YACpF,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACzB,CAAC;IACD,OAAO,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC,CAAA;AAC9B,CAAC;AAED,SAAS,aAAa,CAAC,MAAe,EAAE,MAAsB;IAC5D,IAAI,CAAC,MAAM;QAAE,OAAM;IACnB,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAA;IAClD,CAAC;SAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QACjC,MAAM,KAAK,GAAa,EAAE,CAAA;QAC1B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,MAAM,IAAI,KAAK,EAAE,CAAC;gBACnE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,CAAA;YACtC,CAAC;QACH,CAAC;QACD,IAAI,KAAK,CAAC,MAAM;YAAE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IAC9E,CAAC;AACH,CAAC;AAED,SAAS,uBAAuB,CAAC,GAAY,EAAE,QAAgB;IAC7D,IAAI,IAAI,GAAG,GAAG,CAAA;IACd,IAAI,IAAI,YAAY,UAAU,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACxD,IAAI,CAAC;YACH,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAA;QACnD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAA;QACX,CAAC;IACH,CAAC;IACD,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QACzB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAc,EAAE,CAAC,CAAA;QACpD,CAAC;IACH,CAAC;IACD,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI;QAAE,OAAO,EAAE,CAAA;IAExD,MAAM,GAAG,GAAG,IAA+B,CAAA;IAC3C,MAAM,MAAM,GAAmB,EAAE,CAAA;IAEjC,qCAAqC;IACrC,IAAI,GAAG,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChD,aAAa,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;QACjC,KAAK,MAAM,GAAG,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC/B,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,EAAE,CAAC;gBAC5C,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,MAAM,CAAC;oBAChC,OAAO,EAAE,oBAAoB,CAAC,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;iBACjD,CAAC,CAAA;YACJ,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAA;IACf,CAAC;IAED,kCAAkC;IAClC,IAAI,GAAG,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChD,aAAa,CAAC,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;QAC/C,KAAK,MAAM,GAAG,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC/B,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,EAAE,CAAC;gBAC5C,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,IAAI,MAAM,CAAW,CAAC,CAAC,WAAW,EAAE;oBACtE,OAAO,EAAE,oBAAoB,CAAC,GAAG,CAAC,OAAO,IAAI,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;iBAChE,CAAC,CAAA;YACJ,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAA;IACf,CAAC;IAED,eAAe;IACf,IAAI,GAAG,CAAC,SAAS;QAAE,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAA;IAE5E,kBAAkB;IAClB,IAAI,GAAG,CAAC,MAAM;QAAE,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;IAEtE,OAAO,EAAE,CAAA;AACX,CAAC;AAED,SAAS,0BAA0B,CAAC,QAAiB,EAAE,WAAmB;IACxE,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ;QAAE,OAAO,EAAE,CAAA;IACxD,MAAM,GAAG,GAAG,QAAmC,CAAA;IAE/C,IAAI,WAAW,KAAK,iBAAiB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,GAAG,CAAC,MAA6C,CAAA;QAChE,MAAM,OAAO,GAAG,MAAM,EAAE,OAA8C,CAAA;QACtE,MAAM,OAAO,GAAG,OAAO,EAAE,OAAgC,CAAA;QACzD,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,OAAO,OAAO;iBACX,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACb,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,MAAM,IAAI,KAAK,EAAE,CAAC;oBACnE,OAAO,MAAM,CAAE,KAAiC,CAAC,IAAI,CAAC,CAAA;gBACxD,CAAC;gBACD,OAAO,EAAE,CAAA;YACX,CAAC,CAAC;iBACD,IAAI,CAAC,EAAE,CAAC,CAAA;QACb,CAAC;QACD,OAAO,EAAE,CAAA;IACX,CAAC;IAED,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAA;IACrB,IAAI,IAAI,YAAY,UAAU,IAAI,CAAC,OAAO,MAAM,KAAK,WAAW,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;QAC3F,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAkB,CAAC,CAGrE,CAAA;YACD,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ;gBAAE,OAAO,MAAM,CAAC,UAAU,CAAA;YACnE,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ;gBAAE,OAAO,MAAM,CAAC,UAAU,CAAA;YACnE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAqD,CAAA;YAC5E,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACjD,OAAO,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAA;YAC/D,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAA;QACX,CAAC;IACH,CAAC;IAED,OAAO,EAAE,CAAA;AACX,CAAC;AAED,8EAA8E;AAC9E,iBAAiB;AACjB,8EAA8E;AAE9E,SAAgB,KAAK;IACnB,IAAI,OAAO;QAAE,OAAO,IAAI,CAAA;IAExB,IAAI,oBAAsD,CAAA;IAC1D,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,GAAG,GAAG,OAAO,CAAC,iCAAiC,CAAC,CAAA;QACtD,oBAAoB,GAAG,GAAG,CAAC,oBAAoB,CAAA;QAC/C,IAAI,CAAC,oBAAoB,EAAE,SAAS,EAAE,IAAI;YAAE,OAAO,KAAK,CAAA;IAC1D,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAA;IACd,CAAC;IAED,YAAY,GAAG,oBAAoB,CAAC,SAAS,CAAC,IAA2B,CAAA;IAEzE,oBAAoB,CAAC,SAAS,CAAC,IAAI,GAAG,KAAK,UAAU,WAAW,CAE9D,GAAG,IAAe;QAElB,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE,UAAU,EAAE,mBAAmB,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAA;QAEvF,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAA4B,CAAA;QAClD,MAAM,WAAW,GAAI,OAAO,EAAE,WAAiC,EAAE,IAAI,IAAI,EAAE,CAAA;QAE3E,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;YACvC,OAAO,YAAY,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAA;QACxC,CAAC;QAED,MAAM,KAAK,GAAG,cAAc,EAAE,CAAA;QAC9B,IAAI,CAAC,KAAK;YAAE,OAAO,YAAY,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAA;QAElD,MAAM,KAAK,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAA4B,CAAA;QAC9D,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,IAAI,SAAS,CAAC,CAAA;QAEnE,IAAI,aAA6B,CAAA;QACjC,IAAI,WAAW,KAAK,oBAAoB,EAAE,CAAC;YACzC,aAAa,GAAG,uBAAuB,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAA;QAC9D,CAAC;aAAM,CAAC;YACN,aAAa,GAAG,uBAAuB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAA;QACzD,CAAC;QAED,uEAAuE;QACvE,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;YACzB,IAAI,QAAQ,GAAG,IAAI,CAAA;YAEnB,IAAI,CAAC;gBACH,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,EAAE,OAAO,EAAE;oBAC3D,SAAS,EAAE,aAAa;iBACzB,CAAC,CAAA;YACJ,CAAC;YAAC,OAAO,GAAY,EAAE,CAAC;gBACtB,IAAI,GAAG,YAAY,qBAAa,IAAI,CAAC,UAAU,EAAE;oBAAE,MAAM,GAAG,CAAA;YAC9D,CAAC;YAED,IAAI,QAAQ,EAAE,OAAO,EAAE,CAAC;gBACtB,IAAI,OAAO,EAAE,KAAK,SAAS;oBAAE,MAAM,IAAI,+BAAuB,CAAC,QAAQ,CAAC,CAAA;gBACxE,OAAO,CAAC,IAAI,CACV,uCAAuC,QAAQ,CAAC,UAAU,WAAW,QAAQ,CAAC,OAAO,GAAG,CACzF,CAAA;YACH,CAAC;QACH,CAAC;QAED,uEAAuE;QACvE,MAAM,QAAQ,GAAG,MAAM,YAAY,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAA;QAEtD,uEAAuE;QACvE,IAAI,mBAAmB,EAAE,IAAI,QAAQ,IAAI,KAAK,EAAE,CAAC;YAC/C,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,0BAA0B,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAA;gBAC9D,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM,YAAY,GAAG,MAAM,KAAK,CAAC,IAAI,CACnC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EACtC,QAAQ,EACR,OAAO,CACR,CAAA;oBACD,IAAI,YAAY,CAAC,OAAO,IAAI,OAAO,EAAE,KAAK,SAAS,EAAE,CAAC;wBACpD,MAAM,IAAI,+BAAuB,CAAC,YAAY,CAAC,CAAA;oBACjD,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,OAAO,GAAY,EAAE,CAAC;gBACtB,IAAI,GAAG,YAAY,+BAAuB;oBAAE,MAAM,GAAG,CAAA;gBACrD,IAAI,GAAG,YAAY,qBAAa,IAAI,CAAC,UAAU,EAAE;oBAAE,MAAM,GAAG,CAAA;YAC9D,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAA;IACjB,CAAC,CAAA;IAED,OAAO,GAAG,IAAI,CAAA;IACd,OAAO,IAAI,CAAA;AACb,CAAC;AAED,SAAgB,MAAM;IACpB,IAAI,CAAC,OAAO,IAAI,CAAC,YAAY;QAAE,OAAM;IAErC,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,GAAG,GAAG,OAAO,CAAC,iCAAiC,CAAC,CAAA;QACtD,IAAI,GAAG,CAAC,oBAAoB,EAAE,SAAS,EAAE,CAAC;YACxC,GAAG,CAAC,oBAAoB,CAAC,SAAS,CAAC,IAAI,GAAG,YAAY,CAAA;QACxD,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;IAED,YAAY,GAAG,IAAI,CAAA;IACnB,OAAO,GAAG,KAAK,CAAA;AACjB,CAAC"}

package/dist/patches/cohere.d.ts

@@ -0,0 +1,14 @@
+/**
+ * Cohere SDK patch — wraps `Client.prototype.chat` and
+ * `ClientV2.prototype.chat` from the `cohere-ai` npm package.
+ *
+ * Covers direct Cohere usage and frameworks that use it (LangChain.js
+ * ChatCohere, Haystack, etc.).
+ *
+ * Cohere has two client generations:
+ *  - V1 (`Client`): `chat({ message, chatHistory })` returning `{ text }`
+ *  - V2 (`ClientV2`): `chat({ messages })` returning `{ message.content }`
+ */
+export declare function apply(): boolean;
+export declare function revert(): void;
+//# sourceMappingURL=cohere.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"cohere.d.ts","sourceRoot":"","sources":["../../src/patches/cohere.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAkIH,wBAAgB,KAAK,IAAI,OAAO,CA8B/B;AAED,wBAAgB,MAAM,IAAI,IAAI,CAqB7B"}

package/dist/patches/cohere.js

@@ -0,0 +1,183 @@
+"use strict";
+/**
+ * Cohere SDK patch — wraps `Client.prototype.chat` and
+ * `ClientV2.prototype.chat` from the `cohere-ai` npm package.
+ *
+ * Covers direct Cohere usage and frameworks that use it (LangChain.js
+ * ChatCohere, Haystack, etc.).
+ *
+ * Cohere has two client generations:
+ *  - V1 (`Client`): `chat({ message, chatHistory })` returning `{ text }`
+ *  - V2 (`ClientV2`): `chat({ messages })` returning `{ message.content }`
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.apply = apply;
+exports.revert = revert;
+const guard_1 = require("../guard");
+const originals = new Map();
+let patched = false;
+// ---------------------------------------------------------------------------
+// Message conversion
+// ---------------------------------------------------------------------------
+function messagesToGuardFormat(params) {
+    const result = [];
+    if (params.messages && Array.isArray(params.messages)) {
+        for (const msg of params.messages) {
+            if (!msg)
+                continue;
+            const role = String(msg.role ?? "user");
+            const content = String(msg.content ?? "");
+            result.push({ role, content });
+        }
+        return result;
+    }
+    if (params.chatHistory && Array.isArray(params.chatHistory)) {
+        for (const msg of params.chatHistory) {
+            if (!msg)
+                continue;
+            const role = String(msg.role ?? "user");
+            const content = String(msg.message ?? msg.content ?? "");
+            result.push({ role, content });
+        }
+    }
+    if (params.message) {
+        result.push({ role: "user", content: String(params.message) });
+    }
+    return result;
+}
+function extractResponseText(response) {
+    try {
+        const r = response;
+        if (typeof r?.text === "string")
+            return r.text;
+        const message = r?.message;
+        if (message?.content) {
+            const parts = message.content;
+            if (Array.isArray(parts)) {
+                const texts = [];
+                for (const p of parts) {
+                    if (p?.text)
+                        texts.push(String(p.text));
+                }
+                return texts.length ? texts.join("\n") : null;
+            }
+            return String(parts);
+        }
+    }
+    catch {
+        // ignore
+    }
+    return null;
+}
+// ---------------------------------------------------------------------------
+// Patch factory
+// ---------------------------------------------------------------------------
+function createPatchedChat(originalChat) {
+    return async function patchedChat(...args) {
+        const { getGuardClient, getMode, isFailOpen, shouldScanResponses } = require("../auto");
+        const guard = getGuardClient();
+        if (!guard)
+            return originalChat.apply(this, args);
+        const params = (args[0] ?? {});
+        const guardMessages = messagesToGuardFormat(params);
+        const model = params.model ? String(params.model) : "cohere";
+        // -- Pre-call scan ---------------------------------------------------
+        if (guardMessages.length) {
+            let decision = null;
+            try {
+                decision = await guard.scan(guardMessages, "input", model, {
+                    framework: "cohere",
+                });
+            }
+            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})`);
+            }
+        }
+        // -- Original call ---------------------------------------------------
+        const response = await originalChat.apply(this, args);
+        // -- Post-call scan --------------------------------------------------
+        if (shouldScanResponses() && response && guard) {
+            try {
+                const text = extractResponseText(response);
+                if (text) {
+                    const respDecision = await guard.scan([{ role: "assistant", content: text }], "output", model);
+                    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;
+    };
+}
+// ---------------------------------------------------------------------------
+// Apply / revert
+// ---------------------------------------------------------------------------
+function apply() {
+    if (patched)
+        return true;
+    let cohere;
+    try {
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        cohere = require("cohere-ai");
+    }
+    catch {
+        return false;
+    }
+    let patchedAny = false;
+    for (const className of ["CohereClientV2", "ClientV2", "CohereClient", "Client"]) {
+        try {
+            const cls = cohere[className];
+            if (!cls?.prototype?.chat)
+                continue;
+            const key = `${className}.chat`;
+            if (originals.has(key))
+                continue;
+            originals.set(key, cls.prototype.chat);
+            cls.prototype.chat = createPatchedChat(cls.prototype.chat);
+            patchedAny = true;
+        }
+        catch {
+            // class not available in this version
+        }
+    }
+    patched = patchedAny;
+    return patched;
+}
+function revert() {
+    if (!patched)
+        return;
+    try {
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const cohere = require("cohere-ai");
+        for (const [key, original] of originals) {
+            const className = key.split(".")[0];
+            try {
+                const cls = cohere[className];
+                if (cls?.prototype)
+                    cls.prototype.chat = original;
+            }
+            catch {
+                // ignore
+            }
+        }
+    }
+    catch {
+        // ignore
+    }
+    originals.clear();
+    patched = false;
+}
+//# sourceMappingURL=cohere.js.map

package/dist/patches/cohere.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cohere.js","sourceRoot":"","sources":["../../src/patches/cohere.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;GAUG;;AAkIH,sBA8BC;AAED,wBAqBC;AArLD,oCAAoF;AAEpF,MAAM,SAAS,GAAiD,IAAI,GAAG,EAAE,CAAA;AACzE,IAAI,OAAO,GAAG,KAAK,CAAA;AAEnB,8EAA8E;AAC9E,qBAAqB;AACrB,8EAA8E;AAE9E,SAAS,qBAAqB,CAAC,MAA+B;IAC5D,MAAM,MAAM,GAAmB,EAAE,CAAA;IAEjC,IAAI,MAAM,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;QACtD,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YAClC,IAAI,CAAC,GAAG;gBAAE,SAAQ;YAClB,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,MAAM,CAAC,CAAA;YACvC,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC,CAAA;YACzC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;QAChC,CAAC;QACD,OAAO,MAAM,CAAA;IACf,CAAC;IAED,IAAI,MAAM,CAAC,WAAW,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;QAC5D,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;YACrC,IAAI,CAAC,GAAG;gBAAE,SAAQ;YAClB,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,MAAM,CAAC,CAAA;YACvC,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,IAAI,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC,CAAA;YACxD,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;QAChC,CAAC;IACH,CAAC;IAED,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;IAChE,CAAC;IAED,OAAO,MAAM,CAAA;AACf,CAAC;AAED,SAAS,mBAAmB,CAAC,QAAiB;IAC5C,IAAI,CAAC;QACH,MAAM,CAAC,GAAG,QAAmC,CAAA;QAC7C,IAAI,OAAO,CAAC,EAAE,IAAI,KAAK,QAAQ;YAAE,OAAO,CAAC,CAAC,IAAI,CAAA;QAC9C,MAAM,OAAO,GAAG,CAAC,EAAE,OAA8C,CAAA;QACjE,IAAI,OAAO,EAAE,OAAO,EAAE,CAAC;YACrB,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAA;YAC7B,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzB,MAAM,KAAK,GAAa,EAAE,CAAA;gBAC1B,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;oBACtB,IAAI,CAAC,EAAE,IAAI;wBAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;gBACzC,CAAC;gBACD,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;YAC/C,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAA;QACtB,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;IACD,OAAO,IAAI,CAAA;AACb,CAAC;AAED,8EAA8E;AAC9E,gBAAgB;AAChB,8EAA8E;AAE9E,SAAS,iBAAiB,CACxB,YAA6C;IAE7C,OAAO,KAAK,UAAU,WAAW,CAAgB,GAAG,IAAe;QACjE,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE,UAAU,EAAE,mBAAmB,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAA;QAEvF,MAAM,KAAK,GAAG,cAAc,EAAE,CAAA;QAC9B,IAAI,CAAC,KAAK;YAAE,OAAO,YAAY,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAA;QAEjD,MAAM,MAAM,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAA4B,CAAA;QACzD,MAAM,aAAa,GAAG,qBAAqB,CAAC,MAAM,CAAC,CAAA;QACnD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAA;QAE5D,uEAAuE;QACvE,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;YACzB,IAAI,QAAQ,GAAG,IAAI,CAAA;YAEnB,IAAI,CAAC;gBACH,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,EAAE,KAAK,EAAE;oBACzD,SAAS,EAAE,QAAQ;iBACpB,CAAC,CAAA;YACJ,CAAC;YAAC,OAAO,GAAY,EAAE,CAAC;gBACtB,IAAI,GAAG,YAAY,qBAAa,IAAI,CAAC,UAAU,EAAE;oBAAE,MAAM,GAAG,CAAA;YAC9D,CAAC;YAED,IAAI,QAAQ,EAAE,OAAO,EAAE,CAAC;gBACtB,IAAI,OAAO,EAAE,KAAK,SAAS;oBAAE,MAAM,IAAI,+BAAuB,CAAC,QAAQ,CAAC,CAAA;gBACxE,OAAO,CAAC,IAAI,CACV,uCAAuC,QAAQ,CAAC,UAAU,WAAW,QAAQ,CAAC,OAAO,GAAG,CACzF,CAAA;YACH,CAAC;QACH,CAAC;QAED,uEAAuE;QACvE,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAA;QAErD,uEAAuE;QACvE,IAAI,mBAAmB,EAAE,IAAI,QAAQ,IAAI,KAAK,EAAE,CAAC;YAC/C,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAA;gBAC1C,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM,YAAY,GAAG,MAAM,KAAK,CAAC,IAAI,CACnC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EACtC,QAAQ,EACR,KAAK,CACN,CAAA;oBACD,IAAI,YAAY,CAAC,OAAO,IAAI,OAAO,EAAE,KAAK,SAAS,EAAE,CAAC;wBACpD,MAAM,IAAI,+BAAuB,CAAC,YAAY,CAAC,CAAA;oBACjD,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,OAAO,GAAY,EAAE,CAAC;gBACtB,IAAI,GAAG,YAAY,+BAAuB;oBAAE,MAAM,GAAG,CAAA;gBACrD,IAAI,GAAG,YAAY,qBAAa,IAAI,CAAC,UAAU,EAAE;oBAAE,MAAM,GAAG,CAAA;YAC9D,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAA;IACjB,CAAC,CAAA;AACH,CAAC;AAED,8EAA8E;AAC9E,iBAAiB;AACjB,8EAA8E;AAE9E,SAAgB,KAAK;IACnB,IAAI,OAAO;QAAE,OAAO,IAAI,CAAA;IAExB,IAAI,MAAyD,CAAA;IAC7D,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAAA;IAC/B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAA;IACd,CAAC;IAED,IAAI,UAAU,GAAG,KAAK,CAAA;IAEtB,KAAK,MAAM,SAAS,IAAI,CAAC,gBAAgB,EAAE,UAAU,EAAE,cAAc,EAAE,QAAQ,CAAC,EAAE,CAAC;QACjF,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,CAAC,CAAA;YAC7B,IAAI,CAAC,GAAG,EAAE,SAAS,EAAE,IAAI;gBAAE,SAAQ;YACnC,MAAM,GAAG,GAAG,GAAG,SAAS,OAAO,CAAA;YAC/B,IAAI,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC;gBAAE,SAAQ;YAEhC,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,SAAS,CAAC,IAAuC,CAAC,CAAA;YACzE,GAAG,CAAC,SAAS,CAAC,IAAI,GAAG,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,IAAuC,CAAC,CAAA;YAC7F,UAAU,GAAG,IAAI,CAAA;QACnB,CAAC;QAAC,MAAM,CAAC;YACP,sCAAsC;QACxC,CAAC;IACH,CAAC;IAED,OAAO,GAAG,UAAU,CAAA;IACpB,OAAO,OAAO,CAAA;AAChB,CAAC;AAED,SAAgB,MAAM;IACpB,IAAI,CAAC,OAAO;QAAE,OAAM;IAEpB,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,CAAsD,CAAA;QACxF,KAAK,MAAM,CAAC,GAAG,EAAE,QAAQ,CAAC,IAAI,SAAS,EAAE,CAAC;YACxC,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;YACnC,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,CAAC,CAAA;gBAC7B,IAAI,GAAG,EAAE,SAAS;oBAAE,GAAG,CAAC,SAAS,CAAC,IAAI,GAAG,QAAQ,CAAA;YACnD,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;QACH,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;IAED,SAAS,CAAC,KAAK,EAAE,CAAA;IACjB,OAAO,GAAG,KAAK,CAAA;AACjB,CAAC"}

package/dist/patches/google.d.ts

@@ -0,0 +1,13 @@
+/**
+ * Google Generative AI SDK patch — wraps
+ * `GenerativeModel.prototype.generateContent` from `@google/generative-ai`.
+ *
+ * Covers direct Google AI usage and frameworks built on top (LangChain.js
+ * ChatGoogleGenerativeAI, etc.).
+ */
+import { type GuardMessage } from "../guard";
+export declare function contentToGuardFormat(contents: unknown): GuardMessage[];
+export declare function extractResponseText(response: unknown): string | null;
+export declare function apply(): boolean;
+export declare function revert(): void;
+//# sourceMappingURL=google.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"google.d.ts","sourceRoot":"","sources":["../../src/patches/google.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAiB,KAAK,YAAY,EAA2B,MAAM,UAAU,CAAA;AASpF,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,OAAO,GAAG,YAAY,EAAE,CAwBtE;AAcD,wBAAgB,mBAAmB,CAAC,QAAQ,EAAE,OAAO,GAAG,MAAM,GAAG,IAAI,CAcpE;AAMD,wBAAgB,KAAK,IAAI,OAAO,CA4E/B;AAED,wBAAgB,MAAM,IAAI,IAAI,CAe7B"}