Richie/dotfiles
1
0
Fork 0
Code Issues 1 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

feat(ingest): ensure session rollback on ingestion error and log mention count

Browse Source
This commit is contained in:
Richie
2026-07-16 13:50:35 -04:00
parent 13efd3de0d
commit 699b9322ee
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ebook_search/ingest.py
+2 -1
View File
@@ -159,13 +159,14 @@ async def ingest_file(session: AsyncSession, path: Path | AsyncPath, config: Ebo
await session.flush() await session.flush()
chunk_index = add_chapter_chunks(session, source, chapter, parsed_chapter, chunk_index, config) chunk_index = add_chapter_chunks(session, source, chapter, parsed_chapter, chunk_index, config)
await session.commit()
mention_count = await index_chunk_phrase_mentions_for_book(session, source.id, config) mention_count = await index_chunk_phrase_mentions_for_book(session, source.id, config)
await session.commit()
logger.info( logger.info(
f"ebook_ingest_file_complete {source.id=} {resolved_path=} chapters={len(parsed.chapters)} {chunk_index=} " f"ebook_ingest_file_complete {source.id=} {resolved_path=} chapters={len(parsed.chapters)} {chunk_index=} "
f"{mention_count=}" f"{mention_count=}"
) )
except Exception: except Exception:
await session.rollback()
logger.exception(f"ebook_ingest_file_error {path=}") logger.exception(f"ebook_ingest_file_error {path=}")
return False return False
else: else: