haystack-vaultak

Vaultak runtime security components for Haystack pipelines.

Installation

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

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