$jndZddlmZmZddlmZerddlmZGddeZGddeZ d S) zXExtension context providing a controlled API for extensions to interact with the system.)ABCabstractmethod) TYPE_CHECKING)MemoryEngineInterfacecJeZdZdZededdfdZeddZdS) ExtensionContexta Abstract context providing a controlled API for extensions. Extensions receive this context instead of direct access to internal components like MemoryEngine or database connections. This provides: - A stable API that won't break when internals change - Security by limiting what extensions can access - Clear documentation of what extensions can do Built-in implementation: hindsight_api.extensions.builtin.context.DefaultExtensionContext Example usage in an extension: class MyTenantExtension(TenantExtension): async def on_startup(self) -> None: # Run migrations for a new tenant schema await self.context.run_migration("tenant_acme") class MyHttpExtension(HttpExtension): def get_router(self, memory): # Use memory engine for custom endpoints engine = self.context.get_memory_engine() ... schemareturnNc KdS)a2 Run database migrations for a specific schema. This creates the schema if it doesn't exist and runs all pending migrations. Uses advisory locks to coordinate between distributed workers. Args: schema: PostgreSQL schema name (e.g., "tenant_acme"). The schema will be created if it doesn't exist. Raises: RuntimeError: If migrations fail to complete. Example: # Provision a new tenant schema await context.run_migration("tenant_acme") N)selfr s h/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/hindsight_api/extensions/context.py run_migrationzExtensionContext.run_migration$s & rcdS)a Get the memory engine interface. Returns the MemoryEngineInterface for performing memory operations like retain, recall, reflect, and entity/document management. Returns: MemoryEngineInterface instance. Example: engine = context.get_memory_engine() result = await engine.recall_async(bank_id, query) Nr r s rget_memory_enginez"ExtensionContext.get_memory_engine9s  rr r)__name__ __module__ __qualname____doc__rstrrrr rrrr sl2 # $   ^ (   ^   rrc>eZdZdZ d deddfdZdeddfd Zd d ZdS)DefaultExtensionContextzw Default implementation of ExtensionContext. Uses the system's database URL and migration infrastructure. N database_url memory_enginezMemoryEngineInterface | Nonec"||_||_dS)z Initialize the context. Args: database_url: SQLAlchemy database URL for migrations. memory_engine: Optional MemoryEngine instance for memory operations. N) _database_url_memory_engine)r rrs r__init__z DefaultExtensionContext.__init__Rs*+rr r c Kddl}ddlm}ddlm}m}m}m}|j}|j t|j dd} | r| }|} | |||| j d{V|j Pt|j dd} | 8t| dd} | %| ||| || j d{V| ||| j | d{V| ||| j| d{VdS) z%Run migrations for a specific schema.rN) get_config)ensure_embedding_dimensionensure_text_search_extensionensure_vector_extensionrun_migrationsdb_url)r migration_database_url embeddings dimension)r vector_extension)r,r )text_search_extensionr )asynciohindsight_api.configr#hindsight_api.migrationsr$r%r&r'rr getattr to_threadr)r,r-) r r r.r#r$r%r&r'r( engine_urlconfigr*r+s rrz%DefaultExtensionContext.run_migrationas 333333            #   * !4hEEJ $# F6&Jg            * !4lDIIJ%#J TBB (!++2!%)/)@ , #Vf>U^d           (&Hdms            rrc<|jtd|jS)z Get the memory engine interface.NzpMemory engine not configured in ExtensionContext. Ensure the context was created with a memory_engine parameter.)r RuntimeErrorrs rrz)DefaultExtensionContext.get_memory_engines.   &Q ""r)Nr)rrrrrr!rrr rrrrKs9= , , ,6 , , , ,5 #5 $5 5 5 5 n######rrN) rabcrrtypingrhindsight_api.engine.interfacerrrr rrr:s^^######## EDDDDDD> > > > > s> > > BT#T#T#T#T#.T#T#T#T#T#r