GuardEx

from guardex import Guard

guard = Guard()  # zero config

result = guard.screen(
    user_msg, gate="input",
)

if result.blocked:
    print(result.classify.category)
else:
    if result.pii.has_pii:
        n = len(result.pii.entities)
        print(f"masked {n} entities")
    print(result.text)

from google import genai
from guardex import Guard

guard = Guard()
client = genai.Client()

# before it reaches the model
safe = guard.screen(
    user_msg, gate="input",
)
reply = client.models.generate_content(
    model="gemini-2.5-flash",
    contents=safe.text,
).text

# before it reaches the user
out = guard.screen(reply, gate="output")

# needs GUARDEX_GROUNDING_ENABLED=1

def check(guard, reply, sources):
    out = guard.screen_grounded(
        response_text=reply,
        sources=sources,
        gate="output",
        grounding_mode="accuracy",
    )
    screen, ground = out

    if screen.blocked:
        return "I can't help with that."

    bad = ground.hallucinated_sentences
    for s in bad:
        flag_for_review(s.sentence)

import asyncio
from guardex import Guard, PIIVault

async def main():
    vault = PIIVault()
    async with Guard() as guard:
        r = await guard.ascreen(
            msg, gate="input",
        )
        # tokens survive chunk splits
        async for c in guard.astream(
            stream, vault=vault,
            restore_mode="buffered",
        ):
            send(c)

asyncio.run(main())