promptguard-sdk 1.2.0 → 1.5.1

package/dist/patches/google.js

@@ -0,0 +1,163 @@
+"use strict";
+/**
+ * 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.).
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.contentToGuardFormat = contentToGuardFormat;
+exports.extractResponseText = extractResponseText;
+exports.apply = apply;
+exports.revert = revert;
+const guard_1 = require("../guard");
+let originalGenerate = null;
+let patched = false;
+// ---------------------------------------------------------------------------
+// Message conversion
+// ---------------------------------------------------------------------------
+function contentToGuardFormat(contents) {
+    if (typeof contents === "string") {
+        return [{ role: "user", content: contents }];
+    }
+    if (!Array.isArray(contents)) {
+        return [{ role: "user", content: String(contents ?? "") }];
+    }
+    const result = [];
+    for (const item of contents) {
+        if (typeof item === "string") {
+            result.push({ role: "user", content: item });
+        }
+        else if (item?.role && item?.parts) {
+            const text = extractTextFromParts(item.parts);
+            result.push({ role: item.role ?? "user", content: text });
+        }
+        else if (item?.parts) {
+            result.push({ role: "user", content: extractTextFromParts(item.parts) });
+        }
+        else {
+            result.push({ role: "user", content: String(item) });
+        }
+    }
+    return result;
+}
+function extractTextFromParts(parts) {
+    if (typeof parts === "string")
+        return parts;
+    if (!Array.isArray(parts))
+        return String(parts ?? "");
+    const texts = [];
+    for (const part of parts) {
+        if (typeof part === "string")
+            texts.push(part);
+        else if (part?.text != null)
+            texts.push(String(part.text));
+    }
+    return texts.join("\n");
+}
+function extractResponseText(response) {
+    try {
+        const r = response;
+        if (typeof r?.text === "function")
+            return r.text();
+        if (typeof r?.text === "string")
+            return r.text;
+        const candidates = r?.candidates;
+        if (candidates?.[0]?.content) {
+            const content = candidates[0].content;
+            if (content?.parts)
+                return extractTextFromParts(content.parts);
+        }
+    }
+    catch {
+        // ignore
+    }
+    return null;
+}
+// ---------------------------------------------------------------------------
+// Apply / revert
+// ---------------------------------------------------------------------------
+function apply() {
+    if (patched)
+        return true;
+    let GenerativeModel;
+    try {
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const mod = require("@google/generative-ai");
+        GenerativeModel = mod.GenerativeModel;
+        if (!GenerativeModel?.prototype?.generateContent)
+            return false;
+    }
+    catch {
+        return false;
+    }
+    originalGenerate = GenerativeModel.prototype.generateContent;
+    GenerativeModel.prototype.generateContent = async function patchedGenerate(...args) {
+        const { getGuardClient, getMode, isFailOpen, shouldScanResponses } = require("../auto");
+        const guard = getGuardClient();
+        if (!guard)
+            return originalGenerate?.apply(this, args);
+        const contents = args[0];
+        const modelName = String(this.model ?? this.modelName ?? "gemini");
+        // -- Pre-call scan ---------------------------------------------------
+        if (contents) {
+            const guardMessages = contentToGuardFormat(contents);
+            let decision = null;
+            try {
+                decision = await guard.scan(guardMessages, "input", modelName, {
+                    framework: "google-generativeai",
+                });
+            }
+            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 originalGenerate?.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", modelName);
+                    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 || !originalGenerate)
+        return;
+    try {
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const mod = require("@google/generative-ai");
+        if (mod.GenerativeModel) {
+            mod.GenerativeModel.prototype.generateContent = originalGenerate;
+        }
+    }
+    catch {
+        // ignore
+    }
+    originalGenerate = null;
+    patched = false;
+}
+//# sourceMappingURL=google.js.map

package/dist/patches/google.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"google.js","sourceRoot":"","sources":["../../src/patches/google.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;AAWH,oDAwBC;AAcD,kDAcC;AAMD,sBA4EC;AAED,wBAeC;AAhKD,oCAAoF;AAEpF,IAAI,gBAAgB,GAA6C,IAAI,CAAA;AACrE,IAAI,OAAO,GAAG,KAAK,CAAA;AAEnB,8EAA8E;AAC9E,qBAAqB;AACrB,8EAA8E;AAE9E,SAAgB,oBAAoB,CAAC,QAAiB;IACpD,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACjC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAA;IAC9C,CAAC;IAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC7B,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC,EAAE,CAAC,CAAA;IAC5D,CAAC;IAED,MAAM,MAAM,GAAmB,EAAE,CAAA;IACjC,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;QAC5B,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAA;QAC9C,CAAC;aAAM,IAAI,IAAI,EAAE,IAAI,IAAI,IAAI,EAAE,KAAK,EAAE,CAAC;YACrC,MAAM,IAAI,GAAG,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YAC7C,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAA;QAC3D,CAAC;aAAM,IAAI,IAAI,EAAE,KAAK,EAAE,CAAC;YACvB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;QAC1E,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QACtD,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAA;AACf,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAc;IAC1C,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAA;IAC3C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAA;IAErD,MAAM,KAAK,GAAa,EAAE,CAAA;IAC1B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,OAAO,IAAI,KAAK,QAAQ;YAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;aACzC,IAAI,IAAI,EAAE,IAAI,IAAI,IAAI;YAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;IAC5D,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;AACzB,CAAC;AAED,SAAgB,mBAAmB,CAAC,QAAiB;IACnD,IAAI,CAAC;QACH,MAAM,CAAC,GAAG,QAAmC,CAAA;QAC7C,IAAI,OAAO,CAAC,EAAE,IAAI,KAAK,UAAU;YAAE,OAAQ,CAAC,CAAC,IAAqB,EAAE,CAAA;QACpE,IAAI,OAAO,CAAC,EAAE,IAAI,KAAK,QAAQ;YAAE,OAAO,CAAC,CAAC,IAAI,CAAA;QAC9C,MAAM,UAAU,GAAG,CAAC,EAAE,UAAmD,CAAA;QACzE,IAAI,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC;YAC7B,MAAM,OAAO,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,OAAkC,CAAA;YAChE,IAAI,OAAO,EAAE,KAAK;gBAAE,OAAO,oBAAoB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;QAChE,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,eAA4D,CAAA;IAChE,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,GAAG,GAAG,OAAO,CAAC,uBAAuB,CAAC,CAAA;QAC5C,eAAe,GAAG,GAAG,CAAC,eAAe,CAAA;QACrC,IAAI,CAAC,eAAe,EAAE,SAAS,EAAE,eAAe;YAAE,OAAO,KAAK,CAAA;IAChE,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAA;IACd,CAAC;IAED,gBAAgB,GAAG,eAAe,CAAC,SAAS,CAAC,eAA0C,CAAA;IAEvF,eAAe,CAAC,SAAS,CAAC,eAAe,GAAG,KAAK,UAAU,eAAe,CAExE,GAAG,IAAe;QAElB,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,gBAAgB,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAA;QAEtD,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAA;QACxB,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,SAAS,IAAI,QAAQ,CAAC,CAAA;QAElE,uEAAuE;QACvE,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,aAAa,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAA;YACpD,IAAI,QAAQ,GAAG,IAAI,CAAA;YAEnB,IAAI,CAAC;gBACH,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,EAAE,SAAS,EAAE;oBAC7D,SAAS,EAAE,qBAAqB;iBACjC,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,gBAAgB,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAA;QAE1D,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,SAAS,CACV,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,gBAAgB;QAAE,OAAM;IAEzC,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,GAAG,GAAG,OAAO,CAAC,uBAAuB,CAAC,CAAA;QAC5C,IAAI,GAAG,CAAC,eAAe,EAAE,CAAC;YACxB,GAAG,CAAC,eAAe,CAAC,SAAS,CAAC,eAAe,GAAG,gBAAgB,CAAA;QAClE,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;IAED,gBAAgB,GAAG,IAAI,CAAA;IACvB,OAAO,GAAG,KAAK,CAAA;AACjB,CAAC"}

package/dist/patches/openai.d.ts

@@ -0,0 +1,14 @@
+/**
+ * OpenAI SDK patch — wraps the `create` method on the Chat Completions
+ * prototype (sync and async).
+ *
+ * This single patch covers every framework built on the `openai` npm
+ * package: LangChain.js (ChatOpenAI), Vercel AI SDK (openai provider),
+ * AutoGen.js, CrewAI.js, and direct usage.
+ */
+import { type GuardMessage } from "../guard";
+export declare function messagesToGuardFormat(messages: unknown[]): GuardMessage[];
+export declare function extractResponseContent(response: unknown): string | null;
+export declare function apply(): boolean;
+export declare function revert(): void;
+//# sourceMappingURL=openai.d.ts.map

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

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

package/dist/patches/openai.js

@@ -0,0 +1,165 @@
+"use strict";
+/**
+ * OpenAI SDK patch — wraps the `create` method on the Chat Completions
+ * prototype (sync and async).
+ *
+ * This single patch covers every framework built on the `openai` npm
+ * package: LangChain.js (ChatOpenAI), Vercel AI SDK (openai provider),
+ * AutoGen.js, CrewAI.js, and direct usage.
+ */
+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) {
+    if (!messages)
+        return [];
+    const 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 part of msg.content) {
+                if (typeof part === "string")
+                    textParts.push(part);
+                else if (part?.type === "text")
+                    textParts.push(part.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?.choices?.[0]?.message?.content) {
+            return r.choices[0].message.content;
+        }
+    }
+    catch {
+        // ignore
+    }
+    return null;
+}
+function applyRedaction(messages, redacted) {
+    return messages.map((msg, i) => {
+        if (i < redacted.length && typeof msg === "object" && msg !== null) {
+            return { ...msg, content: redacted[i].content };
+        }
+        return msg;
+    });
+}
+// ---------------------------------------------------------------------------
+// Apply / revert
+// ---------------------------------------------------------------------------
+function apply() {
+    if (patched)
+        return true;
+    let Completions;
+    try {
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const mod = require("openai/resources/chat/completions");
+        Completions = mod.Completions;
+        if (!Completions?.prototype?.create)
+            return false;
+    }
+    catch {
+        return false;
+    }
+    originalCreate = Completions.prototype.create;
+    Completions.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 model = params.model;
+        // -- Pre-call scan ---------------------------------------------------
+        if (messages) {
+            const guardMessages = messagesToGuardFormat(messages);
+            let decision = null;
+            try {
+                decision = await guard.scan(guardMessages, "input", model ? String(model) : undefined, {
+                    framework: "openai",
+                });
+            }
+            catch (err) {
+                if (err instanceof guard_1.GuardApiError && !isFailOpen())
+                    throw err;
+                // fail-open: continue
+            }
+            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") {
+                    args[0] = {
+                        ...params,
+                        messages: applyRedaction(messages, decision.redactedMessages),
+                    };
+                }
+            }
+        }
+        // -- 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;
+                // fail-open
+            }
+        }
+        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("openai/resources/chat/completions");
+        mod.Completions.prototype.create = originalCreate;
+    }
+    catch {
+        // ignore
+    }
+    originalCreate = null;
+    patched = false;
+}
+//# sourceMappingURL=openai.js.map

package/dist/patches/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/patches/openai.ts"],"names":[],"mappings":";AAAA;;;;;;;GAOG;;AAWH,sDA2BC;AAED,wDAUC;AAeD,sBAqFC;AAED,wBAaC;AAnKD,oCAAoF;AAEpF,IAAI,cAAc,GAA6C,IAAI,CAAA;AACnE,IAAI,OAAO,GAAG,KAAK,CAAA;AAEnB,8EAA8E;AAC9E,qBAAqB;AACrB,8EAA8E;AAE9E,SAAgB,qBAAqB,CAAC,QAAmB;IACvD,IAAI,CAAC,QAAQ;QAAE,OAAO,EAAE,CAAA;IACxB,MAAM,MAAM,GAAmB,EAAE,CAAA;IAEjC,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,IAAI,CAAC,GAAG;YAAE,SAAQ;QAClB,MAAM,GAAG,GAAG,GAA8B,CAAA;QAC1C,MAAM,IAAI,GAAW,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,MAAM,CAAC,CAAA;QAC/C,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,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,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,QAA8D,CAAA;QACxE,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;YACtC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAA;QACrC,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;IACD,OAAO,IAAI,CAAA;AACb,CAAC;AAED,SAAS,cAAc,CAAC,QAAmB,EAAE,QAAwB;IACnE,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE;QAC7B,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,EAAE,CAAC;YACnE,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;AAED,8EAA8E;AAC9E,iBAAiB;AACjB,8EAA8E;AAE9E,SAAgB,KAAK;IACnB,IAAI,OAAO;QAAE,OAAO,IAAI,CAAA;IAExB,IAAI,WAA+C,CAAA;IACnD,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,GAAG,GAAG,OAAO,CAAC,mCAAmC,CAAC,CAAA;QACxD,WAAW,GAAG,GAAG,CAAC,WAAW,CAAA;QAC7B,IAAI,CAAC,WAAW,EAAE,SAAS,EAAE,MAAM;YAAE,OAAO,KAAK,CAAA;IACnD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAA;IACd,CAAC;IAED,cAAc,GAAG,WAAW,CAAC,SAAS,CAAC,MAA+B,CAAA;IAEtE,WAAW,CAAC,SAAS,CAAC,MAAM,GAAG,KAAK,UAAU,aAAa,CAAgB,GAAG,IAAe;QAC3F,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,KAAK,GAAG,MAAM,CAAC,KAAK,CAAA;QAE1B,uEAAuE;QACvE,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,aAAa,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAA;YACrD,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,QAAQ;iBACpB,CAAC,CAAA;YACJ,CAAC;YAAC,OAAO,GAAY,EAAE,CAAC;gBACtB,IAAI,GAAG,YAAY,qBAAa,IAAI,CAAC,UAAU,EAAE;oBAAE,MAAM,GAAG,CAAA;gBAC5D,sBAAsB;YACxB,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,IAAI,CAAC,CAAC,CAAC,GAAG;wBACR,GAAG,MAAM;wBACT,QAAQ,EAAE,cAAc,CAAC,QAAQ,EAAE,QAAQ,CAAC,gBAAgB,CAAC;qBAC9D,CAAA;gBACH,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;gBAC5D,YAAY;YACd,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,mCAAmC,CAAC,CAAA;QACxD,GAAG,CAAC,WAAW,CAAC,SAAS,CAAC,MAAM,GAAG,cAAc,CAAA;IACnD,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;IAED,cAAc,GAAG,IAAI,CAAA;IACrB,OAAO,GAAG,KAAK,CAAA;AACjB,CAAC"}

package/package.json

@@ -1,13 +1,38 @@
 {
   "name": "promptguard-sdk",
-  "version": "1.2.0",
-  "description": "Drop-in security for AI applications - AI Firewall SDK",
+  "version": "1.5.1",
+  "description": "Drop-in security for AI applications - AI Firewall SDK with auto-instrumentation",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "require": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    },
+    "./auto": {
+      "require": "./dist/auto.js",
+      "types": "./dist/auto.d.ts"
+    },
+    "./guard": {
+      "require": "./dist/guard.js",
+      "types": "./dist/guard.d.ts"
+    },
+    "./integrations/langchain": {
+      "require": "./dist/integrations/langchain.js",
+      "types": "./dist/integrations/langchain.d.ts"
+    },
+    "./integrations/vercel-ai": {
+      "require": "./dist/integrations/vercel-ai.js",
+      "types": "./dist/integrations/vercel-ai.d.ts"
+    }
+  },
   "scripts": {
     "build": "tsc",
-    "test": "jest",
-    "lint": "eslint src/",
+    "test": "node --experimental-vm-modules node_modules/.bin/jest",
+    "lint": "tsc --noEmit && npx biome check .",
+    "format": "npx biome format --write .",
+    "format:check": "npx biome format .",
+    "check": "npx biome check .",
     "prepublishOnly": "npm run build"
   },
   "keywords": [
@@ -17,31 +42,78 @@
     "prompt-injection",
     "openai",
     "anthropic",
+    "google-generativeai",
     "ai-firewall",
     "pii-detection",
-    "llm-security"
+    "llm-security",
+    "ai-security",
+    "langchain",
+    "vercel-ai",
+    "auto-instrumentation",
+    "genai-security",
+    "cohere",
+    "aws-bedrock"
   ],
   "author": "PromptGuard <support@promptguard.co>",
   "license": "MIT",
   "repository": {
     "type": "git",
-    "url": "https://github.com/acebot712/promptguard.git",
-    "directory": "packages/sdk-node"
+    "url": "https://github.com/acebot712/promptguard-node.git"
   },
   "homepage": "https://promptguard.co",
   "bugs": {
-    "url": "https://github.com/acebot712/promptguard/issues"
+    "url": "https://github.com/acebot712/promptguard-node/issues"
   },
   "engines": {
-    "node": ">=16.0.0"
+    "node": ">=20.0.0"
   },
   "publishConfig": {
     "access": "public"
   },
-  "files": ["dist", "README.md", "LICENSE"],
-  "dependencies": {},
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ],
   "devDependencies": {
+    "@biomejs/biome": "^2.4.4",
+    "@types/jest": "^29.5.14",
     "@types/node": "^20.0.0",
+    "jest": "^29.7.0",
+    "ts-jest": "^29.4.6",
+    "ts-node": "^10.9.2",
     "typescript": "^5.0.0"
+  },
+  "peerDependencies": {
+    "@anthropic-ai/sdk": ">=0.20.0",
+    "@aws-sdk/client-bedrock-runtime": ">=3.0.0",
+    "@google/generative-ai": ">=0.5.0",
+    "@langchain/core": ">=0.1.0",
+    "ai": ">=3.0.0",
+    "cohere-ai": ">=7.0.0",
+    "openai": ">=4.0.0"
+  },
+  "peerDependenciesMeta": {
+    "openai": {
+      "optional": true
+    },
+    "@anthropic-ai/sdk": {
+      "optional": true
+    },
+    "@aws-sdk/client-bedrock-runtime": {
+      "optional": true
+    },
+    "@google/generative-ai": {
+      "optional": true
+    },
+    "@langchain/core": {
+      "optional": true
+    },
+    "ai": {
+      "optional": true
+    },
+    "cohere-ai": {
+      "optional": true
+    }
   }
 }