kodit 0.2.8__py3-none-any.whl → 0.3.0__py3-none-any.whl

kodit/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.2.8'
-__version_tuple__ = version_tuple = (0, 2, 8)
+__version__ = version = '0.3.0'
+__version_tuple__ = version_tuple = (0, 3, 0)

kodit/app.py

@@ -1,14 +1,49 @@
 """FastAPI application for kodit API."""
 
+from collections.abc import AsyncIterator
+from contextlib import asynccontextmanager
+
 from asgi_correlation_id import CorrelationIdMiddleware
 from fastapi import FastAPI
 
+from kodit.config import AppContext
+from kodit.infrastructure.indexing.auto_indexing_service import AutoIndexingService
 from kodit.mcp import mcp
 from kodit.middleware import ASGICancelledErrorMiddleware, logging_middleware
 
+# Global auto-indexing service
+_auto_indexing_service: AutoIndexingService | None = None
+
+
+@asynccontextmanager
+async def app_lifespan(_: FastAPI) -> AsyncIterator[None]:
+    """Manage application lifespan for auto-indexing."""
+    global _auto_indexing_service  # noqa: PLW0603
+    # Start auto-indexing service
+    app_context = AppContext()
+    db = await app_context.get_db()
+    _auto_indexing_service = AutoIndexingService(
+        app_context=app_context,
+        session_factory=db.session_factory,
+    )
+    await _auto_indexing_service.start_background_indexing()
+    yield
+    if _auto_indexing_service:
+        await _auto_indexing_service.stop()
+
+
 # See https://gofastmcp.com/deployment/asgi#fastapi-integration
 mcp_app = mcp.sse_app()
-app = FastAPI(title="kodit API", lifespan=mcp_app.router.lifespan_context)
+
+
+@asynccontextmanager
+async def combined_lifespan(app: FastAPI) -> AsyncIterator[None]:
+    """Combine app and MCP lifespans."""
+    async with app_lifespan(app), mcp_app.router.lifespan_context(app):
+        yield
+
+
+app = FastAPI(title="kodit API", lifespan=combined_lifespan)
 
 # Add middleware
 app.middleware("http")(logging_middleware)

kodit/application/factories/__init__.py

@@ -0,0 +1 @@
+"""Application factories package."""

kodit/application/factories/code_indexing_factory.py

@@ -0,0 +1,119 @@
+"""Factory for creating the unified code indexing application service."""
+
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from kodit.application.services.code_indexing_application_service import (
+    CodeIndexingApplicationService,
+)
+from kodit.config import AppContext
+from kodit.domain.entities import EmbeddingType
+from kodit.domain.services.bm25_service import BM25DomainService
+from kodit.domain.services.embedding_service import EmbeddingDomainService
+from kodit.domain.services.enrichment_service import EnrichmentDomainService
+from kodit.domain.services.source_service import SourceService
+from kodit.infrastructure.bm25.bm25_factory import bm25_repository_factory
+from kodit.infrastructure.embedding.embedding_factory import (
+    embedding_domain_service_factory,
+)
+from kodit.infrastructure.embedding.embedding_providers import (
+    hash_embedding_provider,
+)
+from kodit.infrastructure.embedding.local_vector_search_repository import (
+    LocalVectorSearchRepository,
+)
+from kodit.infrastructure.enrichment.enrichment_factory import (
+    enrichment_domain_service_factory,
+)
+from kodit.infrastructure.enrichment.null_enrichment_provider import (
+    NullEnrichmentProvider,
+)
+from kodit.infrastructure.indexing.indexing_factory import (
+    indexing_domain_service_factory,
+)
+from kodit.infrastructure.indexing.snippet_domain_service_factory import (
+    snippet_domain_service_factory,
+)
+from kodit.infrastructure.sqlalchemy.embedding_repository import (
+    SqlAlchemyEmbeddingRepository,
+)
+
+
+def create_code_indexing_application_service(
+    app_context: AppContext,
+    session: AsyncSession,
+    source_service: SourceService,
+) -> CodeIndexingApplicationService:
+    """Create a unified code indexing application service with all dependencies."""
+    # Create domain services
+    indexing_domain_service = indexing_domain_service_factory(session)
+    snippet_domain_service = snippet_domain_service_factory(session)
+    bm25_service = BM25DomainService(bm25_repository_factory(app_context, session))
+    code_search_service = embedding_domain_service_factory("code", app_context, session)
+    text_search_service = embedding_domain_service_factory("text", app_context, session)
+    enrichment_service = enrichment_domain_service_factory(app_context)
+
+    # Create and return the unified application service
+    return CodeIndexingApplicationService(
+        indexing_domain_service=indexing_domain_service,
+        snippet_domain_service=snippet_domain_service,
+        source_service=source_service,
+        bm25_service=bm25_service,
+        code_search_service=code_search_service,
+        text_search_service=text_search_service,
+        enrichment_service=enrichment_service,
+        session=session,
+    )
+
+
+def create_fast_test_code_indexing_application_service(
+    app_context: AppContext,
+    session: AsyncSession,
+    source_service: SourceService,
+) -> CodeIndexingApplicationService:
+    """Create a fast test version of CodeIndexingApplicationService."""
+    # Create domain services
+    indexing_domain_service = indexing_domain_service_factory(session)
+    snippet_domain_service = snippet_domain_service_factory(session)
+    bm25_service = BM25DomainService(bm25_repository_factory(app_context, session))
+
+    # Create fast embedding services using HashEmbeddingProvider
+    embedding_repository = SqlAlchemyEmbeddingRepository(session=session)
+
+    # Fast code search service
+    code_search_repository = LocalVectorSearchRepository(
+        embedding_repository=embedding_repository,
+        embedding_provider=hash_embedding_provider.HashEmbeddingProvider(),
+        embedding_type=EmbeddingType.CODE,
+    )
+    code_search_service = EmbeddingDomainService(
+        embedding_provider=hash_embedding_provider.HashEmbeddingProvider(),
+        vector_search_repository=code_search_repository,
+    )
+
+    # Fast text search service
+    text_search_repository = LocalVectorSearchRepository(
+        embedding_repository=embedding_repository,
+        embedding_provider=hash_embedding_provider.HashEmbeddingProvider(),
+        embedding_type=EmbeddingType.TEXT,
+    )
+    text_search_service = EmbeddingDomainService(
+        embedding_provider=hash_embedding_provider.HashEmbeddingProvider(),
+        vector_search_repository=text_search_repository,
+    )
+
+    # Fast enrichment service using NullEnrichmentProvider
+    enrichment_service = EnrichmentDomainService(
+        enrichment_provider=NullEnrichmentProvider()
+    )
+
+    # Create and return the unified application service
+    return CodeIndexingApplicationService(
+        indexing_domain_service=indexing_domain_service,
+        snippet_domain_service=snippet_domain_service,
+        source_service=source_service,
+        bm25_service=bm25_service,
+        code_search_service=code_search_service,
+        text_search_service=text_search_service,
+        enrichment_service=enrichment_service,
+        session=session,
+    )