llm_guardrail 1.0.0

package/README.md

@@ -0,0 +1 @@
+# llm_Guardrails 

package/index.js

@@ -0,0 +1,42 @@
+import { execFile } from "child_process";
+import path from "path";
+import { fileURLToPath } from "url";
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+const PYTHON = "D:/npm/llm_Guardrails/venv/Scripts/python.exe";
+
+
+export function check(prompt) {
+  return new Promise((resolve, reject) => {
+    if (typeof prompt !== "string") {
+      return reject(new Error("Prompt must be a string"));
+    }
+
+    const scriptPath = path.join(__dirname, "model", "predict.py");
+    
+
+    execFile(
+      PYTHON,
+      [scriptPath, prompt],
+      {},
+      (err, stdout, stderr) => {
+        if (err) return reject(err);
+        if (stderr) return reject(new Error(stderr));
+
+        try {
+          const result = JSON.parse(stdout);
+          resolve({
+            allowed: result.injective === 0,
+            ...result
+          });
+        } catch (e) {
+          reject(new Error("Invalid JSON from Python"));
+        }
+      }
+    );
+  });
+}
+
+export default { check };

package/model/predict.py

@@ -0,0 +1,32 @@
+import sys
+import json
+import joblib
+from pathlib import Path
+
+# ---- safety check ----
+if len(sys.argv) < 2:
+    print(json.dumps({"error": "No input text provided"}))
+    sys.exit(1)
+
+BASE_DIR = Path(__file__).resolve().parent
+text = sys.argv[1]
+
+# ---- load combined artifact ----
+artifact = joblib.load(BASE_DIR / "prompt_injection_detector.joblib")
+
+model = artifact["model"]
+vectorizer = artifact["vectorizer"]
+threshold = artifact.get("threshold", 0.5)
+
+# ---- predict ----
+X = vectorizer.transform([text])
+prob = model.predict_proba(X)[0][1]
+flagged = prob >= threshold
+
+result = {
+    "injective": int(flagged),
+    "probability": round(float(prob), 4),
+    "threshold": threshold
+}
+
+print(json.dumps(result))

package/package.json

@@ -0,0 +1,34 @@
+{
+  "name": "llm_guardrail",
+  "version": "1.0.0",
+  "description": "A lightweight, low-latency ML-powered guardrail to stop prompt injection attacks before they reach your LLM.",
+  "homepage": "https://github.com/Frank2006x/llm_Guardrails#readme",
+  "bugs": {
+    "url": "https://github.com/Frank2006x/llm_Guardrails/issues"
+  },
+  "keywords": [
+    "llm",
+    "ai",
+    "llm-security",
+    "prompt-injection",
+    "guardrails",
+    "ai-safety",
+    "llm-guardrail",
+    "jailbreak-detection",
+    "rag-security",
+    "prompt-security",
+    "ml-security",
+    "low-latency"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/Frank2006x/llm_Guardrails.git"
+  },
+  "license": "ISC",
+  "author": "Frank2006x",
+  "type": "module",
+  "main": "index.js",
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  }
+}