norm_toolkit 1.0.1__tar.gz → 1.0.2__tar.gz

{norm_toolkit-1.0.1 → norm_toolkit-1.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: norm_toolkit
-Version: 1.0.1
+Version: 1.0.2
 Summary: Toolkit to normalize text to UMLS / ontologies
 Author: Haydn Jones
 Author-email: Haydn Jones <haydnjonest@gmail.com>

{norm_toolkit-1.0.1 → norm_toolkit-1.0.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "norm_toolkit"
-version = "1.0.1"
+version = "1.0.2"
 description = "Toolkit to normalize text to UMLS / ontologies"
 readme = "README.md"
 authors = [{ name = "Haydn Jones", email = "haydnjonest@gmail.com" }]

{norm_toolkit-1.0.1 → norm_toolkit-1.0.2}/src/norm_toolkit/normalizer_postgres.py

@@ -10,6 +10,7 @@ from __future__ import annotations
 import asyncio
 import json
 from collections.abc import Mapping, Sequence
+from typing import Any
 
 import asyncpg
 import polars as pl
@@ -42,13 +43,20 @@ class PostgresNormalizer:
     Uses VALUES clauses instead of temp tables for efficiency with small batches.
     """
 
-    def __init__(self, pool: asyncpg.Pool, schema: str = "public") -> None:
+    def __init__(
+        self,
+        pool: asyncpg.Pool,
+        schema: str = "public",
+        owned_resource: Any | None = None,
+    ) -> None:
         """
         Initialize the normalizer with an external connection pool.
 
         Args:
             pool: asyncpg connection pool (caller manages lifecycle)
             schema: PostgreSQL schema where tables are located (default: "public")
+            owned_resource: Optional resource with async close() method to clean up
+                when this normalizer is closed (e.g., AlloyDB AsyncConnector)
 
         Note:
             After creating the normalizer, call `await normalizer.initialize()`
@@ -57,6 +65,7 @@ class PostgresNormalizer:
         self._pool = pool
         self._schema = schema
         self._loop: asyncio.AbstractEventLoop | None = None
+        self._owned_resource = owned_resource
         self._has_types = False
         self._has_defs = False
         self._has_edges = False
@@ -820,12 +829,14 @@ WHERE concept_id != $1
 
     async def close(self) -> None:
         """
-        Close the connection pool.
+        Close the connection pool and any owned resources.
 
         Note: Only call this if you want to close the pool. If the pool
         is managed externally, the caller should close it instead.
         """
         await self._pool.close()
+        if self._owned_resource is not None:
+            await self._owned_resource.close()
 
     def close_sync(self) -> None:
         """