kodit 0.5.1__py3-none-any.whl → 0.5.2__py3-none-any.whl

kodit/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.5.1'
-__version_tuple__ = version_tuple = (0, 5, 1)
+__version__ = version = '0.5.2'
+__version_tuple__ = version_tuple = (0, 5, 2)
 
 __commit_id__ = commit_id = None

kodit/application/factories/server_factory.py

@@ -12,6 +12,9 @@ from kodit.application.services.code_search_application_service import (
 from kodit.application.services.commit_indexing_application_service import (
     CommitIndexingApplicationService,
 )
+from kodit.application.services.enrichment_query_service import (
+    EnrichmentQueryService,
+)
 from kodit.application.services.queue_service import QueueService
 from kodit.application.services.reporting import ProgressTracker
 from kodit.application.services.sync_scheduler import SyncSchedulerService
@@ -39,6 +42,7 @@ from kodit.domain.services.git_repository_service import (
 from kodit.domain.services.physical_architecture_service import (
     PhysicalArchitectureService,
 )
+from kodit.domain.tracking.resolution_service import TrackableResolutionService
 from kodit.infrastructure.bm25.local_bm25_repository import LocalBM25Repository
 from kodit.infrastructure.bm25.vectorchord_bm25_repository import (
     VectorChordBM25Repository,
@@ -127,6 +131,8 @@ class ServerFactory:
         self._architecture_service: PhysicalArchitectureService | None = None
         self._enrichment_v2_repository: EnrichmentV2Repository | None = None
         self._architecture_formatter: PhysicalArchitectureFormatter | None = None
+        self._trackable_resolution_service: TrackableResolutionService | None = None
+        self._enrichment_query_service: EnrichmentQueryService | None = None
 
     def architecture_formatter(self) -> PhysicalArchitectureFormatter:
         """Create a PhysicalArchitectureFormatter instance."""
@@ -351,3 +357,22 @@ class ServerFactory:
                 session_factory=self.session_factory
             )
         return self._git_tag_repository
+
+    def trackable_resolution_service(self) -> TrackableResolutionService:
+        """Create a TrackableResolutionService instance."""
+        if not self._trackable_resolution_service:
+            self._trackable_resolution_service = TrackableResolutionService(
+                commit_repo=self.git_commit_repository(),
+                branch_repo=self.git_branch_repository(),
+                tag_repo=self.git_tag_repository(),
+            )
+        return self._trackable_resolution_service
+
+    def enrichment_query_service(self) -> EnrichmentQueryService:
+        """Create a EnrichmentQueryService instance."""
+        if not self._enrichment_query_service:
+            self._enrichment_query_service = EnrichmentQueryService(
+                trackable_resolution=self.trackable_resolution_service(),
+                enrichment_repo=self.enrichment_v2_repository(),
+            )
+        return self._enrichment_query_service

kodit/application/services/enrichment_query_service.py

@@ -0,0 +1,95 @@
+"""Application service for querying enrichments."""
+
+import structlog
+
+from kodit.domain.enrichments.enrichment import EnrichmentV2
+from kodit.domain.tracking.resolution_service import TrackableResolutionService
+from kodit.domain.tracking.trackable import Trackable
+from kodit.infrastructure.sqlalchemy.enrichment_v2_repository import (
+    EnrichmentV2Repository,
+)
+
+
+class EnrichmentQueryService:
+    """Finds the latest commit with enrichments for a trackable.
+
+    Orchestrates domain services and repositories to fulfill the use case.
+    """
+
+    def __init__(
+        self,
+        trackable_resolution: TrackableResolutionService,
+        enrichment_repo: EnrichmentV2Repository,
+    ) -> None:
+        """Initialize the enrichment query service."""
+        self.trackable_resolution = trackable_resolution
+        self.enrichment_repo = enrichment_repo
+        self.log = structlog.get_logger(__name__)
+
+    async def find_latest_enriched_commit(
+        self,
+        trackable: Trackable,
+        enrichment_type: str | None = None,
+        max_commits_to_check: int = 100,
+    ) -> str | None:
+        """Find the most recent commit with enrichments.
+
+        Args:
+            trackable: What to track (branch, tag, or commit)
+            enrichment_type: Optional filter for specific enrichment type
+            max_commits_to_check: How far back in history to search
+
+        Returns:
+            Commit SHA of the most recent commit with enrichments, or None
+
+        """
+        # Get candidate commits from the trackable
+        candidate_commits = await self.trackable_resolution.resolve_to_commits(
+            trackable, max_commits_to_check
+        )
+
+        if not candidate_commits:
+            return None
+
+        # Check which commits have enrichments
+        enrichments = await self.enrichment_repo.enrichments_for_entity_type(
+            entity_type="git_commit",
+            entity_ids=candidate_commits,
+        )
+
+        # Filter by type if specified
+        if enrichment_type:
+            enrichments = [e for e in enrichments if e.type == enrichment_type]
+
+        # Find the first commit (newest) that has enrichments
+        for commit_sha in candidate_commits:
+            if any(e.entity_id == commit_sha for e in enrichments):
+                return commit_sha
+
+        return None
+
+    async def get_enrichments_for_commit(
+        self,
+        commit_sha: str,
+        enrichment_type: str | None = None,
+    ) -> list[EnrichmentV2]:
+        """Get all enrichments for a specific commit.
+
+        Args:
+            commit_sha: The commit SHA to get enrichments for
+            enrichment_type: Optional filter for specific enrichment type
+
+        Returns:
+            List of enrichments for the commit
+
+        """
+        enrichments = await self.enrichment_repo.enrichments_for_entity_type(
+            entity_type="git_commit",
+            entity_ids=[commit_sha],
+        )
+
+        # Filter by type if specified
+        if enrichment_type:
+            enrichments = [e for e in enrichments if e.type == enrichment_type]
+
+        return enrichments

kodit/domain/tracking/__init__.py

@@ -0,0 +1 @@
+"""Tracking domain module."""

kodit/domain/tracking/resolution_service.py

@@ -0,0 +1,81 @@
+"""Domain service for resolving trackables to commits."""
+
+import structlog
+
+from kodit.domain.protocols import (
+    GitBranchRepository,
+    GitCommitRepository,
+    GitTagRepository,
+)
+from kodit.domain.tracking.trackable import Trackable, TrackableReferenceType
+
+
+class TrackableResolutionService:
+    """Resolves trackables to ordered lists of commits.
+
+    This is a domain service because it orchestrates multiple aggregates
+    (branches, tags, commits) without belonging to any single entity.
+    """
+
+    def __init__(
+        self,
+        commit_repo: GitCommitRepository,
+        branch_repo: GitBranchRepository,
+        tag_repo: GitTagRepository,
+    ) -> None:
+        """Initialize the trackable resolution service."""
+        self.commit_repo = commit_repo
+        self.branch_repo = branch_repo
+        self.tag_repo = tag_repo
+        self.log = structlog.get_logger(__name__)
+
+    async def resolve_to_commits(
+        self, trackable: Trackable, limit: int = 100
+    ) -> list[str]:
+        """Resolve a trackable to an ordered list of commit SHAs.
+
+        Returns commits from newest to oldest based on git history.
+        """
+        if trackable.type == TrackableReferenceType.BRANCH:
+            return await self._resolve_branch(trackable, limit)
+        if trackable.type == TrackableReferenceType.TAG:
+            return await self._resolve_tag(trackable, limit)
+        # COMMIT_SHA
+        return [trackable.identifier]
+
+    async def _resolve_branch(
+        self, trackable: Trackable, limit: int
+    ) -> list[str]:
+        """Get commits from branch HEAD backwards through history."""
+        branch = await self.branch_repo.get_by_name(
+            trackable.identifier, trackable.repo_id
+        )
+        # Walk commit history from head_commit backwards
+        return await self._walk_commit_history(
+            branch.head_commit.commit_sha, limit
+        )
+
+    async def _resolve_tag(self, trackable: Trackable, limit: int) -> list[str]:
+        """Get commits from tag target backwards through history."""
+        tag = await self.tag_repo.get_by_name(
+            trackable.identifier, trackable.repo_id
+        )
+        return await self._walk_commit_history(
+            tag.target_commit.commit_sha, limit
+        )
+
+    async def _walk_commit_history(
+        self, start_sha: str, limit: int
+    ) -> list[str]:
+        """Walk commit history backwards from start_sha."""
+        result = []
+        current_sha: str | None = start_sha
+
+        for _ in range(limit):
+            if not current_sha:
+                break
+            result.append(current_sha)
+            commit = await self.commit_repo.get_by_sha(current_sha)
+            current_sha = commit.parent_commit_sha or None
+
+        return result

kodit/domain/tracking/trackable.py

@@ -0,0 +1,21 @@
+"""Trackable value objects."""
+
+from dataclasses import dataclass
+from enum import StrEnum
+
+
+class TrackableReferenceType(StrEnum):
+    """Types of git references that can be tracked."""
+
+    BRANCH = "branch"
+    TAG = "tag"
+    COMMIT_SHA = "commit_sha"
+
+
+@dataclass(frozen=True)
+class Trackable:
+    """Represents a trackable reference point in a git repository."""
+
+    type: TrackableReferenceType
+    identifier: str  # e.g., "main", "v1.0.0", "abc123..."
+    repo_id: int

kodit/infrastructure/api/v1/dependencies.py

@@ -13,6 +13,9 @@ from kodit.application.services.code_search_application_service import (
 from kodit.application.services.commit_indexing_application_service import (
     CommitIndexingApplicationService,
 )
+from kodit.application.services.enrichment_query_service import (
+    EnrichmentQueryService,
+)
 from kodit.application.services.queue_service import QueueService
 from kodit.config import AppContext
 from kodit.domain.protocols import (
@@ -155,3 +158,15 @@ async def get_code_search_app_service(
 CodeSearchAppServiceDep = Annotated[
     CodeSearchApplicationService, Depends(get_code_search_app_service)
 ]
+
+
+async def get_enrichment_query_service(
+    server_factory: ServerFactoryDep,
+) -> EnrichmentQueryService:
+    """Get enrichment query service dependency."""
+    return server_factory.enrichment_query_service()
+
+
+EnrichmentQueryServiceDep = Annotated[
+    EnrichmentQueryService, Depends(get_enrichment_query_service)
+]

kodit/infrastructure/api/v1/routers/repositories.py

@@ -2,15 +2,22 @@
 
 from fastapi import APIRouter, Depends, HTTPException
 
+from kodit.domain.tracking.trackable import Trackable, TrackableReferenceType
 from kodit.infrastructure.api.middleware.auth import api_key_auth
 from kodit.infrastructure.api.v1.dependencies import (
     CommitIndexingAppServiceDep,
+    EnrichmentQueryServiceDep,
     GitBranchRepositoryDep,
     GitCommitRepositoryDep,
     GitRepositoryDep,
     GitTagRepositoryDep,
     TaskStatusQueryServiceDep,
 )
+from kodit.infrastructure.api.v1.schemas.enrichment import (
+    EnrichmentAttributes,
+    EnrichmentData,
+    EnrichmentListResponse,
+)
 from kodit.infrastructure.api.v1.schemas.repository import (
     RepositoryBranchData,
     RepositoryCommitData,
@@ -259,6 +266,98 @@ async def get_repository_tag(
     )
 
 
+@router.get(
+    "/{repo_id}/enrichments",
+    summary="List latest repository enrichments",
+    responses={404: {"description": "Repository not found"}},
+)
+async def list_repository_enrichments(  # noqa: PLR0913
+    repo_id: str,
+    git_repository: GitRepositoryDep,
+    enrichment_query_service: EnrichmentQueryServiceDep,
+    ref_type: str = "branch",
+    ref_name: str | None = None,
+    enrichment_type: str | None = None,
+    limit: int = 10,
+) -> EnrichmentListResponse:
+    """List the most recent enrichments for a repository.
+
+    Query parameters:
+    - ref_type: Type of reference (branch, tag, or commit_sha). Defaults to "branch".
+    - ref_name: Name of the reference. For branches, defaults to the tracking branch.
+    - enrichment_type: Optional filter for specific enrichment type.
+    - limit: Maximum number of enrichments to return. Defaults to 10.
+    """
+    # Get repository
+    repo = await git_repository.get_by_id(int(repo_id))
+    if not repo:
+        raise HTTPException(status_code=404, detail="Repository not found")
+
+    # Determine the reference to track
+    if ref_name is None:
+        if ref_type == "branch":
+            # Default to tracking branch
+            if not repo.tracking_branch:
+                raise HTTPException(
+                    status_code=400, detail="No tracking branch configured"
+                )
+            ref_name = repo.tracking_branch.name
+        else:
+            raise HTTPException(
+                status_code=400,
+                detail="ref_name is required for tag and commit_sha references",
+            )
+
+    # Parse ref_type
+    try:
+        trackable_type = TrackableReferenceType(ref_type)
+    except ValueError:
+        raise HTTPException(
+            status_code=400,
+            detail=f"Invalid ref_type: {ref_type}. Must be branch, tag, or commit_sha",
+        ) from None
+
+    # Create trackable
+    trackable = Trackable(
+        type=trackable_type, identifier=ref_name, repo_id=int(repo_id)
+    )
+
+    # Find the latest enriched commit
+    enriched_commit = await enrichment_query_service.find_latest_enriched_commit(
+        trackable=trackable,
+        enrichment_type=enrichment_type,
+        max_commits_to_check=limit * 10,  # Check more commits to find enriched ones
+    )
+
+    # If no enriched commit found, return empty list
+    if not enriched_commit:
+        return EnrichmentListResponse(data=[])
+
+    # Get enrichments for the commit
+    enrichments = await enrichment_query_service.get_enrichments_for_commit(
+        commit_sha=enriched_commit,
+        enrichment_type=enrichment_type,
+    )
+
+    # Map enrichments to API response format
+    enrichment_data = [
+        EnrichmentData(
+            type="enrichment",
+            id=str(enrichment.id) if enrichment.id else "0",
+            attributes=EnrichmentAttributes(
+                type=enrichment.type,
+                subtype=enrichment.subtype,
+                content=enrichment.content,
+                created_at=enrichment.created_at,
+                updated_at=enrichment.updated_at,
+            ),
+        )
+        for enrichment in enrichments
+    ]
+
+    return EnrichmentListResponse(data=enrichment_data)
+
+
 @router.delete(
     "/{repo_id}",
     status_code=204,

{kodit-0.5.1.dist-info → kodit-0.5.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kodit
-Version: 0.5.1
+Version: 0.5.2
 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.5.1.dist-info → kodit-0.5.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
 kodit/.gitignore,sha256=ztkjgRwL9Uud1OEi36hGQeDGk3OLK1NfDEO8YqGYy8o,11
 kodit/__init__.py,sha256=aEKHYninUq1yh6jaNfvJBYg-6fenpN132nJt1UU6Jxs,59
-kodit/_version.py,sha256=cYMOhuaBHd0MIZmumuccsEQ-AxM8LIJy9dsBAWgOpqE,704
+kodit/_version.py,sha256=LGYtjQ6cyPZC_N0AovMIeSYYDK21050nm3HYgDanQBM,704
 kodit/app.py,sha256=niIfZiuuDp7mLzrBwQhx_FU7RvKfUALNV5y0o43miss,5802
 kodit/cli.py,sha256=QSTXIUDxZo3anIONY-grZi9_VSehWoS8QoVJZyOmWPQ,3086
 kodit/cli_utils.py,sha256=umkvt4kWNapk6db6RGz6bmn7oxgDpsW2Vo09MZ37OGg,2430
@@ -13,10 +13,11 @@ kodit/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kodit/application/__init__.py,sha256=mH50wTpgP9dhbKztFsL8Dda9Hi18TSnMVxXtpp4aGOA,35
 kodit/application/factories/__init__.py,sha256=bU5CvEnaBePZ7JbkCOp1MGTNP752bnU2uEqmfy5FdRk,37
 kodit/application/factories/reporting_factory.py,sha256=3IpRiAw_olM69db-jbDAtjyGtd6Nh5o8jUJX3-rXCA8,1421
-kodit/application/factories/server_factory.py,sha256=NrOFLbkKsLpymNBYbuvoWdRznV5YRh_BsoLQjx3_i_o,15185
+kodit/application/factories/server_factory.py,sha256=m6tWMH9rSENqEEePpoWtHT51iQZ1IjZ2ZtvLugsbFmo,16456
 kodit/application/services/__init__.py,sha256=p5UQNw-H5sxQvs5Etfte93B3cJ1kKW6DNxK34uFvU1E,38
 kodit/application/services/code_search_application_service.py,sha256=sqMgyAw7e2d2FWroaonaL8G1Hwigb-Yku71dut3wOpQ,4963
 kodit/application/services/commit_indexing_application_service.py,sha256=S5Gep4aXB9_1CWxs9xcIMnGmsrfrwJqvfDAHIPhoS1k,29860
+kodit/application/services/enrichment_query_service.py,sha256=4wtBPwLbmgFCDgYam1D6x6EgSDGEJeQgAt_opzQz5rs,3102
 kodit/application/services/indexing_worker_service.py,sha256=8J8CaUdPd5nF6MyvJbJQpXeGkP2oClmFjZel1xBXELU,4065
 kodit/application/services/queue_service.py,sha256=nXplzN-nehPEeEvygzjJwWg4oQmu3SPodsZzY1Z3MtE,2509
 kodit/application/services/reporting.py,sha256=cwe-S-UpSOE6xSAEhoD1hi4hSWk1bW3YRLJ7463fIvM,3518
@@ -55,6 +56,9 @@ kodit/domain/services/git_repository_service.py,sha256=b-zAAFVxU22KKp2ACyKUgOpFK
 kodit/domain/services/git_service.py,sha256=nVQCfXQ8kW-MAAoAd8bgSQmCdgPMVftUh5qd4du_bes,11352
 kodit/domain/services/physical_architecture_service.py,sha256=0YgoAvbUxT_VwgIh_prftSYnil_XIqNPSoP0g37eIt4,7209
 kodit/domain/services/task_status_query_service.py,sha256=rI93pTMHeycigQryCWkimXSDzRqx_nJOr07UzPAacPE,736
+kodit/domain/tracking/__init__.py,sha256=5FvxhDxB2Fpr8Dw8BLtjfPE1YUWn2rr3u7njQlqcosk,30
+kodit/domain/tracking/resolution_service.py,sha256=w9OMgNv0ZS1DiduY-XFTA_pWK9vXgoEvEGLBl1vqRHE,2765
+kodit/domain/tracking/trackable.py,sha256=-9UT-c5iDkg5LxLl-BEyysvrKgHoYfofuGUacjPhVk4,479
 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=8MjEc6cHCqiI-LtIyng3uKD7a2wUaR-QUdIAePYyIRg,292
@@ -65,11 +69,11 @@ 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=xWtkR3UP7daksCXW_Eyvcqsh091OREqfBPnlFs027_o,22
-kodit/infrastructure/api/v1/dependencies.py,sha256=S0U1oY0CH8rQpF685d1eT2TE6KgbCWsav1E4HIss5KQ,4457
+kodit/infrastructure/api/v1/dependencies.py,sha256=VZhSBgPF1lTc7UnayXl5RsHkA6EvHkGl8NLHPUedjU0,4875
 kodit/infrastructure/api/v1/routers/__init__.py,sha256=pz_7kFwHcxztbTiFI-57Q2tCAllI7u0fgTP4rpQeUoQ,22
 kodit/infrastructure/api/v1/routers/commits.py,sha256=osjmm2Po-MOshD7zBv01C12UcYukkaKATtjEa3hSAJU,11624
 kodit/infrastructure/api/v1/routers/queue.py,sha256=srZmOCZqvcCBlDcPYt1ZWhwVhvVWARWJ3Qp4Tn5eK4Y,2148
-kodit/infrastructure/api/v1/routers/repositories.py,sha256=pxdCCMQWAsezkwyj-rNG-eQbtlW8wMaXZwfzHmae3sI,9328
+kodit/infrastructure/api/v1/routers/repositories.py,sha256=fDc73mt9tc1oqa35LGXr8WpG3zIS6QNuQu9qu15l0I0,12758
 kodit/infrastructure/api/v1/routers/search.py,sha256=eMlofqcy9xWCsE9ugfBZHtcPo1hb-A06_Xfv4XR3FfY,3187
 kodit/infrastructure/api/v1/schemas/__init__.py,sha256=capaxPe7y28pWj6Pu5hfTOxLnVL9pwW-hJu7ZdN2klw,41
 kodit/infrastructure/api/v1/schemas/commit.py,sha256=UVGkwZNjwUMiitUbFws1_mlZN7IALq99P99HJCg5h2c,1794
@@ -161,8 +165,8 @@ kodit/utils/dump_config.py,sha256=dd5uPgqh6ATk02Zt59t2JFKR9X17YWjHudV0nE8VktE,11
 kodit/utils/dump_openapi.py,sha256=EasYOnnpeabwb_sTKQUBrrOLHjPcOFQ7Zx0YKpx9fmM,1239
 kodit/utils/generate_api_paths.py,sha256=TMtx9v55podDfUmiWaHgJHLtEWLV2sLL-5ejGFMPzAo,3569
 kodit/utils/path_utils.py,sha256=UB_81rx7Y1G1jalVv2PX8miwaprBbcqEdtoQ3hPT3kU,2451
-kodit-0.5.1.dist-info/METADATA,sha256=HyFZ0U2-rvusI_5CIa3Z1BOaER_--WZEyy_g3i1BHg8,7703
-kodit-0.5.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-kodit-0.5.1.dist-info/entry_points.txt,sha256=hoTn-1aKyTItjnY91fnO-rV5uaWQLQ-Vi7V5et2IbHY,40
-kodit-0.5.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-kodit-0.5.1.dist-info/RECORD,,
+kodit-0.5.2.dist-info/METADATA,sha256=gcQYxwPZ5mx36u7ejs76xkZSBKUEnyGzDc5qJAcrrlQ,7703
+kodit-0.5.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+kodit-0.5.2.dist-info/entry_points.txt,sha256=hoTn-1aKyTItjnY91fnO-rV5uaWQLQ-Vi7V5et2IbHY,40
+kodit-0.5.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+kodit-0.5.2.dist-info/RECORD,,