$j5 NdZddlmZddlmZerddlmZGddeZdS)z2Base Extension class for all Hindsight extensions.)ABC) TYPE_CHECKING)ExtensionContextc^eZdZdZdeeeffdZd dZed d Z dd Z dd Z dS) Extensiona Base class for all Hindsight extensions. Extensions are loaded via environment variables and receive configuration from prefixed environment variables. Example: HINDSIGHT_API_MY_EXTENSION=mypackage.ext:MyExtension HINDSIGHT_API_MY_SOME_CONFIG=value The extension receives: {"some_config": "value"} Extensions also receive an ExtensionContext that provides a controlled API for interacting with the system (e.g., running migrations for tenant schemas). configc"||_d|_dS)z Initialize the extension with configuration. Args: config: Dictionary of configuration values from environment variables. Keys are lowercased with the prefix stripped. N)r_context)selfrs e/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/hindsight_api/extensions/base.py__init__zExtension.__init__s 37 contextrreturnNc||_dS)z Set the extension context. Called by the extension loader after instantiation. Extensions should not call this directly. Args: context: The ExtensionContext providing system APIs. N)r )r rs r set_contextzExtension.set_context&s  rc<|jtd|jS)z Get the extension context. Returns: The ExtensionContext providing system APIs. Raises: RuntimeError: If context has not been set yet. Nz\Extension context not set. Context is available after the extension is loaded by the system.)r RuntimeErrorr s r rzExtension.context2s+ = n }rc KdS)z Called when the application starts. Override to perform initialization tasks like connecting to external services. Nrs r on_startupzExtension.on_startupC rc KdS)z~ Called when the application shuts down. Override to perform cleanup tasks like closing connections. Nrrs r on_shutdownzExtension.on_shutdownKrr)rrrN)rr)rN) __name__ __module__ __qualname____doc__dictstrr rpropertyrrrrrr rr s 8tCH~ 8 8 8 8     X           rrN)rabcrtypingr hindsight_api.extensions.contextrrrrr r&s88 BAAAAAAG G G G G G G G G G r