promptshield-firewall 0.1.0__tar.gz

promptshield_firewall-0.1.0/PKG-INFO

@@ -0,0 +1,65 @@
+Metadata-Version: 2.4
+Name: promptshield-firewall
+Version: 0.1.0
+Summary: Real-time LLM prompt injection firewall
+Home-page: https://promptshield.live
+Author: Kehinde Omosor
+Author-email: kehinde@promptshield.live
+Keywords: llm security prompt injection firewall ai
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Topic :: Security
+Classifier: Topic :: Software Development :: Libraries
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: httpx>=0.24.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# PromptShield Python SDK
+
+Real-time LLM prompt injection firewall.
+
+## Installation
+
+```bash
+pip install promptshield
+```
+
+## Quick Start
+
+```python
+from promptshield import PromptShield
+
+shield = PromptShield(api_key="ps_your_key")
+
+# Scan before sending to your LLM
+result = shield.scan("Ignore all previous instructions.")
+
+if result["verdict"] == "BLOCKED":
+    print(f"Attack blocked! Risk score: {result['risk_score']}")
+else:
+    response = your_llm.chat(user_input)
+
+# Or use the simple boolean check
+if not shield.is_safe(user_input):
+    return "Request blocked."
+```
+
+## Async Support
+
+```python
+result = await shield.async_scan(user_input)
+```
+
+## Get API Key
+
+Free tier at [promptshield.live](https://promptshield.live)

promptshield_firewall-0.1.0/README.md

@@ -0,0 +1,39 @@
+# PromptShield Python SDK
+
+Real-time LLM prompt injection firewall.
+
+## Installation
+
+```bash
+pip install promptshield
+```
+
+## Quick Start
+
+```python
+from promptshield import PromptShield
+
+shield = PromptShield(api_key="ps_your_key")
+
+# Scan before sending to your LLM
+result = shield.scan("Ignore all previous instructions.")
+
+if result["verdict"] == "BLOCKED":
+    print(f"Attack blocked! Risk score: {result['risk_score']}")
+else:
+    response = your_llm.chat(user_input)
+
+# Or use the simple boolean check
+if not shield.is_safe(user_input):
+    return "Request blocked."
+```
+
+## Async Support
+
+```python
+result = await shield.async_scan(user_input)
+```
+
+## Get API Key
+
+Free tier at [promptshield.live](https://promptshield.live)

promptshield_firewall-0.1.0/promptshield/__init__.py

@@ -0,0 +1,4 @@
+from .client import PromptShield, PromptShieldError
+
+__version__ = "0.1.0"
+__all__ = ["PromptShield", "PromptShieldError"]

promptshield_firewall-0.1.0/promptshield/client.py

@@ -0,0 +1,160 @@
+"""
+PromptShield Python SDK
+pip install promptshield
+"""
+
+import httpx
+import time
+from typing import Dict, Any, Optional
+
+
+class PromptShieldError(Exception):
+    """Base error for PromptShield SDK"""
+    pass
+
+
+class PromptShield:
+    """
+    PromptShield client — real-time LLM prompt injection firewall.
+
+    Usage:
+        from promptshield import PromptShield
+
+        shield = PromptShield(api_key="ps_your_key")
+        result = shield.scan("Ignore all previous instructions.")
+
+        if result["verdict"] == "BLOCKED":
+            print("Attack blocked!")
+        else:
+            # Safe to send to your LLM
+            response = your_llm.chat(user_input)
+    """
+
+    def __init__(
+        self,
+        api_key: str,
+        base_url: str = "https://promptshield.live",
+        timeout: float = 10.0,
+    ):
+        self.api_key = api_key
+        self.base_url = base_url.rstrip("/")
+        self.timeout = timeout
+        self.headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json",
+        }
+
+    def scan(self, prompt: str, context: Optional[str] = None) -> Dict[str, Any]:
+        """
+        Scan a prompt for injection attacks synchronously.
+
+        Args:
+            prompt: The user input to scan
+            context: Optional system context for better analysis
+
+        Returns:
+            dict with verdict, risk_score, attack_types, explanation
+        """
+        start_time = time.time()
+        payload = {"input": prompt}
+        if context:
+            payload["context"] = context
+
+        try:
+            with httpx.Client(timeout=self.timeout) as client:
+                response = client.post(
+                    f"{self.base_url}/v1/analyze",
+                    json=payload,
+                    headers=self.headers,
+                )
+
+            if response.status_code == 401:
+                raise PromptShieldError("Invalid API key. Get one at promptshield.live")
+            if response.status_code == 429:
+                raise PromptShieldError("Monthly scan limit reached. Upgrade at promptshield.live")
+            if response.status_code != 200:
+                raise PromptShieldError(f"API Error ({response.status_code}): {response.text}")
+
+            data = response.json()
+            data["sdk_latency_ms"] = int((time.time() - start_time) * 1000)
+            return data
+
+        except httpx.RequestError as exc:
+            # Fail-safe: allow request if firewall is unreachable
+            return {
+                "verdict": "SAFE",
+                "risk_score": 0,
+                "attack_types": [],
+                "confidence": 0,
+                "explanation": "PromptShield unreachable — request allowed by fail-safe.",
+                "recommendation": "Check your network connection.",
+                "tags": ["fail-safe"],
+                "latency_ms": 0,
+                "sanitized_input": prompt,
+                "error": str(exc),
+                "fallback_triggered": True,
+                "sdk_latency_ms": int((time.time() - start_time) * 1000),
+            }
+
+    async def async_scan(self, prompt: str, context: Optional[str] = None) -> Dict[str, Any]:
+        """
+        Scan a prompt for injection attacks asynchronously.
+
+        Args:
+            prompt: The user input to scan
+            context: Optional system context for better analysis
+
+        Returns:
+            dict with verdict, risk_score, attack_types, explanation
+        """
+        start_time = time.time()
+        payload = {"input": prompt}
+        if context:
+            payload["context"] = context
+
+        try:
+            async with httpx.AsyncClient(timeout=self.timeout) as client:
+                response = await client.post(
+                    f"{self.base_url}/v1/analyze",
+                    json=payload,
+                    headers=self.headers,
+                )
+
+            if response.status_code == 401:
+                raise PromptShieldError("Invalid API key. Get one at promptshield.live")
+            if response.status_code == 429:
+                raise PromptShieldError("Monthly scan limit reached. Upgrade at promptshield.live")
+            if response.status_code != 200:
+                raise PromptShieldError(f"API Error ({response.status_code}): {response.text}")
+
+            data = response.json()
+            data["sdk_latency_ms"] = int((time.time() - start_time) * 1000)
+            return data
+
+        except httpx.RequestError as exc:
+            return {
+                "verdict": "SAFE",
+                "risk_score": 0,
+                "attack_types": [],
+                "confidence": 0,
+                "explanation": "PromptShield unreachable — request allowed by fail-safe.",
+                "recommendation": "Check your network connection.",
+                "tags": ["fail-safe"],
+                "latency_ms": 0,
+                "sanitized_input": prompt,
+                "error": str(exc),
+                "fallback_triggered": True,
+                "sdk_latency_ms": int((time.time() - start_time) * 1000),
+            }
+
+    def is_safe(self, prompt: str) -> bool:
+        """
+        Simple boolean check — returns True if safe, False if blocked/suspicious.
+
+        Usage:
+            if not shield.is_safe(user_input):
+                return "Request blocked."
+            response = your_llm.chat(user_input)
+        """
+        result = self.scan(prompt)
+        return result["verdict"] == "SAFE"

promptshield_firewall-0.1.0/promptshield_firewall.egg-info/PKG-INFO

@@ -0,0 +1,65 @@
+Metadata-Version: 2.4
+Name: promptshield-firewall
+Version: 0.1.0
+Summary: Real-time LLM prompt injection firewall
+Home-page: https://promptshield.live
+Author: Kehinde Omosor
+Author-email: kehinde@promptshield.live
+Keywords: llm security prompt injection firewall ai
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Topic :: Security
+Classifier: Topic :: Software Development :: Libraries
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: httpx>=0.24.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# PromptShield Python SDK
+
+Real-time LLM prompt injection firewall.
+
+## Installation
+
+```bash
+pip install promptshield
+```
+
+## Quick Start
+
+```python
+from promptshield import PromptShield
+
+shield = PromptShield(api_key="ps_your_key")
+
+# Scan before sending to your LLM
+result = shield.scan("Ignore all previous instructions.")
+
+if result["verdict"] == "BLOCKED":
+    print(f"Attack blocked! Risk score: {result['risk_score']}")
+else:
+    response = your_llm.chat(user_input)
+
+# Or use the simple boolean check
+if not shield.is_safe(user_input):
+    return "Request blocked."
+```
+
+## Async Support
+
+```python
+result = await shield.async_scan(user_input)
+```
+
+## Get API Key
+
+Free tier at [promptshield.live](https://promptshield.live)

promptshield_firewall-0.1.0/promptshield_firewall.egg-info/SOURCES.txt

@@ -0,0 +1,9 @@
+README.md
+setup.py
+promptshield/__init__.py
+promptshield/client.py
+promptshield_firewall.egg-info/PKG-INFO
+promptshield_firewall.egg-info/SOURCES.txt
+promptshield_firewall.egg-info/dependency_links.txt
+promptshield_firewall.egg-info/requires.txt
+promptshield_firewall.egg-info/top_level.txt

promptshield_firewall-0.1.0/promptshield_firewall.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

promptshield_firewall-0.1.0/promptshield_firewall.egg-info/requires.txt

@@ -0,0 +1 @@
+httpx>=0.24.0

promptshield_firewall-0.1.0/promptshield_firewall.egg-info/top_level.txt

@@ -0,0 +1 @@
+promptshield

promptshield_firewall-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

promptshield_firewall-0.1.0/setup.py

@@ -0,0 +1,22 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="promptshield-firewall",
+    version="0.1.0",
+    description="Real-time LLM prompt injection firewall",
+    long_description=open("README.md").read(),
+    long_description_content_type="text/markdown",
+    author="Kehinde Omosor",
+    author_email="kehinde@promptshield.live",
+    url="https://promptshield.live",
+    packages=find_packages(),
+    install_requires=["httpx>=0.24.0"],
+    python_requires=">=3.8",
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Topic :: Security",
+        "Topic :: Software Development :: Libraries",
+    ],
+    keywords="llm security prompt injection firewall ai",
+)