Middleware Registry • AI Agent SDK

MiddlewareRegistry: Registry for runtime-scoped tool result middleware.

MiddlewareRegistry

Defined in the Middleware Registry module.

AI Agent

Registry for runtime-scoped tool result middleware.

Thread-safe singleton-like registry that manages middleware instances for different runtime environments. Plugin harnesses register their middleware here, and the core tool execution system uses it to normalize results before they reach hooks and memory adapters.

Methods

register()

unregister()

Unregister middleware for a runtime.

get_middleware()

Get middleware for a runtime.

has_middleware()

Check if middleware is registered for a runtime.

list_runtimes()

List all registered runtime IDs.

clear()

Clear all registered middleware.

Usage

python

# In plugin harness initialization
    registry = MiddlewareRegistry()
    registry.register("claude_harness", ClaudeHarnessMiddleware())

    # In core tool execution (agent.execute_tool)
    registry = MiddlewareRegistry()
    middleware = registry.get_middleware("claude_harness")
    normalized = middleware.normalize(result, tool_name, ctx)

Source

View on GitHub

praisonaiagents/runtime/middleware_registry.py at line 20

Middleware Registry • AI Agent SDK