nat.middleware.defense.register#

Registration module for defense middleware.

Functions#

content_safety_guard_middleware(...)

Build a Content Safety Guard middleware from configuration.

output_verifier_middleware(...)

Build an Output Verifier middleware from configuration.

pii_defense_middleware(...)

Build a PII Defense middleware from configuration.

Module Contents#

async content_safety_guard_middleware(
config: nat.middleware.defense.defense_middleware_content_guard.ContentSafetyGuardMiddlewareConfig,
builder: nat.builder.builder.Builder,
) collections.abc.AsyncGenerator[nat.middleware.defense.defense_middleware_content_guard.ContentSafetyGuardMiddleware, None]#

Build a Content Safety Guard middleware from configuration.

Args:

config: The content safety guard middleware configuration builder: The workflow builder used to resolve the LLM

Yields:

A configured Content Safety Guard middleware instance

async output_verifier_middleware(
config: nat.middleware.defense.defense_middleware_output_verifier.OutputVerifierMiddlewareConfig,
builder: nat.builder.builder.Builder,
) collections.abc.AsyncGenerator[nat.middleware.defense.defense_middleware_output_verifier.OutputVerifierMiddleware, None]#

Build an Output Verifier middleware from configuration.

Args:

config: The Output Verifier middleware configuration builder: The workflow builder used to resolve the LLM

Yields:

A configured Output Verifier middleware instance

async pii_defense_middleware(
config: nat.middleware.defense.defense_middleware_pii.PIIDefenseMiddlewareConfig,
builder: nat.builder.builder.Builder,
) collections.abc.AsyncGenerator[nat.middleware.defense.defense_middleware_pii.PIIDefenseMiddleware, None]#

Build a PII Defense middleware from configuration.

Args:

config: The PII Defense middleware configuration builder: The workflow builder (not used for PII defense)

Yields:

A configured PII Defense middleware instance