LlamaIndex Integration

pip install llamaindex-ambientmeta

from llama_index.core import VectorStoreIndex, Settings
from llama_index.llms.openai import OpenAI
from llamaindex_ambientmeta import PrivacyLLM

# Wrap your LLM with privacy protection
Settings.llm = PrivacyLLM(
    llm=OpenAI(),
    api_key="am_live_xxx",
)

# Build your index as normal
index = VectorStoreIndex.from_documents(documents)
query_engine = index.as_query_engine()

# Queries are automatically sanitized
response = query_engine.query("What's in John Smith's contract?")
# LLM never sees real names

chat_engine = index.as_chat_engine()

# Multi-turn conversations stay private
response = chat_engine.chat("Tell me about employee EMP-123456")
response = chat_engine.chat("What's their email?")

Settings.llm = PrivacyLLM(
    llm=OpenAI(model="gpt-4"),
    api_key="am_live_xxx",
    entities=["PERSON", "EMAIL_ADDRESS", "SSN"],  # Optional: detect specific entities only
    auto_rehydrate=True,   # Automatically restore PII in responses (default: True)
)

from llama_index.llms.anthropic import Anthropic

Settings.llm = PrivacyLLM(
    llm=Anthropic(model="claude-sonnet-4-20250514"),
    api_key="am_live_xxx",
)