kodit 0.4.1__py3-none-any.whl → 0.4.3__py3-none-any.whl

kodit/infrastructure/sqlalchemy/task_repository.py

@@ -1,5 +1,7 @@
 """Task repository for the task queue."""
 
+from collections.abc import Callable
+
 import structlog
 from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncSession
@@ -9,14 +11,23 @@ from kodit.domain.protocols import TaskRepository
 from kodit.domain.value_objects import TaskType
 from kodit.infrastructure.mappers.task_mapper import TaskMapper, TaskTypeMapper
 from kodit.infrastructure.sqlalchemy import entities as db_entities
+from kodit.infrastructure.sqlalchemy.unit_of_work import SqlAlchemyUnitOfWork
+
+
+def create_task_repository(
+    session_factory: Callable[[], AsyncSession],
+) -> TaskRepository:
+    """Create an index repository."""
+    uow = SqlAlchemyUnitOfWork(session_factory=session_factory)
+    return SqlAlchemyTaskRepository(uow)
 
 
 class SqlAlchemyTaskRepository(TaskRepository):
     """Repository for task persistence using the existing Task entity."""
 
-    def __init__(self, session: AsyncSession) -> None:
+    def __init__(self, uow: SqlAlchemyUnitOfWork) -> None:
         """Initialize the repository."""
-        self.session = session
+        self.uow = uow
         self.log = structlog.get_logger(__name__)
 
     async def add(
@@ -24,58 +35,63 @@ class SqlAlchemyTaskRepository(TaskRepository):
         task: Task,
     ) -> None:
         """Create a new task in the database."""
-        self.session.add(TaskMapper.from_domain_task(task))
+        async with self.uow:
+            self.uow.session.add(TaskMapper.from_domain_task(task))
 
     async def get(self, task_id: str) -> Task | None:
         """Get a task by ID."""
-        stmt = select(db_entities.Task).where(db_entities.Task.dedup_key == task_id)
-        result = await self.session.execute(stmt)
-        db_task = result.scalar_one_or_none()
-        if not db_task:
-            return None
-        return TaskMapper.to_domain_task(db_task)
+        async with self.uow:
+            stmt = select(db_entities.Task).where(db_entities.Task.dedup_key == task_id)
+            result = await self.uow.session.execute(stmt)
+            db_task = result.scalar_one_or_none()
+            if not db_task:
+                return None
+            return TaskMapper.to_domain_task(db_task)
 
     async def take(self) -> Task | None:
         """Take a task for processing and remove it from the database."""
-        stmt = (
-            select(db_entities.Task)
-            .order_by(db_entities.Task.priority.desc(), db_entities.Task.created_at)
-            .limit(1)
-        )
-        result = await self.session.execute(stmt)
-        db_task = result.scalar_one_or_none()
-        if not db_task:
-            return None
-        await self.session.delete(db_task)
-        return TaskMapper.to_domain_task(db_task)
+        async with self.uow:
+            stmt = (
+                select(db_entities.Task)
+                .order_by(db_entities.Task.priority.desc(), db_entities.Task.created_at)
+                .limit(1)
+            )
+            result = await self.uow.session.execute(stmt)
+            db_task = result.scalar_one_or_none()
+            if not db_task:
+                return None
+            await self.uow.session.delete(db_task)
+            return TaskMapper.to_domain_task(db_task)
 
     async def update(self, task: Task) -> None:
         """Update a task in the database."""
-        stmt = select(db_entities.Task).where(db_entities.Task.dedup_key == task.id)
-        result = await self.session.execute(stmt)
-        db_task = result.scalar_one_or_none()
+        async with self.uow:
+            stmt = select(db_entities.Task).where(db_entities.Task.dedup_key == task.id)
+            result = await self.uow.session.execute(stmt)
+            db_task = result.scalar_one_or_none()
 
-        if not db_task:
-            raise ValueError(f"Task not found: {task.id}")
+            if not db_task:
+                raise ValueError(f"Task not found: {task.id}")
 
-        db_task.priority = task.priority
-        db_task.payload = task.payload
+            db_task.priority = task.priority
+            db_task.payload = task.payload
 
     async def list(self, task_type: TaskType | None = None) -> list[Task]:
         """List tasks with optional status filter."""
-        stmt = select(db_entities.Task)
+        async with self.uow:
+            stmt = select(db_entities.Task)
 
-        if task_type:
-            stmt = stmt.where(
-                db_entities.Task.type == TaskTypeMapper.from_domain_type(task_type)
-            )
+            if task_type:
+                stmt = stmt.where(
+                    db_entities.Task.type == TaskTypeMapper.from_domain_type(task_type)
+                )
 
-        stmt = stmt.order_by(
-            db_entities.Task.priority.desc(), db_entities.Task.created_at
-        )
+            stmt = stmt.order_by(
+                db_entities.Task.priority.desc(), db_entities.Task.created_at
+            )
 
-        result = await self.session.execute(stmt)
-        records = result.scalars().all()
+            result = await self.uow.session.execute(stmt)
+            records = result.scalars().all()
 
-        # Convert to domain entities
-        return [TaskMapper.to_domain_task(record) for record in records]
+            # Convert to domain entities
+            return [TaskMapper.to_domain_task(record) for record in records]

kodit/infrastructure/sqlalchemy/task_status_repository.py

@@ -0,0 +1,79 @@
+"""Task repository for the task queue."""
+
+from collections.abc import Callable
+
+import structlog
+from sqlalchemy import delete, select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from kodit.domain import entities as domain_entities
+from kodit.domain.protocols import TaskStatusRepository
+from kodit.infrastructure.mappers.task_status_mapper import TaskStatusMapper
+from kodit.infrastructure.sqlalchemy import entities as db_entities
+from kodit.infrastructure.sqlalchemy.unit_of_work import SqlAlchemyUnitOfWork
+
+
+def create_task_status_repository(
+    session_factory: Callable[[], AsyncSession],
+) -> TaskStatusRepository:
+    """Create an index repository."""
+    uow = SqlAlchemyUnitOfWork(session_factory=session_factory)
+    return SqlAlchemyTaskStatusRepository(uow)
+
+
+class SqlAlchemyTaskStatusRepository(TaskStatusRepository):
+    """Repository for persisting TaskStatus entities."""
+
+    def __init__(self, uow: SqlAlchemyUnitOfWork) -> None:
+        """Initialize the repository."""
+        self.uow = uow
+        self.log = structlog.get_logger(__name__)
+        self.mapper = TaskStatusMapper()
+
+    async def save(self, status: domain_entities.TaskStatus) -> None:
+        """Save a TaskStatus to database."""
+        async with self.uow:
+            # Convert domain entity to database entity
+            db_status = self.mapper.from_domain_task_status(status)
+            stmt = select(db_entities.TaskStatus).where(
+                db_entities.TaskStatus.id == db_status.id,
+            )
+            result = await self.uow.session.execute(stmt)
+            existing = result.scalar_one_or_none()
+
+            if not existing:
+                self.uow.session.add(db_status)
+            else:
+                # Update existing record with new values
+                existing.operation = db_status.operation
+                existing.state = db_status.state
+                existing.error = db_status.error
+                existing.total = db_status.total
+                existing.current = db_status.current
+                existing.updated_at = db_status.updated_at
+                existing.parent = db_status.parent
+                existing.trackable_id = db_status.trackable_id
+                existing.trackable_type = db_status.trackable_type
+
+    async def load_with_hierarchy(
+        self, trackable_type: str, trackable_id: int
+    ) -> list[domain_entities.TaskStatus]:
+        """Load TaskStatus entities with hierarchy from database."""
+        async with self.uow:
+            stmt = select(db_entities.TaskStatus).where(
+                db_entities.TaskStatus.trackable_id == trackable_id,
+                db_entities.TaskStatus.trackable_type == trackable_type,
+            )
+            result = await self.uow.session.execute(stmt)
+            db_statuses = list(result.scalars().all())
+
+            # Use mapper to convert and reconstruct hierarchy
+            return self.mapper.to_domain_task_status_with_hierarchy(db_statuses)
+
+    async def delete(self, status: domain_entities.TaskStatus) -> None:
+        """Delete a TaskStatus."""
+        async with self.uow:
+            stmt = delete(db_entities.TaskStatus).where(
+                db_entities.TaskStatus.id == status.id,
+            )
+            await self.uow.session.execute(stmt)

kodit/infrastructure/sqlalchemy/unit_of_work.py

@@ -0,0 +1,59 @@
+"""SQLAlchemy implementation of Unit of Work pattern."""
+
+from collections.abc import Callable
+from types import TracebackType
+
+from sqlalchemy.ext.asyncio import AsyncSession
+
+
+class SqlAlchemyUnitOfWork:
+    """SQLAlchemy implementation of Unit of Work pattern."""
+
+    def __init__(self, session_factory: Callable[[], AsyncSession]) -> None:
+        """Initialize the unit of work with a session factory."""
+        self._session_factory = session_factory
+        self._session: AsyncSession | None = None
+
+    @property
+    def session(self) -> AsyncSession:
+        """Get the current session."""
+        if self._session is None:
+            raise RuntimeError("UnitOfWork must be used within async context")
+        return self._session
+
+    async def __aenter__(self) -> "SqlAlchemyUnitOfWork":
+        """Enter the unit of work context."""
+        self._session = self._session_factory()
+        return self
+
+    async def __aexit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        exc_tb: TracebackType | None,
+    ) -> None:
+        """Exit the unit of work context."""
+        if self._session:
+            if exc_type is not None:
+                await self._session.rollback()
+            await self._session.commit()
+            await self._session.close()
+            self._session = None
+
+    async def commit(self) -> None:
+        """Commit the current transaction."""
+        if self._session is None:
+            raise RuntimeError("UnitOfWork must be used within async context")
+        await self._session.commit()
+
+    async def rollback(self) -> None:
+        """Rollback the current transaction."""
+        if self._session is None:
+            raise RuntimeError("UnitOfWork must be used within async context")
+        await self._session.rollback()
+
+    async def flush(self) -> None:
+        """Flush pending changes to the database without committing."""
+        if self._session is None:
+            raise RuntimeError("UnitOfWork must be used within async context")
+        await self._session.flush()

kodit/mcp.py

@@ -1,6 +1,6 @@
 """MCP server for kodit."""
 
-from collections.abc import AsyncIterator
+from collections.abc import AsyncIterator, Callable
 from contextlib import asynccontextmanager
 from dataclasses import dataclass
 from pathlib import Path
@@ -15,6 +15,7 @@ from kodit._version import version
 from kodit.application.factories.code_indexing_factory import (
     create_code_indexing_application_service,
 )
+from kodit.application.factories.reporting_factory import create_server_operation
 from kodit.config import AppContext
 from kodit.database import Database
 from kodit.domain.value_objects import (
@@ -22,6 +23,9 @@ from kodit.domain.value_objects import (
     MultiSearchResult,
     SnippetSearchFilters,
 )
+from kodit.infrastructure.sqlalchemy.task_status_repository import (
+    create_task_status_repository,
+)
 
 # Global database connection for MCP server
 _mcp_db: Database | None = None
@@ -32,6 +36,7 @@ class MCPContext:
     """Context for the MCP server."""
 
     session: AsyncSession
+    session_factory: Callable[[], AsyncSession]
     app_context: AppContext
 
 
@@ -55,7 +60,11 @@ async def mcp_lifespan(_: FastMCP) -> AsyncIterator[MCPContext]:
     if _mcp_db is None:
         _mcp_db = await app_context.get_db()
     async with _mcp_db.session_factory() as session:
-        yield MCPContext(session=session, app_context=app_context)
+        yield MCPContext(
+            session=session,
+            app_context=app_context,
+            session_factory=_mcp_db.session_factory,
+        )
 
 
 def create_mcp_server(name: str, instructions: str | None = None) -> FastMCP:
@@ -173,7 +182,10 @@ def register_mcp_tools(mcp_server: FastMCP) -> None:
         # Use the unified application service
         service = create_code_indexing_application_service(
             app_context=mcp_context.app_context,
-            session=mcp_context.session,
+            session_factory=mcp_context.session_factory,
+            operation=create_server_operation(
+                create_task_status_repository(mcp_context.session_factory)
+            ),
         )
 
         log.debug("Searching for snippets")

kodit/migrations/env.py

@@ -1,4 +1,3 @@
-# ruff: noqa: F401
 """Alembic environment file for kodit."""
 
 import asyncio

kodit/migrations/versions/b9cd1c3fd762_add_task_status.py

@@ -0,0 +1,77 @@
+# ruff: noqa
+"""add task status
+
+Revision ID: b9cd1c3fd762
+Revises: 9cf0e87de578
+Create Date: 2025-09-05 13:41:29.645898
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "b9cd1c3fd762"
+down_revision: Union[str, None] = "9cf0e87de578"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "task_status",
+        sa.Column("id", sa.String(length=255), nullable=False),
+        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
+        sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
+        sa.Column("operation", sa.String(length=255), nullable=False),
+        sa.Column("trackable_id", sa.Integer(), nullable=True),
+        sa.Column("trackable_type", sa.String(length=255), nullable=True),
+        sa.Column("parent", sa.String(length=255), nullable=True),
+        sa.Column("message", sa.UnicodeText(), nullable=False),
+        sa.Column("state", sa.String(length=255), nullable=False),
+        sa.Column("error", sa.UnicodeText(), nullable=False),
+        sa.Column("total", sa.Integer(), nullable=False),
+        sa.Column("current", sa.Integer(), nullable=False),
+        sa.ForeignKeyConstraint(
+            ["parent"],
+            ["task_status.id"],
+        ),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index(op.f("ix_task_status_id"), "task_status", ["id"], unique=False)
+    op.create_index(
+        op.f("ix_task_status_operation"), "task_status", ["operation"], unique=False
+    )
+    op.create_index(
+        op.f("ix_task_status_parent"), "task_status", ["parent"], unique=False
+    )
+    op.create_index(
+        op.f("ix_task_status_trackable_id"),
+        "task_status",
+        ["trackable_id"],
+        unique=False,
+    )
+    op.create_index(
+        op.f("ix_task_status_trackable_type"),
+        "task_status",
+        ["trackable_type"],
+        unique=False,
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_index(op.f("ix_task_status_trackable_type"), table_name="task_status")
+    op.drop_index(op.f("ix_task_status_trackable_id"), table_name="task_status")
+    op.drop_index(op.f("ix_task_status_parent"), table_name="task_status")
+    op.drop_index(op.f("ix_task_status_operation"), table_name="task_status")
+    op.drop_index(op.f("ix_task_status_id"), table_name="task_status")
+    op.drop_table("task_status")
+    # ### end Alembic commands ###

{kodit-0.4.1.dist-info → kodit-0.4.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kodit
-Version: 0.4.1
+Version: 0.4.3
 Summary: Code indexing for better AI code generation
 Project-URL: Homepage, https://docs.helixml.tech/kodit/
 Project-URL: Documentation, https://docs.helixml.tech/kodit/

{kodit-0.4.1.dist-info → kodit-0.4.3.dist-info}/RECORD

@@ -1,36 +1,37 @@
 kodit/.gitignore,sha256=ztkjgRwL9Uud1OEi36hGQeDGk3OLK1NfDEO8YqGYy8o,11
 kodit/__init__.py,sha256=aEKHYninUq1yh6jaNfvJBYg-6fenpN132nJt1UU6Jxs,59
-kodit/_version.py,sha256=k7cu0JKra64gmMNU_UfA5sw2eNc_GRvf3QmesiYAy8g,704
-kodit/app.py,sha256=FKqHbJNoHpnBb5KLJaUINDm0a5cHpxlrQ5qNoWnHsBc,4326
-kodit/cli.py,sha256=VUZD4cPRgAnrKEWUl2PbS-nOA0FkDVqmJ2SR0g1yJsk,28202
+kodit/_version.py,sha256=bmI9ViMEsJ1Rjce-6ExwiNh2B7sZKTyBkze4k8NsTrU,704
+kodit/app.py,sha256=_AXw_ICyPXkRxXaFPEbeQdSsj5Ybq5z2IEA4XBngXkE,4637
+kodit/cli.py,sha256=1n4pGjnzlZCBhWOJayHo_zFixs1mqkhPP3EosoZRbfE,27385
 kodit/cli_utils.py,sha256=bW4rIm-elrsyM_pSGHh30zV0_oX7V-64pL3YSaBcOt0,2810
-kodit/config.py,sha256=EOyB3BfSDfGJeOts2vrOyX8lhGXbAHVUl69kzjqGHXE,10326
-kodit/database.py,sha256=kI9yBm4uunsgV4-QeVoCBL0wLzU4kYmYv5qZilGnbPE,1740
+kodit/config.py,sha256=7q0MR8wKqSeS7_qhswWZPOM0E7aTATvLSW2mkze9CC8,11058
+kodit/database.py,sha256=k93byjVUX1VjAb0hLZxUo4liEKKxAWUBJNw2e7rzaiI,1771
 kodit/log.py,sha256=ZpM0eMo_DVGQqrHxg0VV6dMrN2AAmu_3C0I3G7p2nMw,8828
-kodit/mcp.py,sha256=aEcPc8dQiZaR0AswCZZNxcm_rhhUZNsEBimYti0ibSI,7221
+kodit/mcp.py,sha256=5Qvme_dRW4qeK33av8_PfBKGrhaVx3vOVRs6alG2HP4,7708
 kodit/middleware.py,sha256=TiwebNpaEmiP7QRuZrfZcCL51IUefQyNLSPuzVyk8UM,2813
-kodit/reporting.py,sha256=icce1ZyiADsA_Qz-mSjgn2H4SSqKuGfLKnw-yrl9nsg,2722
 kodit/application/__init__.py,sha256=mH50wTpgP9dhbKztFsL8Dda9Hi18TSnMVxXtpp4aGOA,35
 kodit/application/factories/__init__.py,sha256=bU5CvEnaBePZ7JbkCOp1MGTNP752bnU2uEqmfy5FdRk,37
-kodit/application/factories/code_indexing_factory.py,sha256=R9f0wsj4-3NJFS5SEt_-OIGR_s_01gJXaL3PkZd8MlU,5911
+kodit/application/factories/code_indexing_factory.py,sha256=KfQnmJmMI3G9yamg3fd3tK0_taPGnqcscmHMFS9LBLw,7353
+kodit/application/factories/reporting_factory.py,sha256=6zZ6Rjh0WXmO0ehafeAOb2xe36bxnOuMBuIFr7DLgfw,1287
 kodit/application/services/__init__.py,sha256=p5UQNw-H5sxQvs5Etfte93B3cJ1kKW6DNxK34uFvU1E,38
-kodit/application/services/auto_indexing_service.py,sha256=O5BNR5HypgghzUFG4ykIWMl9mxHCUExnBmJuITIhECk,3457
-kodit/application/services/code_indexing_application_service.py,sha256=nrnd_Md-D0AfNKku7Aqt3YHDbXsBV9f44Z6XsjhiF3E,15877
-kodit/application/services/indexing_worker_service.py,sha256=Un4PytnWJU4uwROcxOMUFkt4cD7nmPezaBLsEHrMN6U,5185
-kodit/application/services/queue_service.py,sha256=vf_TEl76B0F0RSvfCeGDuM-QFzW-VUuj3zQaRmDPEYI,1921
-kodit/application/services/sync_scheduler.py,sha256=aLpEczZdTM8ubfAEY0Ajdh3MLfDcB9s-0ILZJrtIuZs,3504
+kodit/application/services/auto_indexing_service.py,sha256=TiA809X9TUTzRsh_uMCqPQZjq_HpznzN-A4ypDJhWAk,3740
+kodit/application/services/code_indexing_application_service.py,sha256=WzxmIsjrkQMeJViGU-v95AaWr5AKfL2vrWL98Jw7h1c,16708
+kodit/application/services/indexing_worker_service.py,sha256=z5fZuUk_iPfcCST0ywPqiEWZaQHhyRbxEGps8e1y2BY,5182
+kodit/application/services/queue_service.py,sha256=G42lR31maFRZ9cSvnWZrzeyb4P1R6yFqrcHWVKAqc9U,1924
+kodit/application/services/reporting.py,sha256=UkDa23ifz9eF8vxrD8L2T4-_sYtMTgKQUfZkiy3SN8s,3598
+kodit/application/services/sync_scheduler.py,sha256=FUUpDtxUh7Eg-lnzOrUHzmSWGpzdpYJQYgPhnQYwTcg,3446
 kodit/domain/__init__.py,sha256=TCpg4Xx-oF4mKV91lo4iXqMEfBT1OoRSYnbG-zVWolA,66
-kodit/domain/entities.py,sha256=QsCzKXT7gF9jTPAjJo5lqjFGRsIklAFC2qRy_Gt3RbA,10377
+kodit/domain/entities.py,sha256=pCvBSLK5hutZtiZ0OAcIW1WBswZNKQBh8BzZeuwGvyk,13353
 kodit/domain/errors.py,sha256=yIsgCjM_yOFIg8l7l-t7jM8pgeAX4cfPq0owf7iz3DA,106
-kodit/domain/interfaces.py,sha256=Jkd0Ob4qSvhZHI9jRPFQ1n5Cv0SvU-y3Z-HCw2ikc4I,742
-kodit/domain/protocols.py,sha256=GA0CCvmhvQ3F4MseeQUVw3NeIgUoaV7V_7TdAaU70Is,2587
-kodit/domain/value_objects.py,sha256=dkfbg99PSCrfj6nJ7tZ2UzDG3QUgNa_Cpj2gLakDM5k,17512
+kodit/domain/protocols.py,sha256=tKl7zzS3k5n4czN7TSwjJp0DauPz_IcZjJN3qgE2d_M,3317
+kodit/domain/value_objects.py,sha256=CQ3qS3g_zYghiY1pYrC6kVngSuJwctFSW3yKdOvVOpQ,18827
 kodit/domain/services/__init__.py,sha256=Q1GhCK_PqKHYwYE4tkwDz5BIyXkJngLBBOHhzvX8nzo,42
-kodit/domain/services/bm25_service.py,sha256=nsfTan3XtDwXuuAu1LUv-6Jukm6qFKVqqCVymjyepZQ,3625
+kodit/domain/services/bm25_service.py,sha256=seRo0V-zW6Uq-Y67j0-zp1xz93gbfQgvlEbpQeYHN1U,3529
 kodit/domain/services/embedding_service.py,sha256=7drYRC2kjg0WJmo06a2E9N0vDnwInUlBB96twjz2BT8,4526
 kodit/domain/services/enrichment_service.py,sha256=XsXg3nV-KN4rqtC7Zro_ZiZ6RSq-1eA1MG6IDzFGyBA,1316
 kodit/domain/services/index_query_service.py,sha256=cDQkgpJ3JbyeZ3z3GTIqH1JzhhKE_LBIwYE6b-lakwU,2172
-kodit/domain/services/index_service.py,sha256=uVwDUEQWfZ5yJRvcjaWW7P9gCZttmnlkI51IHz52eew,11554
+kodit/domain/services/index_service.py,sha256=4iROJC5RGKhf9P3SrvW9jjichGTQRqweVWskYF5ltac,10637
+kodit/domain/services/task_status_query_service.py,sha256=R2YPl42sdt8stUvPmIPxVG5WDcohT0yLESs_i0Y14lI,727
 kodit/infrastructure/__init__.py,sha256=HzEYIjoXnkz_i_MHO2e0sIVYweUcRnl2RpyBiTbMObU,28
 kodit/infrastructure/api/__init__.py,sha256=U0TSMPpHrlj1zbAtleuZjU3nXGwudyMe-veNBgvODwM,34
 kodit/infrastructure/api/client/__init__.py,sha256=6RSYqeuxjDe_zTUq48D0F-VfBBUvDmTkO3K3vD61q3I,349
@@ -42,9 +43,9 @@ kodit/infrastructure/api/client/search_client.py,sha256=f4mM5ZJpAuR7w-i9yASbh4SY
 kodit/infrastructure/api/middleware/__init__.py,sha256=6m7eE5k5buboJbuzyX5E9-Tf99yNwFaeJF0f_6HwLyM,30
 kodit/infrastructure/api/middleware/auth.py,sha256=QSnMcMLWvfumqN1iG4ePj2vEZb2Dlsgr-WHptkEkkhE,1064
 kodit/infrastructure/api/v1/__init__.py,sha256=hQ03es21FSgzQlmdP5xWZzK80woIvuYGjiZLwFYuYwk,151
-kodit/infrastructure/api/v1/dependencies.py,sha256=jaM000IfSnvU8uzwnC1cBZsfsMC-19jWFjObHfqBYuM,2475
+kodit/infrastructure/api/v1/dependencies.py,sha256=NPfRXe-Tsj4R4GPSrU1jOp6JTMr3E-0M19xG78pft2I,3457
 kodit/infrastructure/api/v1/routers/__init__.py,sha256=YYyeiuyphIPc-Q_2totF8zfR0BoseOH4ZYFdHP0ed_M,218
-kodit/infrastructure/api/v1/routers/indexes.py,sha256=_lUir1M0SW6kPHeGqjiPjtSa50rY4PN2es5TZEpSHYE,3442
+kodit/infrastructure/api/v1/routers/indexes.py,sha256=BslFWHw1DWH8rpm9fWOH1foQl62F22jZ8rNF7YGBIZE,4905
 kodit/infrastructure/api/v1/routers/queue.py,sha256=EZbR-G0qDO9W5ajV_75GRk2pW1Qdgc0ggOwrGKlBE2A,2138
 kodit/infrastructure/api/v1/routers/search.py,sha256=da9YTR6VTzU85_6X3aaZemdTHGCEvcPNeKuMFBgmT_A,2452
 kodit/infrastructure/api/v1/schemas/__init__.py,sha256=_5BVqv4EUi_vvWlAQOE_VfRulUDAF21ZQ7z27y7YOdw,498
@@ -52,6 +53,7 @@ kodit/infrastructure/api/v1/schemas/context.py,sha256=NlsIn9j1R3se7JkGZivS_CUN4g
 kodit/infrastructure/api/v1/schemas/index.py,sha256=NtL09YtO50h-ddpAFxNf-dyxu_Xi5v3yOpKW0W4xsAM,1950
 kodit/infrastructure/api/v1/schemas/queue.py,sha256=oa4wumWOvGzi53Q3cjwIrQJRoentp5nsQSsaj-l-B4U,652
 kodit/infrastructure/api/v1/schemas/search.py,sha256=CWzg5SIMUJ_4yM-ZfgSLWCanMxov6AyGgQQcOMkRlGw,5618
+kodit/infrastructure/api/v1/schemas/task_status.py,sha256=WVB7QdSEg1lKZh_H4sCe-zFdAZA3hCUoxvqCZmhlMXQ,1220
 kodit/infrastructure/bm25/__init__.py,sha256=DmGbrEO34FOJy4e685BbyxLA7gPW1eqs2gAxsp6JOuM,34
 kodit/infrastructure/bm25/bm25_factory.py,sha256=I4eo7qRslnyXIRkBf-StZ5ga2Evrr5J5YFocTChFD3g,884
 kodit/infrastructure/bm25/local_bm25_repository.py,sha256=qdDRiiap168XivjkukRHVj1fxlprqRIG0iRSaseLv5I,4695
@@ -59,45 +61,49 @@ kodit/infrastructure/bm25/vectorchord_bm25_repository.py,sha256=p6ht5K-jlDTvEkmo
 kodit/infrastructure/cloning/__init__.py,sha256=IzIvX-yeRRFZ-lfvPVSEe_qXszO6DGQdjKwwDigexyQ,30
 kodit/infrastructure/cloning/metadata.py,sha256=GD2UnCC1oR82RD0SVUqk9CJOqzXPxhOAHVOp7jqN6Qc,3571
 kodit/infrastructure/cloning/git/__init__.py,sha256=20ePcp0qE6BuLsjsv4KYB1DzKhMIMsPXwEqIEZtjTJs,34
-kodit/infrastructure/cloning/git/working_copy.py,sha256=qYcrR5qP1rhWZiYGMT1p-1Alavi_YvQLXx4MgIV7eXs,2611
+kodit/infrastructure/cloning/git/working_copy.py,sha256=-DPHW9YZ-wyCC-etkIurFurW-Ef-xIJfa05TCPAD4tc,3909
 kodit/infrastructure/embedding/__init__.py,sha256=F-8nLlWAerYJ0MOIA4tbXHLan8bW5rRR84vzxx6tRKI,39
-kodit/infrastructure/embedding/embedding_factory.py,sha256=wngBD2g6NniHDq_-KcYhhwSvmcMYyI8yIzoXvGQvu1U,3287
+kodit/infrastructure/embedding/embedding_factory.py,sha256=BNhrrYQAkcnXkuuQy-Q-lwJhyoGONsTsbgN4t0UdGeY,3395
 kodit/infrastructure/embedding/local_vector_search_repository.py,sha256=ExweyNEL5cP-g3eDhGqZSih7zhdOrop2WdFPPJL-tB4,3505
 kodit/infrastructure/embedding/vectorchord_vector_search_repository.py,sha256=PIoU0HsDlaoXDXnGjOR0LAkAcW4JiE3ymJy_SBhEopc,8030
 kodit/infrastructure/embedding/embedding_providers/__init__.py,sha256=qeZ-oAIAxMl5QqebGtO1lq-tHjl_ucAwOXePklcwwGk,34
 kodit/infrastructure/embedding/embedding_providers/batching.py,sha256=a8CL9PX2VLmbeg616fc_lQzfC4BWTVn32m4SEhXpHxc,3279
 kodit/infrastructure/embedding/embedding_providers/hash_embedding_provider.py,sha256=V6OdCuWyQQOvo3OJGRi-gBKDApIcrELydFg7T696P5s,2257
-kodit/infrastructure/embedding/embedding_providers/litellm_embedding_provider.py,sha256=5LCrPSQn3ZaLZ1XTKzJV_LzANH7FdaR4NL-gJupaiDE,5579
+kodit/infrastructure/embedding/embedding_providers/litellm_embedding_provider.py,sha256=9Q5he_MI8xXENODwCvYCbhVawTjTv1bArGQrmxoWLas,5297
 kodit/infrastructure/embedding/embedding_providers/local_embedding_provider.py,sha256=9aLV1Zg4KMhYWlGRwgAUtswW4aIabNqbsipWhAn64RI,4133
 kodit/infrastructure/enrichment/__init__.py,sha256=8acZKNzql8Fs0lceFu9U3KoUrOptRBtVIxr_Iw6lz3Y,40
 kodit/infrastructure/enrichment/enrichment_factory.py,sha256=NFGY6u9SJ_GOgiB_RtotbQmte0kGFQUymwzZCbbsx34,1530
 kodit/infrastructure/enrichment/litellm_enrichment_provider.py,sha256=AM4-4KApDndzWzQzzKAedy21iGMhkwylR5VCmV9K-uI,6040
-kodit/infrastructure/enrichment/local_enrichment_provider.py,sha256=aVU3_kbLJ0BihwGIwvJ00DBe0voHkiKdFSjPxxkVfVA,4150
+kodit/infrastructure/enrichment/local_enrichment_provider.py,sha256=6WVVt5Nlq8mDGSRCOTjjVPMDCUyMgW-cxCzZv3WfP7k,4589
 kodit/infrastructure/enrichment/null_enrichment_provider.py,sha256=DhZkJBnkvXg_XSAs-oKiFnKqYFPnmTl3ikdxrqeEfbc,713
 kodit/infrastructure/enrichment/utils.py,sha256=FE9UCuxxzSdoHrmAC8Si2b5D6Nf6kVqgM1yjUVyCvW0,930
 kodit/infrastructure/git/__init__.py,sha256=0iMosFzudj4_xNIMe2SRbV6l5bWqkjnUsZoFsoZFuM8,33
-kodit/infrastructure/git/git_utils.py,sha256=3Fg2ZX9pkp8Mk1mWuW30PSO_ZKXrPL7wTCS9TMTfIUM,765
+kodit/infrastructure/git/git_utils.py,sha256=5lH94AcF7Hac4h6kBzo_B9pzC1S6AK2-Dy13gz--Zf0,781
 kodit/infrastructure/ignore/__init__.py,sha256=VzFv8XOzHmsu0MEGnWVSF6KsgqLBmvHlRqAkT1Xb1MY,36
 kodit/infrastructure/ignore/ignore_pattern_provider.py,sha256=zdxun3GodLfXxyssBK8QDUK58xb4fBJ0SKcHUyn3pzM,2131
 kodit/infrastructure/indexing/__init__.py,sha256=7UPRa2jwCAsa0Orsp6PqXSF8iIXJVzXHMFmrKkI9yH8,38
 kodit/infrastructure/indexing/fusion_service.py,sha256=2B0guBsuKz19uWcs18sIJpUJPzXoRvULgl7UNWQGysA,1809
 kodit/infrastructure/mappers/__init__.py,sha256=QPHOjNreXmBPPovZ6elnYFS0vD-IsmrGl4TT01FCKro,77
-kodit/infrastructure/mappers/index_mapper.py,sha256=ZSfu8kjTaa8_UY0nTqr4b02NS3VrjqZYkduCN71AL2g,12743
+kodit/infrastructure/mappers/index_mapper.py,sha256=XWtv_him2Sd9dR-Jy_ndy9jYXVtv3LttzmmUGzNK6CE,12825
 kodit/infrastructure/mappers/task_mapper.py,sha256=QW7uL8rji6QJ7RRdHwbvkWqmwDcUDGTYPLwbwiKlViY,2919
+kodit/infrastructure/mappers/task_status_mapper.py,sha256=5lo_jS2EKYg4czOOAFmrWfQl3OojIokwpzhaR3b1wE0,3433
+kodit/infrastructure/reporting/__init__.py,sha256=4Qu38YbDOaeDqLdT_CbK8tOZHTKGrHRXncVKlGRzOeQ,32
+kodit/infrastructure/reporting/db_progress.py,sha256=VVaCKjC_UFwdRptXbBroG9qhXCxI4bZmElf1PMsBzWA,819
+kodit/infrastructure/reporting/log_progress.py,sha256=dXy-xwIFLBnVTDPJd_h6e9iszmpQHvUUXAiVMLkOCWQ,1202
+kodit/infrastructure/reporting/tdqm_progress.py,sha256=-7ZrxTcb1L1tNiBGb5W_HKQ_uBNyMdeuelS3ZW3Wl3E,1288
 kodit/infrastructure/slicing/__init__.py,sha256=x7cjvHA9Ay2weUYE_dpdAaPaStp20M-4U2b5MLgT5KM,37
 kodit/infrastructure/slicing/language_detection_service.py,sha256=JGJXrq9bLyfnisWJXeP7y1jbZMmKAISdPBlRBCosUcE,684
 kodit/infrastructure/slicing/slicer.py,sha256=GOqJykd00waOTO1WJHyE5KUgJ2RLx2rOQ7M7T_u5LLg,35600
 kodit/infrastructure/sqlalchemy/__init__.py,sha256=UXPMSF_hgWaqr86cawRVqM8XdVNumQyyK5B8B97GnlA,33
-kodit/infrastructure/sqlalchemy/embedding_repository.py,sha256=dC2Wzj_zQiWExwfScE1LAGiiyxPyg0YepwyLOgDwcs4,7905
-kodit/infrastructure/sqlalchemy/entities.py,sha256=2iG_2NoKS26rJXimLFL8whRqFsUvKNGFAXCkQYK5GtE,6951
-kodit/infrastructure/sqlalchemy/index_repository.py,sha256=QQNsyLBI09YLUPLguB9qvqPZMxtg1p2twpm7sO_gNlo,23598
-kodit/infrastructure/sqlalchemy/task_repository.py,sha256=yazMO6Kw0Pb2b3L8wlGKOFA0QuMFcWBUXYFGZdtZo0w,2874
-kodit/infrastructure/ui/__init__.py,sha256=CzbLOBwIZ6B6iAHEd1L8cIBydCj-n_kobxJAhz2I9_Y,32
-kodit/infrastructure/ui/progress.py,sha256=SHEUoQA_x36z4nqHrQduVrrWIvFfX6QxAawC7zQ50pw,6433
-kodit/infrastructure/ui/spinner.py,sha256=GcP115qtR0VEnGfMEtsGoAUpRzVGUSfiUXfoJJERngA,2357
+kodit/infrastructure/sqlalchemy/embedding_repository.py,sha256=YYxbUEdzDdlKdy0FyAP4EzhJMAIdEnNZiXT6hzPHk9I,7731
+kodit/infrastructure/sqlalchemy/entities.py,sha256=QTtLlWFeFH8R3y4RWPESXk5IYMlVwAAAaJJsmg5OacY,9978
+kodit/infrastructure/sqlalchemy/index_repository.py,sha256=x8MPl0j7GrW_lEZh464EZyb0w935p_EHv2NIMNxjJu0,25680
+kodit/infrastructure/sqlalchemy/task_repository.py,sha256=60ECbxiXC2_UR80f4uPSmJiP_so7PTBzZG_w1WXSiuE,3546
+kodit/infrastructure/sqlalchemy/task_status_repository.py,sha256=xZm4j7V0NUD3OFj4o1Mwv1K8ngSoqWKlgFBgrVlWBxo,3283
+kodit/infrastructure/sqlalchemy/unit_of_work.py,sha256=gK-C8yk2HYBrAEDrblWxBrldrGb83SBHn-8lURkFeMg,2093
 kodit/migrations/README,sha256=ISVtAOvqvKk_5ThM5ioJE-lMkvf9IbknFUFVU_vPma4,58
 kodit/migrations/__init__.py,sha256=lP5MuwlyWRMO6UcDWnQcQ3G-GYHcFb6rl9gYPHJ1sjo,40
-kodit/migrations/env.py,sha256=m57TkFLYjQ4w2aw1YICXkeek27M6qjwRDMHvThWqIL0,2383
+kodit/migrations/env.py,sha256=asxYDVHeIPjM43SkF6b0raVOe-xKykyJTtjxXiSj4Bo,2364
 kodit/migrations/script.py.mako,sha256=zWziKtiwYKEWuwPV_HBNHwa9LCT45_bi01-uSNFaOOE,703
 kodit/migrations/versions/4073b33f9436_add_file_processing_flag.py,sha256=c8dMcKQ-BBBr_2-92eJZFS3Fwe3__B2eNqvQeMZHs0w,917
 kodit/migrations/versions/4552eb3f23ce_add_summary.py,sha256=WjyBQzFK8IXuvta15YBE23yaTMM1rZCXvPxW98MStng,870
@@ -106,13 +112,14 @@ kodit/migrations/versions/85155663351e_initial.py,sha256=h8DWmSxVwTtWlmWNH8-S4Ax
 kodit/migrations/versions/9cf0e87de578_add_queue.py,sha256=FYrco38f3_-1ZRfGOU617bgR1Ta0YTwwz3QDQMw_NKY,1600
 kodit/migrations/versions/9e53ea8bb3b0_add_authors.py,sha256=a32Zm8KUQyiiLkjKNPYdaJDgjW6VsV-GhaLnPnK_fpI,3884
 kodit/migrations/versions/__init__.py,sha256=9-lHzptItTzq_fomdIRBegQNm4Znx6pVjwD4MiqRIdo,36
+kodit/migrations/versions/b9cd1c3fd762_add_task_status.py,sha256=7cgdMKu4OBidL0ER8AGgyerfSfEk1I3XHo6Kh0sCzIQ,2754
 kodit/migrations/versions/c3f5137d30f5_index_all_the_things.py,sha256=r7ukmJ_axXLAWewYx-F1fEmZ4JbtFd37i7cSb0tq3y0,1722
 kodit/utils/__init__.py,sha256=DPEB1i8evnLF4Ns3huuAYg-0pKBFKUFuiDzOKG9r-sw,33
 kodit/utils/dump_openapi.py,sha256=29VdjHpNSaGAg7RjQw0meq1OLhljCx1ElgBlTC8xoF4,1247
 kodit/utils/generate_api_paths.py,sha256=TMtx9v55podDfUmiWaHgJHLtEWLV2sLL-5ejGFMPzAo,3569
 kodit/utils/path_utils.py,sha256=thK6YGGNvQThdBaCYCCeCvS1L8x-lwl3AoGht2jnjGw,1645
-kodit-0.4.1.dist-info/METADATA,sha256=Mf4UuPg2D08hfp1STtsZ1DwOHA2a1J4ba5_-T7Pifr4,7702
-kodit-0.4.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-kodit-0.4.1.dist-info/entry_points.txt,sha256=hoTn-1aKyTItjnY91fnO-rV5uaWQLQ-Vi7V5et2IbHY,40
-kodit-0.4.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-kodit-0.4.1.dist-info/RECORD,,
+kodit-0.4.3.dist-info/METADATA,sha256=HGyP_rA98DrqrsLJWOvUmRkbmRrrYd-p9czPjiF38aI,7702
+kodit-0.4.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+kodit-0.4.3.dist-info/entry_points.txt,sha256=hoTn-1aKyTItjnY91fnO-rV5uaWQLQ-Vi7V5et2IbHY,40
+kodit-0.4.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+kodit-0.4.3.dist-info/RECORD,,

kodit/domain/interfaces.py

@@ -1,27 +0,0 @@
-"""Domain interfaces."""
-
-from abc import ABC, abstractmethod
-
-from kodit.domain.value_objects import ProgressEvent
-
-
-class ProgressCallback(ABC):
-    """Abstract interface for progress callbacks."""
-
-    @abstractmethod
-    async def on_progress(self, event: ProgressEvent) -> None:
-        """On progress hook."""
-
-    @abstractmethod
-    async def on_complete(self, operation: str) -> None:
-        """On complete hook."""
-
-
-class NullProgressCallback(ProgressCallback):
-    """Null implementation of progress callback that does nothing."""
-
-    async def on_progress(self, event: ProgressEvent) -> None:
-        """Do nothing on progress."""
-
-    async def on_complete(self, operation: str) -> None:
-        """Do nothing on complete."""

kodit/infrastructure/ui/__init__.py

@@ -1 +0,0 @@
-"""UI infrastructure module."""