Benjamin_Boenisch/breakpilot-compliance
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix stale DB session after long embedding pre-load

Browse Source 
The embedding pre-load for 4998 existing controls takes ~16 minutes,
causing the SQLAlchemy session to become invalid. Added rollback after
pre-load completes to reset the session before subsequent DB operations.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Benjamin Admin
2026-03-17 14:34:44 +01:00
parent 36ef34169a
commit 567e82ddf5
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backend-compliance/compliance/services/control_generator.py
View File

@@ -1391,6 +1391,12 @@ Gib ein JSON-Array zurueck mit GENAU {len(chunks)} Objekten. Jedes Objekt hat di
loaded = sum(1 for emb in embeddings if emb) loaded = sum(1 for emb in embeddings if emb)
logger.info("Pre-loaded %d/%d embeddings", loaded, len(texts)) logger.info("Pre-loaded %d/%d embeddings", loaded, len(texts))
# Reset DB session after long-running embedding operation to avoid stale connections
try:
self.db.rollback()
except Exception:
pass
def _load_existing_controls(self) -> list[dict]: def _load_existing_controls(self) -> list[dict]:
"""Load existing controls from DB (cached per pipeline run).""" """Load existing controls from DB (cached per pipeline run)."""
if self._existing_controls is not None: if self._existing_controls is not None: