remove direct db call from ingestion service

Author: AJaccP

src/infrastructure/db/repository.py

@@ -1,6 +1,6 @@
 from uuid import UUID
 
-from sqlalchemy import func, select
+from sqlalchemy import func, select, update
 from sqlalchemy.dialects.postgresql import insert
 from sqlalchemy.ext.asyncio import AsyncSession
 
@@ -60,9 +60,9 @@ async def upsert_chunk(
 
     @staticmethod
     async def update_source_synced(session: AsyncSession, source_id: UUID) -> None:
-        result = await session.execute(select(SourceRow).where(SourceRow.id == source_id))
-        source = result.scalar_one()
-        source.last_synced_at = func.now()
+        await session.execute(
+            update(SourceRow).where(SourceRow.id == source_id).values(last_synced_at=func.now())
+        )
 
     @staticmethod
     async def top_k_chunks(

src/ingestion/services/ingestion_service.py

@@ -1,11 +1,9 @@
 import hashlib
 
 from langchain_text_splitters import RecursiveCharacterTextSplitter
-from sqlalchemy import func, update
 
 from src.config.logger import get_logger
 from src.infrastructure.db import async_session_factory
-from src.infrastructure.db.models import SourceRow
 from src.infrastructure.db.repository import Repository
 from src.ingestion.scrapers.html_scraper import scrape
 from src.retrieval.services import embedding_service
@@ -75,9 +73,7 @@ async def ingest_url(url: str) -> None:
                 source_id=source.id,
             )
 
-        await session.execute(
-            update(SourceRow).where(SourceRow.id == source.id).values(last_synced_at=func.now())
-        )
+        await Repository.update_source_synced(session, source.id)
         await session.commit()
 
     log.info("ingest_complete", url=url, n_chunks=len(chunks))