iatoolkit 2.53.0__tar.gz → 2.53.1__tar.gz

{iatoolkit-2.53.0/src/iatoolkit.egg-info → iatoolkit-2.53.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 2.53.0
+Version: 2.53.1
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT

{iatoolkit-2.53.0 → iatoolkit-2.53.1}/src/iatoolkit/__init__.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-__version__ = "2.53.0"
+__version__ = "2.53.1"
 
 # Expose main classes and functions at the top level of the package
 

{iatoolkit-2.53.0 → iatoolkit-2.53.1}/src/iatoolkit/core.py

@@ -175,6 +175,13 @@ class IAToolkit:
                     logging.info("🔥 Startup warm-up completed for endpoints=%s", woken_endpoints)
             except Exception:
                 logging.exception("⚠️ Startup warm-up failed.")
+            finally:
+                # The scoped session is per thread: whatever this thread read
+                # (the endpoint's credentials, from the secrets table) keeps its
+                # connection "idle in transaction" until the thread's own session
+                # is removed - and nothing else ever removes it.
+                if self.db_manager is not None:
+                    self.db_manager.remove_session()
 
         thread = threading.Thread(target=_run_warmup, name="iatoolkit-startup-warmup", daemon=True)
         thread.start()

{iatoolkit-2.53.0 → iatoolkit-2.53.1}/src/iatoolkit/infra/llm_proxy.py

@@ -89,6 +89,23 @@ class LLMProxy:
                 self._telemetry_service = NoopTelemetryService()
         return self._telemetry_service
 
+    @staticmethod
+    def _release_database_transaction() -> None:
+        """
+        Hands the request's database connection back to the pool before the
+        provider call. Resolved lazily, like telemetry: the proxy is also built
+        outside a running toolkit (tests, scripts), where there is nothing to
+        release.
+        """
+        try:
+            from iatoolkit import current_iatoolkit
+            from iatoolkit.repositories.database_manager import DatabaseManager
+
+            db_manager = current_iatoolkit().get_injector().get(DatabaseManager)
+        except Exception:
+            return
+        db_manager.release_idle_transaction()
+
     # -------------------------------------------------------------------------
     # Public API
     # -------------------------------------------------------------------------
@@ -146,6 +163,10 @@ class LLMProxy:
         if telemetry_execution is not None:
             request_kwargs["telemetry_execution"] = telemetry_execution
 
+        # Everything the call needs from the database is resolved by now; the
+        # model can take minutes, and the connection should not wait with it.
+        self._release_database_transaction()
+
         # Delegate to the adapter (OpenAI, Gemini, DeepSeek, xAI, Anthropic, etc.)
         return adapter.create_response(
             model=self._wire_model(model, model_config),

{iatoolkit-2.53.0 → iatoolkit-2.53.1}/src/iatoolkit/locales/en.yaml

@@ -1247,6 +1247,13 @@ ui:
     run_duration_seconds: "Duration (s)"
     run_duration_minutes: "Duration (min)"
     duration_minutes_suffix: "mins."
+    timing_duration: "Duration"
+    timing_rate: "Rate"
+    timing_eta: "ETA"
+    timing_eta_after_discovery: "ETA after discovery"
+    timing_eta_pending: "ETA pending"
+    timing_processed_rate_suffix: "processed"
+    timing_submitted_rate_suffix: "submitted"
     run_finished: "Finished"
     run_actions: "Actions"
     run_discovered: "Discovered"

{iatoolkit-2.53.0 → iatoolkit-2.53.1}/src/iatoolkit/locales/es.yaml

@@ -1262,6 +1262,13 @@ ui:
     run_duration_seconds: "Duración (s)"
     run_duration_minutes: "Duración (min)"
     duration_minutes_suffix: "mins."
+    timing_duration: "Duración"
+    timing_rate: "Ritmo"
+    timing_eta: "ETA"
+    timing_eta_after_discovery: "ETA después de discovery"
+    timing_eta_pending: "ETA pendiente"
+    timing_processed_rate_suffix: "procesados"
+    timing_submitted_rate_suffix: "enviados"
     run_finished: "Fin"
     run_actions: "Acciones"
     run_discovered: "Descubiertos"

{iatoolkit-2.53.0 → iatoolkit-2.53.1}/src/iatoolkit/repositories/database_manager.py

@@ -324,6 +324,37 @@ class DatabaseManager(DatabaseProvider):
     def remove_session(self):
         self.scoped_session.remove()
 
+    def release_idle_transaction(self) -> bool:
+        """
+        Ends the current session's transaction and returns its connection to the
+        pool, before a call that waits on something other than the database.
+
+        Any read opens a transaction that lasts until the next commit, so a
+        session that read config or secrets and then waited minutes on an LLM
+        held its connection "idle in transaction" the whole time. A web process
+        serves many requests from one small pool, and a few such waits were
+        enough to time out every other request.
+
+        Commits rather than rolls back: a rollback expires every loaded object
+        and would discard writes already flushed. The session factory keeps
+        objects on commit, so callers go on using what they loaded. Leaves the
+        session alone when it has pending changes - committing them here would
+        be the caller's decision, not ours.
+        """
+        try:
+            if not self.scoped_session.registry.has():
+                return False
+            session = self.scoped_session.registry()
+            if not session.in_transaction():
+                return False
+            if session.new or session.dirty or session.deleted:
+                return False
+            session.commit()
+            return True
+        except Exception as e:
+            logging.debug("Could not release the idle database transaction: %s", e)
+            return False
+
     # -- execution methods ----
 
     def execute_query(self, query: str, commit: bool = False, params: dict | None = None) -> list[dict] | dict:

{iatoolkit-2.53.0 → iatoolkit-2.53.1/src/iatoolkit.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 2.53.0
+Version: 2.53.1
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT