haystack-vaultak 0.1.0__tar.gz

haystack_vaultak-0.1.0/PKG-INFO

@@ -0,0 +1,57 @@
+Metadata-Version: 2.4
+Name: haystack-vaultak
+Version: 0.1.0
+Summary: Vaultak runtime security components for Haystack pipelines
+License: MIT
+Project-URL: Homepage, https://vaultak.com
+Project-URL: Documentation, https://docs.vaultak.com
+Project-URL: Repository, https://github.com/samueloladji-beep/haystack-vaultak
+Keywords: haystack,vaultak,security,ai-agents,runtime-security,pii
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: Security
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: haystack-ai>=2.0.0
+Requires-Dist: vaultak>=0.1.0
+
+# haystack-vaultak
+
+Vaultak runtime security components for [Haystack](https://haystack.deepset.ai) pipelines.
+
+## Installation
+
+```bash
+pip install haystack-vaultak
+```
+
+## Components
+
+- **VaultakSecurityChecker** — Risk-scores every query before it enters your pipeline. Blocks high-risk inputs.
+- **VaultakPIIMasker** — Masks PII in LLM replies before they reach users.
+
+## Quick start
+
+```python
+from haystack import Pipeline
+from haystack.components.generators.chat import OpenAIChatGenerator
+from haystack_vaultak import VaultakSecurityChecker, VaultakPIIMasker
+import os
+
+pipeline = Pipeline()
+pipeline.add_component("security", VaultakSecurityChecker(api_key=os.environ["VAULTAK_API_KEY"]))
+pipeline.add_component("llm", OpenAIChatGenerator(model="gpt-4o-mini"))
+pipeline.add_component("pii_masker", VaultakPIIMasker(api_key=os.environ["VAULTAK_API_KEY"]))
+
+pipeline.connect("security.query", "llm.query")
+pipeline.connect("llm.replies", "pii_masker.replies")
+
+result = pipeline.run({"security": {"query": "What is the capital of France?"}})
+print(result["pii_masker"]["replies"][0].text)
+```
+
+## License
+
+MIT

haystack_vaultak-0.1.0/README.md

@@ -0,0 +1,38 @@
+# haystack-vaultak
+
+Vaultak runtime security components for [Haystack](https://haystack.deepset.ai) pipelines.
+
+## Installation
+
+```bash
+pip install haystack-vaultak
+```
+
+## Components
+
+- **VaultakSecurityChecker** — Risk-scores every query before it enters your pipeline. Blocks high-risk inputs.
+- **VaultakPIIMasker** — Masks PII in LLM replies before they reach users.
+
+## Quick start
+
+```python
+from haystack import Pipeline
+from haystack.components.generators.chat import OpenAIChatGenerator
+from haystack_vaultak import VaultakSecurityChecker, VaultakPIIMasker
+import os
+
+pipeline = Pipeline()
+pipeline.add_component("security", VaultakSecurityChecker(api_key=os.environ["VAULTAK_API_KEY"]))
+pipeline.add_component("llm", OpenAIChatGenerator(model="gpt-4o-mini"))
+pipeline.add_component("pii_masker", VaultakPIIMasker(api_key=os.environ["VAULTAK_API_KEY"]))
+
+pipeline.connect("security.query", "llm.query")
+pipeline.connect("llm.replies", "pii_masker.replies")
+
+result = pipeline.run({"security": {"query": "What is the capital of France?"}})
+print(result["pii_masker"]["replies"][0].text)
+```
+
+## License
+
+MIT

haystack_vaultak-0.1.0/haystack_vaultak/__init__.py

@@ -0,0 +1,4 @@
+from .components import VaultakSecurityChecker, VaultakPIIMasker
+
+__version__ = "0.1.0"
+__all__ = ["VaultakSecurityChecker", "VaultakPIIMasker"]

haystack_vaultak-0.1.0/haystack_vaultak/components.py

@@ -0,0 +1,162 @@
+"""
+Vaultak runtime security components for Haystack pipelines.
+"""
+import os
+from typing import Optional
+
+from haystack import component, default_from_dict, default_to_dict
+from haystack.dataclasses import ChatMessage
+
+
+@component
+class VaultakSecurityChecker:
+    """
+    Haystack component that risk-scores every query before it enters your pipeline.
+
+    Insert before your retriever or LLM. Queries whose risk score meets or exceeds
+    the threshold raise a RuntimeError so the pipeline halts cleanly.
+
+    Usage:
+        from haystack_vaultak import VaultakSecurityChecker
+
+        checker = VaultakSecurityChecker(api_key="vtk_...", threshold=7.0)
+        pipeline.add_component("security", checker)
+        pipeline.connect("security.query", "retriever.query")
+    """
+
+    def __init__(
+        self,
+        api_key: Optional[str] = None,
+        agent_name: str = "haystack-pipeline",
+        threshold: float = 7.0,
+        verbose: bool = False,
+    ):
+        from vaultak import Vaultak
+        self.api_key = api_key or os.environ.get("VAULTAK_API_KEY", "")
+        self.agent_name = agent_name
+        self.threshold = threshold
+        self.verbose = verbose
+        self._vt = Vaultak(api_key=self.api_key, agent_name=agent_name)
+
+    @component.output_types(query=str)
+    def run(self, query: str) -> dict:
+        """
+        Risk-score the query. Passes it through if safe; raises RuntimeError if blocked.
+
+        Args:
+            query: The user query to screen.
+
+        Returns:
+            dict with key 'query' containing the original query if it passes.
+
+        Raises:
+            RuntimeError: If the risk score meets or exceeds the threshold.
+        """
+        try:
+            result = self._vt.check(
+                action_type="query",
+                resource=query[:200],
+                agent_id=self.agent_name,
+            )
+            score = result.get("score", 0)
+            if self.verbose:
+                print(f"[Vaultak] Query risk score: {score}/100")
+
+            if score >= self.threshold * 10:  # Convert 0-10 to 0-100
+                raise RuntimeError(
+                    f"[Vaultak] Query blocked — risk score {score}/100 meets or exceeds "
+                    f"threshold {self.threshold * 10}/100. Review at app.vaultak.com"
+                )
+        except RuntimeError:
+            raise
+        except Exception as e:
+            if self.verbose:
+                print(f"[Vaultak] Security check failed (non-blocking): {e}")
+
+        self._vt.log_action(
+            action_type="query",
+            resource=query[:200],
+            agent_id=self.agent_name,
+        )
+        return {"query": query}
+
+    def to_dict(self) -> dict:
+        return default_to_dict(
+            self,
+            agent_name=self.agent_name,
+            threshold=self.threshold,
+            verbose=self.verbose,
+        )
+
+    @classmethod
+    def from_dict(cls, data: dict):
+        return default_from_dict(cls, data)
+
+
+@component
+class VaultakPIIMasker:
+    """
+    Haystack component that masks PII in LLM replies before they reach users.
+
+    Insert after your LLM generator. Scans every reply and replaces PII
+    with safe placeholders.
+
+    Usage:
+        from haystack_vaultak import VaultakPIIMasker
+
+        masker = VaultakPIIMasker(api_key="vtk_...")
+        pipeline.add_component("pii_masker", masker)
+        pipeline.connect("llm.replies", "pii_masker.replies")
+    """
+
+    def __init__(
+        self,
+        api_key: Optional[str] = None,
+        agent_name: str = "haystack-pipeline",
+        verbose: bool = False,
+    ):
+        from vaultak import Vaultak
+        self.api_key = api_key or os.environ.get("VAULTAK_API_KEY", "")
+        self.agent_name = agent_name
+        self.verbose = verbose
+        self._vt = Vaultak(api_key=self.api_key, agent_name=agent_name)
+
+    @component.output_types(replies=list)
+    def run(self, replies: list) -> dict:
+        """
+        Mask PII in a list of ChatMessage replies.
+
+        Args:
+            replies: List of ChatMessage objects from the LLM.
+
+        Returns:
+            dict with key 'replies' containing the masked messages.
+        """
+        masked_replies = []
+        for reply in replies:
+            try:
+                if hasattr(reply, 'text') and reply.text:
+                    masked_text = self._vt.mask_pii(reply.text)
+                    if self.verbose and masked_text != reply.text:
+                        print(f"[Vaultak] PII masked in reply")
+                    masked_replies.append(
+                        ChatMessage.from_assistant(masked_text)
+                    )
+                else:
+                    masked_replies.append(reply)
+            except Exception as e:
+                if self.verbose:
+                    print(f"[Vaultak] PII masking failed (non-blocking): {e}")
+                masked_replies.append(reply)
+        return {"replies": masked_replies}
+
+    def to_dict(self) -> dict:
+        return default_to_dict(
+            self,
+            agent_name=self.agent_name,
+            verbose=self.verbose,
+        )
+
+    @classmethod
+    def from_dict(cls, data: dict):
+        return default_from_dict(cls, data)

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

@@ -0,0 +1,57 @@
+Metadata-Version: 2.4
+Name: haystack-vaultak
+Version: 0.1.0
+Summary: Vaultak runtime security components for Haystack pipelines
+License: MIT
+Project-URL: Homepage, https://vaultak.com
+Project-URL: Documentation, https://docs.vaultak.com
+Project-URL: Repository, https://github.com/samueloladji-beep/haystack-vaultak
+Keywords: haystack,vaultak,security,ai-agents,runtime-security,pii
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: Security
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: haystack-ai>=2.0.0
+Requires-Dist: vaultak>=0.1.0
+
+# haystack-vaultak
+
+Vaultak runtime security components for [Haystack](https://haystack.deepset.ai) pipelines.
+
+## Installation
+
+```bash
+pip install haystack-vaultak
+```
+
+## Components
+
+- **VaultakSecurityChecker** — Risk-scores every query before it enters your pipeline. Blocks high-risk inputs.
+- **VaultakPIIMasker** — Masks PII in LLM replies before they reach users.
+
+## Quick start
+
+```python
+from haystack import Pipeline
+from haystack.components.generators.chat import OpenAIChatGenerator
+from haystack_vaultak import VaultakSecurityChecker, VaultakPIIMasker
+import os
+
+pipeline = Pipeline()
+pipeline.add_component("security", VaultakSecurityChecker(api_key=os.environ["VAULTAK_API_KEY"]))
+pipeline.add_component("llm", OpenAIChatGenerator(model="gpt-4o-mini"))
+pipeline.add_component("pii_masker", VaultakPIIMasker(api_key=os.environ["VAULTAK_API_KEY"]))
+
+pipeline.connect("security.query", "llm.query")
+pipeline.connect("llm.replies", "pii_masker.replies")
+
+result = pipeline.run({"security": {"query": "What is the capital of France?"}})
+print(result["pii_masker"]["replies"][0].text)
+```
+
+## License
+
+MIT

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

@@ -0,0 +1,9 @@
+README.md
+pyproject.toml
+haystack_vaultak/__init__.py
+haystack_vaultak/components.py
+haystack_vaultak.egg-info/PKG-INFO
+haystack_vaultak.egg-info/SOURCES.txt
+haystack_vaultak.egg-info/dependency_links.txt
+haystack_vaultak.egg-info/requires.txt
+haystack_vaultak.egg-info/top_level.txt

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

@@ -0,0 +1 @@
+

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

@@ -0,0 +1,2 @@
+haystack-ai>=2.0.0
+vaultak>=0.1.0

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

@@ -0,0 +1 @@
+haystack_vaultak

haystack_vaultak-0.1.0/pyproject.toml

@@ -0,0 +1,32 @@
+[build-system]
+requires = ["setuptools>=68", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "haystack-vaultak"
+version = "0.1.0"
+description = "Vaultak runtime security components for Haystack pipelines"
+readme = "README.md"
+license = { text = "MIT" }
+requires-python = ">=3.8"
+keywords = ["haystack", "vaultak", "security", "ai-agents", "runtime-security", "pii"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Topic :: Security",
+]
+dependencies = [
+    "haystack-ai>=2.0.0",
+    "vaultak>=0.1.0",
+]
+
+[project.urls]
+Homepage = "https://vaultak.com"
+Documentation = "https://docs.vaultak.com"
+Repository = "https://github.com/samueloladji-beep/haystack-vaultak"
+
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["haystack_vaultak*"]

haystack_vaultak-0.1.0/setup.cfg

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