promptguard-sdk 1.2.0 → 1.6.0

package/dist/patches/openai.js

@@ -0,0 +1,122 @@
+"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 base_1 = require("./base");
+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 {
+        const mod = require("openai/resources/chat/completions");
+        Completions = mod.Completions;
+        if (!Completions?.prototype?.create)
+            return false;
+    }
+    catch {
+        return false;
+    }
+    const original = Completions.prototype.create;
+    originalCreate = original;
+    Completions.prototype.create = (0, base_1.createPatchedMethod)(original, {
+        framework: "openai",
+        extractMessages: (args) => {
+            const params = (args[0] ?? {});
+            const messages = params.messages;
+            return {
+                messages: messages ? messagesToGuardFormat(messages) : [],
+                model: params.model ? String(params.model) : undefined,
+            };
+        },
+        extractResponseText: (response) => extractResponseContent(response),
+        applyRedaction: (args, redactedMessages) => {
+            const params = (args[0] ?? {});
+            const messages = params.messages;
+            return [{ ...params, messages: applyRedaction(messages, redactedMessages) }, ...args.slice(1)];
+        },
+    });
+    patched = true;
+    return true;
+}
+function revert() {
+    if (!patched || !originalCreate)
+        return;
+    try {
+        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;;AAYH,sDA2BC;AAED,wDAUC;AAeD,sBAmCC;AAED,wBAYC;AAhHD,iCAA4C;AAE5C,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,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,MAAM,QAAQ,GAAG,WAAW,CAAC,SAAS,CAAC,MAAyC,CAAA;IAChF,cAAc,GAAG,QAAQ,CAAA;IAEzB,WAAW,CAAC,SAAS,CAAC,MAAM,GAAG,IAAA,0BAAmB,EAAC,QAAQ,EAAE;QAC3D,SAAS,EAAE,QAAQ;QACnB,eAAe,EAAE,CAAC,IAAI,EAAE,EAAE;YACxB,MAAM,MAAM,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAA4B,CAAA;YACzD,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAiC,CAAA;YACzD,OAAO;gBACL,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE;gBACzD,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS;aACvD,CAAA;QACH,CAAC;QACD,mBAAmB,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,sBAAsB,CAAC,QAAQ,CAAC;QACnE,cAAc,EAAE,CAAC,IAAI,EAAE,gBAAgB,EAAE,EAAE;YACzC,MAAM,MAAM,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAA4B,CAAA;YACzD,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAqB,CAAA;YAC7C,OAAO,CAAC,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,cAAc,CAAC,QAAQ,EAAE,gBAAgB,CAAC,EAAE,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;QAChG,CAAC;KACF,CAAC,CAAA;IAEF,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,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/dist/resolve.d.ts

@@ -0,0 +1,9 @@
+/**
+ * Shared credential resolution for all PromptGuard entry points.
+ */
+export declare const DEFAULT_BASE_URL = "https://api.promptguard.co/api/v1";
+export declare function resolveCredentials(apiKey?: string, baseUrl?: string, defaultBaseUrl?: string): {
+    apiKey: string;
+    baseUrl: string;
+};
+//# sourceMappingURL=resolve.d.ts.map

package/dist/resolve.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"resolve.d.ts","sourceRoot":"","sources":["../src/resolve.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,eAAO,MAAM,gBAAgB,sCAAsC,CAAA;AAEnE,wBAAgB,kBAAkB,CAChC,MAAM,CAAC,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,MAAM,EAChB,cAAc,SAAmB,GAChC;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAWrC"}

package/dist/resolve.js

@@ -0,0 +1,19 @@
+"use strict";
+/**
+ * Shared credential resolution for all PromptGuard entry points.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DEFAULT_BASE_URL = void 0;
+exports.resolveCredentials = resolveCredentials;
+exports.DEFAULT_BASE_URL = "https://api.promptguard.co/api/v1";
+function resolveCredentials(apiKey, baseUrl, defaultBaseUrl = exports.DEFAULT_BASE_URL) {
+    const key = apiKey ?? process.env.PROMPTGUARD_API_KEY ?? "";
+    if (!key) {
+        throw new Error("PromptGuard API key required. Pass apiKey or set the PROMPTGUARD_API_KEY environment variable.");
+    }
+    return {
+        apiKey: key,
+        baseUrl: baseUrl ?? process.env.PROMPTGUARD_BASE_URL ?? defaultBaseUrl,
+    };
+}
+//# sourceMappingURL=resolve.js.map

package/dist/resolve.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"resolve.js","sourceRoot":"","sources":["../src/resolve.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAIH,gDAeC;AAjBY,QAAA,gBAAgB,GAAG,mCAAmC,CAAA;AAEnE,SAAgB,kBAAkB,CAChC,MAAe,EACf,OAAgB,EAChB,cAAc,GAAG,wBAAgB;IAEjC,MAAM,GAAG,GAAG,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,EAAE,CAAA;IAC3D,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,MAAM,IAAI,KAAK,CACb,gGAAgG,CACjG,CAAA;IACH,CAAC;IACD,OAAO;QACL,MAAM,EAAE,GAAG;QACX,OAAO,EAAE,OAAO,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,cAAc;KACvE,CAAA;AACH,CAAC"}

package/dist/version.d.ts

@@ -0,0 +1,2 @@
+export declare const SDK_VERSION = "1.6.0";
+//# sourceMappingURL=version.d.ts.map

package/dist/version.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"version.d.ts","sourceRoot":"","sources":["../src/version.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,WAAW,UAAU,CAAA"}

package/dist/version.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SDK_VERSION = void 0;
+exports.SDK_VERSION = "1.6.0";
+//# sourceMappingURL=version.js.map

package/dist/version.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"version.js","sourceRoot":"","sources":["../src/version.ts"],"names":[],"mappings":";;;AAAa,QAAA,WAAW,GAAG,OAAO,CAAA"}

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.6.0",
+  "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": {
-    "@types/node": "^20.0.0",
+    "@biomejs/biome": "^2.4.4",
+    "@types/jest": "^30.0.0",
+    "@types/node": "^25.5.0",
+    "jest": "^30.3.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
+    }
   }
 }