Real-time LLM hallucination guardrail — NLI + RAG fact-checking with token-level streaming halt
Project description
Director-AI
Real-time LLM hallucination guardrail — NLI + RAG fact-checking with token-level streaming halt
Try it live on Hugging Face Spaces →
Sales Pitch & Pricing · Contact Sales · invest@anulum.li
What It Does
Director-AI sits between your LLM and the user. It scores every output for hallucination before it reaches anyone — and can halt generation mid-stream if coherence drops below threshold.
from director_ai import CoherenceAgent
agent = CoherenceAgent()
result = agent.process("What color is the sky?")
print(result.coherence.score) # 0.94 — high coherence
print(result.coherence.approved) # True
print(result.coherence.h_logical) # 0.10 — low contradiction probability
print(result.coherence.h_factual) # 0.10 — low factual deviation
Three things make it different:
- Token-level streaming halt — not post-hoc review. The safety kernel monitors coherence token-by-token and severs output the moment it degrades.
- Dual-entropy scoring — NLI contradiction detection (DeBERTa) + RAG fact-checking against your own knowledge base. Both must pass.
- Your data, your rules — ingest PDFs, directories, or any text into a ChromaDB-backed knowledge base. The scorer checks LLM output against your ground truth, not a generic model.
Architecture
┌──────────────────────────┐
│ Coherence Agent │
│ (Orchestrator) │
└─────────┬────────────────┘
│
┌────────────┼────────────────┐
│ │ │
┌──────▼──────┐ ┌───▼──────────┐ ┌───▼────────────┐
│ Generator │ │ Coherence │ │ Safety │
│ (LLM │ │ Scorer │ │ Kernel │
│ backend) │ │ │ │ (streaming │
│ │ │ NLI + RAG │ │ interlock) │
└─────────────┘ └───┬──────────┘ └────────────────┘
│
┌─────────▼─────────┐
│ Ground Truth │
│ Store │
│ (ChromaDB / RAM) │
└───────────────────┘
Installation
# Basic install (heuristic scoring, no GPU needed)
pip install director-ai
# With NLI model (DeBERTa-based contradiction detection)
pip install director-ai[nli]
# With vector store (ChromaDB for custom knowledge bases)
pip install director-ai[vector]
# With high-quality embeddings (bge-large-en-v1.5)
pip install director-ai[embeddings]
# With 8-bit quantized NLI (<80ms on GPU)
pip install director-ai[quantize]
# Framework integrations
pip install director-ai[langchain]
pip install director-ai[llamaindex]
pip install director-ai[langgraph]
pip install director-ai[haystack]
pip install director-ai[crewai]
# With REST API server
pip install director-ai[server]
# Everything
pip install "director-ai[nli,vector,server,embeddings]"
# Development
git clone https://github.com/anulum/director-ai.git
cd director-ai
pip install -e ".[dev]"
Usage
Score a single response
from director_ai.core import CoherenceScorer, GroundTruthStore
store = GroundTruthStore()
store.add("sky color", "The sky is blue due to Rayleigh scattering.")
scorer = CoherenceScorer(threshold=0.6, ground_truth_store=store)
approved, score = scorer.review("What color is the sky?", "The sky is green.")
print(approved) # False — contradicts ground truth
print(score.score) # 0.42
With a real LLM backend
from director_ai import CoherenceAgent
# Works with any OpenAI-compatible endpoint (llama.cpp, vLLM, Ollama, etc.)
agent = CoherenceAgent(llm_api_url="http://localhost:8080/completion")
result = agent.process("Explain quantum entanglement")
if result.halted:
print("Output blocked — coherence too low")
else:
print(result.output)
Token-level streaming with halt
from director_ai.core import StreamingKernel
kernel = StreamingKernel(hard_limit=0.4, window_size=5, window_threshold=0.5)
session = kernel.stream_tokens(
token_generator=my_token_iterator,
coherence_callback=lambda tok: my_scorer(tok),
)
for event in session.events:
if event.halted:
print(f"\n[HALTED — {session.halt_reason}]")
break
print(event.token, end="")
NLI-based scoring (requires torch)
from director_ai.core import CoherenceScorer
scorer = CoherenceScorer(use_nli=True, threshold=0.6)
approved, score = scorer.review(
"The Earth orbits the Sun.",
"The Sun orbits the Earth."
)
print(score.h_logical) # High — NLI detects contradiction
Custom knowledge base with ChromaDB
from director_ai.core import VectorGroundTruthStore
store = VectorGroundTruthStore() # Uses ChromaDB
store.add_fact("company policy", "Refunds are available within 30 days.")
store.add_fact("pricing", "Enterprise plan starts at $99/month.")
scorer = CoherenceScorer(ground_truth_store=store)
approved, score = scorer.review(
"What is the refund policy?",
"We offer full refunds within 90 days." # Wrong
)
# approved = False — contradicts your KB
LangChain integration
pip install director-ai[langchain,nli]
from director_ai.integrations.langchain import DirectorAIGuard
guard = DirectorAIGuard(
facts={"refund": "Refunds available within 30 days."},
threshold=0.6,
use_nli=True,
)
# Pipe after any LLM in a chain
chain = my_llm | guard
result = chain.invoke({"query": "What is the refund policy?"})
print(result["approved"]) # False if hallucinated
print(result["score"]) # 0.0–1.0 coherence
Raises HallucinationError if raise_on_fail=True. Async supported via ainvoke().
LlamaIndex integration
pip install director-ai[llamaindex,nli]
from director_ai.integrations.llamaindex import DirectorAIPostprocessor
postprocessor = DirectorAIPostprocessor(
facts={"pricing": "Enterprise plan starts at $99/month."},
threshold=0.6,
)
# Filters out hallucinated nodes before they reach the user
query_engine = index.as_query_engine(
node_postprocessors=[postprocessor]
)
response = query_engine.query("What does Enterprise cost?")
Adds director_ai_score metadata to surviving nodes. Also usable standalone via postprocessor.check(query, response).
LangGraph integration
from director_ai.integrations.langgraph import director_ai_node, director_ai_conditional_edge
node = director_ai_node(facts={"policy": "Refunds within 30 days."}, on_fail="flag")
edge = director_ai_conditional_edge("output", "retry")
# Wire into your LangGraph StateGraph
Haystack integration
from director_ai.integrations.haystack import DirectorAIChecker
checker = DirectorAIChecker(facts={"policy": "Refunds within 30 days."})
result = checker.run(query="Refund policy?", replies=["60-day refunds."])
print(result["scores"]) # [CoherenceScore(...)]
CrewAI integration
from director_ai.integrations.crewai import DirectorAITool
tool = DirectorAITool(facts={"policy": "Refunds within 30 days."})
result = tool.check("Refund policy?", "We offer 30-day refunds.")
print(result["approved"]) # True
Score caching
scorer = CoherenceScorer(
threshold=0.6,
cache_size=1024, # LRU cache for streaming deduplication
cache_ttl=300, # TTL in seconds
)
More examples
| Example | Backend | What it shows |
|---|---|---|
quickstart.py |
None | Guard any output in 10 lines |
openai_guard.py |
OpenAI | Score + streaming halt for GPT-4o |
ollama_guard.py |
Ollama | Local LLM guard with Llama 3 |
langchain_guard.py |
LangChain | Full chain guardrail |
streaming_halt_demo.py |
Simulated | All 3 halt mechanisms visualised |
Interactive demo
pip install director-ai gradio
python demo/app.py
Scoring Formula
Coherence = 1 - (0.6 * H_logical + 0.4 * H_factual)
| Component | Source | Range | Meaning |
|---|---|---|---|
| H_logical | NLI model (DeBERTa) | 0-1 | Contradiction probability |
| H_factual | RAG retrieval | 0-1 | Ground truth deviation |
- Score >= 0.6 → approved (configurable)
- Score < 0.5 → safety kernel emergency halt
Benchmarks
Evaluated on LLM-AggreFact (29,320 samples across 11 datasets):
| Model | AggreFact Balanced Acc | Latency (avg) |
|---|---|---|
| DeBERTa-v3-base (baseline) | 66.2% | 220 ms |
| Fine-tuned DeBERTa-v3-large | 64.7% | 223 ms |
| Fine-tuned DeBERTa-v3-base | 59.0% | 220 ms |
Per-dataset highlights:
| Dataset | Balanced Accuracy | Notes |
|---|---|---|
| Reveal | 80.7% | Strong on factual claims |
| FactCheck-GPT | 71.7% | Good on GPT-generated text |
| Lfqa | 64.8% | Long-form QA |
| RAGTruth | 58.9% | RAG-specific hallucination |
| AggreFact-CNN | 53.0% | Summarization (known weak spot) |
Head-to-head (same benchmark, same metric — LLM-AggreFact leaderboard):
| Tool | Bal. Acc | Params | Latency | Streaming |
|---|---|---|---|---|
| Bespoke-MiniCheck-7B | 77.4% | 7B | ~100 ms (GPU) | No |
| MiniCheck-Flan-T5-L | 75.0% | 0.8B | ~120 ms | No |
| MiniCheck-DeBERTa-L | 72.6% | 0.4B | ~120 ms | No |
| HHEM-2.1-Open | 71.8% | ~0.4B | ~200 ms | No |
| Director-AI | 66.2% | 0.4B | 220 ms | Yes |
Honest assessment: The NLI scorer alone is not state-of-the-art. Director-AI's value is in the system — combining NLI with your own KB facts, streaming token-level gating, and configurable halt thresholds. No competitor offers real-time streaming halt. The NLI component is pluggable; swap in any model that improves on these numbers.
Full comparison with SelfCheckGPT, RAGAS, NeMo Guardrails, Lynx, and others
in benchmarks/comparison/. Benchmark scripts in
benchmarks/. Fine-tuning pipeline in training/.
Package Structure
src/director_ai/
├── core/ # Production API
│ ├── agent.py # CoherenceAgent — main orchestrator
│ ├── scorer.py # Dual-entropy coherence scorer
│ ├── kernel.py # Safety kernel (streaming interlock)
│ ├── streaming.py # Token-level streaming oversight
│ ├── async_streaming.py # Non-blocking async streaming
│ ├── nli.py # NLI scorer (DeBERTa)
│ ├── actor.py # LLM generator interface
│ ├── knowledge.py # Ground truth store (in-memory)
│ ├── vector_store.py # Vector store (ChromaDB / sentence-transformers)
│ ├── cache.py # LRU score cache (blake2b, TTL)
│ ├── policy.py # YAML declarative policy engine
│ ├── audit.py # Structured JSONL audit logger
│ ├── tenant.py # Multi-tenant KB isolation
│ ├── sanitizer.py # Prompt injection hardening
│ └── types.py # CoherenceScore, ReviewResult
├── integrations/ # Framework integrations
│ ├── langchain.py # LangChain Runnable guardrail
│ ├── llamaindex.py # LlamaIndex postprocessor
│ ├── langgraph.py # LangGraph node + conditional edge
│ ├── haystack.py # Haystack 2.x component
│ └── crewai.py # CrewAI tool
├── cli.py # CLI: review, process, batch, serve
├── server.py # FastAPI REST wrapper
benchmarks/ # AggreFact evaluation suite
training/ # DeBERTa fine-tuning pipeline
Testing
pytest tests/ -v
License & Pricing
Dual-licensed:
- Open-Source: GNU AGPL v3.0 — research, personal use, open-source projects. Full source, self-host, no restrictions beyond AGPL copyleft obligations.
- Commercial: Proprietary license from ANULUM — removes copyleft, allows closed-source and SaaS deployment.
Commercial Tiers
| Tier | Monthly | Yearly | Best for |
|---|---|---|---|
| Hobbyist | $9 | $90 | Students, side projects, experiments. 1 local deployment, community support (GitHub/Discord), delayed updates. |
| Indie | $49 | $490 | Solo devs, bootstrapped teams (<$2M ARR). 1 production deployment, email support, 12 months updates. |
| Pro | $249 | $2,490 | Startups & scale-ups. Unlimited internal devs, multiple envs, Slack priority support, early releases. |
| Enterprise | Custom | Custom | Large orgs. SLA (99.9%), on-prem/air-gapped, SOC2/HIPAA-ready, dedicated engineer, custom NLI fine-tunes. |
Perpetual license: $1,299 one-time (Indie equivalent). First 50 commercial licensees: 50% off first year.
Contact: anulum.li/contact or invest@anulum.li
See NOTICE for full terms and third-party acknowledgements.
Roadmap
Shipped in v1.2.0
- Score caching — LRU cache with blake2b keys and TTL for streaming dedup
- Framework integrations — LangGraph nodes, Haystack components, CrewAI tools
- Quantized NLI — 8-bit bitsandbytes quantization for <80ms GPU inference
- Upgraded embeddings — bge-large-en-v1.5 via
SentenceTransformerBackend - MkDocs site — full API reference, deployment guides, domain cookbooks
- Enhanced demo — side-by-side comparison with token-level highlighting
Shipped in v1.1.0
- Native SDK interceptors —
guard(OpenAI(), facts={...})wraps any OpenAI/Anthropic client with transparent coherence scoring - MiniCheck backend — 72.6% balanced accuracy on LLM-AggreFact
- Evidence return — every
CoherenceScorecarries top-K chunks, NLI premise/hypothesis, and similarity distances - Graceful fallbacks —
fallback="retrieval"/"disclaimer"+ soft warning zone + streamingon_haltcallback
Completed
- Score caching, LangGraph/Haystack/CrewAI, quantized NLI, MkDocs site
-
director-ai eval— structured CLI benchmarking - Native OpenAI/Anthropic SDK interceptors (
guard()) - Evidence schema on all rejections
- Graceful fallback patterns (retrieval, disclaimer, soft warning)
- End-to-end guardrail benchmark (600+ traces, 8 metrics)
- HuggingFace Spaces live demo
Next
- Chunked NLI scoring for long documents
- Prometheus histogram latency buckets
-
director-ai serve --workers Nmulti-process mode
Citation
@software{sotek2026director,
author = {Sotek, Miroslav},
title = {Director-AI: Real-time LLM Hallucination Guardrail},
year = {2026},
url = {https://github.com/anulum/director-ai},
version = {1.2.0},
license = {AGPL-3.0-or-later}
}
Contributing
See CONTRIBUTING.md for guidelines. By contributing, you agree to the Code of Conduct and AGPL v3 licensing terms.
Security
See SECURITY.md for reporting vulnerabilities.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file director_ai-1.2.0.tar.gz.
File metadata
- Download URL: director_ai-1.2.0.tar.gz
- Upload date:
- Size: 108.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.7
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
c6d9c4fc933eb910baa9cefbd7b11d11f9de85ee798344d068e9c613a665adaa
|
|
| MD5 |
c92664a75c6496385826d14fa928aefd
|
|
| BLAKE2b-256 |
5c25bfee9b87c051c4a3d246ef1f5842e0dc6f51a1dff8fd81f3bdcf25b2bfb2
|
Provenance
The following attestation bundles were made for director_ai-1.2.0.tar.gz:
Publisher:
publish.yml on anulum/director-ai
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
director_ai-1.2.0.tar.gz -
Subject digest:
c6d9c4fc933eb910baa9cefbd7b11d11f9de85ee798344d068e9c613a665adaa - Sigstore transparency entry: 1003724436
- Sigstore integration time:
-
Permalink:
anulum/director-ai@3fcb12d703c0530e68f12845ecd41c27452b94a8 -
Branch / Tag:
refs/tags/v1.2.0 - Owner: https://github.com/anulum
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
publish.yml@3fcb12d703c0530e68f12845ecd41c27452b94a8 -
Trigger Event:
release
-
Statement type:
File details
Details for the file director_ai-1.2.0-py3-none-any.whl.
File metadata
- Download URL: director_ai-1.2.0-py3-none-any.whl
- Upload date:
- Size: 84.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.7
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
da8914b4f05f2463744d218a8da96d977a09d7b1e13159f50b67d360e6692a3c
|
|
| MD5 |
72823dcc918592c69b9a7ff1f2895c4e
|
|
| BLAKE2b-256 |
23c9d8f09b4973d89ae23f4615219f651668b87a5f6bba440bbb71cb953c5451
|
Provenance
The following attestation bundles were made for director_ai-1.2.0-py3-none-any.whl:
Publisher:
publish.yml on anulum/director-ai
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
director_ai-1.2.0-py3-none-any.whl -
Subject digest:
da8914b4f05f2463744d218a8da96d977a09d7b1e13159f50b67d360e6692a3c - Sigstore transparency entry: 1003724439
- Sigstore integration time:
-
Permalink:
anulum/director-ai@3fcb12d703c0530e68f12845ecd41c27452b94a8 -
Branch / Tag:
refs/tags/v1.2.0 - Owner: https://github.com/anulum
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
publish.yml@3fcb12d703c0530e68f12845ecd41c27452b94a8 -
Trigger Event:
release
-
Statement type: