Add zero-trust agent, MCP, and reasoning security controls

Author: Vish Devarajan

src/blackwall_llm_shield/__init__.py

@@ -1,11 +1,19 @@
 from .core import (
     AuditTrail,
+    AgenticCapabilityGater,
+    AgentIdentityRegistry,
     BlackwallFastAPIMiddleware,
     BlackwallShield,
+    CoTScanner,
+    ImageMetadataScanner,
     LightweightIntentScorer,
+    MCPSecurityProxy,
     OutputFirewall,
     RetrievalSanitizer,
+    SessionBuffer,
+    TokenBudgetFirewall,
     ToolPermissionFirewall,
+    VisualInstructionDetector,
     POLICY_PACKS,
     build_admin_dashboard_model,
     create_fastapi_guard,
@@ -26,6 +34,7 @@
     mask_value,
     normalize_messages,
     run_red_team_suite,
+    rehydrate_response,
     sanitize_text,
     summarize_security_events,
     validate_grounding,
@@ -39,16 +48,24 @@
 
 __all__ = [
     "AuditTrail",
+    "AgenticCapabilityGater",
+    "AgentIdentityRegistry",
     "BlackwallFastAPIMiddleware",
     "BlackwallLangChainCallback",
     "BlackwallLlamaIndexCallback",
     "BlackwallMiddleware",
     "BlackwallShield",
+    "CoTScanner",
     "FastTextIntentScorer",
+    "ImageMetadataScanner",
     "LightweightIntentScorer",
+    "MCPSecurityProxy",
     "OutputFirewall",
     "RetrievalSanitizer",
+    "SessionBuffer",
+    "TokenBudgetFirewall",
     "ToolPermissionFirewall",
+    "VisualInstructionDetector",
     "POLICY_PACKS",
     "build_admin_dashboard_model",
     "create_fastapi_guard",
@@ -70,6 +87,7 @@
     "mask_value",
     "normalize_messages",
     "run_red_team_suite",
+    "rehydrate_response",
     "sanitize_text",
     "summarize_security_events",
     "validate_grounding",