kodit 0.1.6__py3-none-any.whl → 0.1.7__py3-none-any.whl

kodit/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.1.6'
-__version_tuple__ = version_tuple = (0, 1, 6)
+__version__ = version = '0.1.7'
+__version_tuple__ = version_tuple = (0, 1, 7)

kodit/cli.py

@@ -8,7 +8,7 @@ from typing import Any
 import click
 import structlog
 import uvicorn
-from pytable_formatter import Table
+from pytable_formatter import Cell, Table
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from kodit.config import (
@@ -84,110 +84,59 @@ def cli(  # noqa: PLR0913
     ctx.obj = config
 
 
-@cli.group()
-def sources() -> None:
-    """Manage code sources."""
-
-
-@sources.command(name="list")
-@with_app_context
-@with_session
-async def list_sources(session: AsyncSession, app_context: AppContext) -> None:
-    """List all code sources."""
-    repository = SourceRepository(session)
-    service = SourceService(app_context.get_clone_dir(), repository)
-    sources = await service.list_sources()
-
-    # Define headers and data
-    headers = ["ID", "Created At", "URI"]
-    data = [[source.id, source.created_at, source.uri] for source in sources]
-
-    # Create and display the table
-    table = Table(headers=headers, data=data)
-    click.echo(table)
-
-
-@sources.command(name="create")
-@click.argument("uri")
-@with_app_context
-@with_session
-async def create_source(
-    session: AsyncSession, app_context: AppContext, uri: str
-) -> None:
-    """Add a new code source."""
-    repository = SourceRepository(session)
-    service = SourceService(app_context.get_clone_dir(), repository)
-    source = await service.create(uri)
-    click.echo(f"Source created: {source.id}")
-
-
-@cli.group()
-def indexes() -> None:
-    """Manage indexes."""
-
-
-@indexes.command(name="create")
-@click.argument("source_id")
+@cli.command()
+@click.argument("sources", nargs=-1)
 @with_app_context
 @with_session
-async def create_index(
-    session: AsyncSession, app_context: AppContext, source_id: int
+async def index(
+    session: AsyncSession,
+    app_context: AppContext,
+    sources: list[str],
 ) -> None:
-    """Create an index for a source."""
+    """List indexes, or index data sources."""
     source_repository = SourceRepository(session)
     source_service = SourceService(app_context.get_clone_dir(), source_repository)
     repository = IndexRepository(session)
     service = IndexService(repository, source_service, app_context.get_data_dir())
-    index = await service.create(source_id)
-    click.echo(f"Index created: {index.id}")
 
-
-@indexes.command(name="list")
-@with_app_context
-@with_session
-async def list_indexes(session: AsyncSession, app_context: AppContext) -> None:
-    """List all indexes."""
-    source_repository = SourceRepository(session)
-    source_service = SourceService(app_context.get_clone_dir(), source_repository)
-    repository = IndexRepository(session)
-    service = IndexService(repository, source_service, app_context.get_data_dir())
-    indexes = await service.list_indexes()
-
-    # Define headers and data
-    headers = [
-        "ID",
-        "Created At",
-        "Updated At",
-        "Num Snippets",
-    ]
-    data = [
-        [
-            index.id,
-            index.created_at,
-            index.updated_at,
-            index.num_snippets,
+    if not sources:
+        # No source specified, list all indexes
+        indexes = await service.list_indexes()
+        headers: list[str | Cell] = [
+            "ID",
+            "Created At",
+            "Updated At",
+            "Source",
+            "Num Snippets",
         ]
-        for index in indexes
-    ]
-
-    # Create and display the table
-    table = Table(headers=headers, data=data)
-    click.echo(table)
-
-
-@indexes.command(name="run")
-@click.argument("index_id")
-@with_app_context
-@with_session
-async def run_index(
-    session: AsyncSession, app_context: AppContext, index_id: int
-) -> None:
-    """Run an index."""
-    source_repository = SourceRepository(session)
-    source_service = SourceService(app_context.get_clone_dir(), source_repository)
-    repository = IndexRepository(session)
-    service = IndexService(repository, source_service, app_context.get_data_dir())
-    await service.run(index_id)
+        data = [
+            [
+                index.id,
+                index.created_at,
+                index.updated_at,
+                index.source,
+                index.num_snippets,
+            ]
+            for index in indexes
+        ]
+        click.echo(Table(headers=headers, data=data))
+        return
+    # Handle source indexing
+    for source in sources:
+        if source.startswith("https://"):
+            msg = "Web or git indexing is not implemented yet"
+            raise click.UsageError(msg)
+        if source.startswith("git"):
+            msg = "Git indexing is not implemented yet"
+            raise click.UsageError(msg)
+        if Path(source).is_file():
+            msg = "File indexing is not implemented yet"
+            raise click.UsageError(msg)
+
+        # Index directory
+        s = await source_service.create(source)
+        index = await service.create(s.id)
+        await service.run(index.id)
 
 
 @cli.command()

kodit/indexing/repository.py

@@ -82,7 +82,7 @@ class IndexRepository:
         result = await self.session.execute(query)
         return list(result.scalars())
 
-    async def list_indexes(self) -> list[Index]:
+    async def list_indexes(self) -> list[tuple[Index, Source]]:
         """List all indexes.
 
         Returns:
@@ -90,9 +90,11 @@ class IndexRepository:
             and counts of files and snippets.
 
         """
-        query = select(Index).limit(10)
+        query = select(Index, Source).join(
+            Source, Index.source_id == Source.id, full=True
+        )
         result = await self.session.execute(query)
-        return list(result.scalars())
+        return list(result.tuples())
 
     async def num_snippets_for_index(self, index_id: int) -> int:
         """Get the number of snippets for an index."""

kodit/indexing/service.py

@@ -33,6 +33,7 @@ class IndexView(pydantic.BaseModel):
     id: int
     created_at: datetime
     updated_at: datetime | None = None
+    source: str | None = None
     num_snippets: int | None = None
 
 
@@ -105,8 +106,9 @@ class IndexService:
                 created_at=index.created_at,
                 updated_at=index.updated_at,
                 num_snippets=await self.repository.num_snippets_for_index(index.id),
+                source=source.uri,
             )
-            for index in indexes
+            for index, source in indexes
         ]
 
     async def run(self, index_id: int) -> None:

{kodit-0.1.6.dist-info → kodit-0.1.7.dist-info}/METADATA

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

@@ -1,8 +1,8 @@
 kodit/.gitignore,sha256=ztkjgRwL9Uud1OEi36hGQeDGk3OLK1NfDEO8YqGYy8o,11
 kodit/__init__.py,sha256=aEKHYninUq1yh6jaNfvJBYg-6fenpN132nJt1UU6Jxs,59
-kodit/_version.py,sha256=ESbJO0YD7TYfOUv_WDIJJgWELGepEWsoyhqVifEcXPA,511
+kodit/_version.py,sha256=W_EoL8cAL4KhujvbYWEpb9NqRLbbrH0T024lJvRRWHI,511
 kodit/app.py,sha256=Mr5BFHOHx5zppwjC4XPWVvHjwgl1yrKbUjTWXKubJQM,891
-kodit/cli.py,sha256=CNewAn4aCDBbBRKfTaO5hosklCySvJM7afDN04h3lvY,8307
+kodit/cli.py,sha256=x1zw2zOlGhhU6D3E-GU3cMw3l9CqKC76geQREAQKweY,6915
 kodit/config.py,sha256=nlm9U-nVx5riH2SrU1XY4XcCMhQK4DrwO_1H8bPOBjA,2927
 kodit/database.py,sha256=vtTlmrXHyHJH3Ek-twZTCqEjB0jun-NncALFze2fqhA,2350
 kodit/logging.py,sha256=cFEQXWI27LzWScSxly9ApwkbBDamUG17pA-jEfVakXQ,5316
@@ -18,8 +18,8 @@ kodit/bm25/__init__.py,sha256=j8zyriNWhbwE5Lbybzg1hQAhANlU9mKHWw4beeUR6og,19
 kodit/bm25/bm25.py,sha256=3wyNRSrTaYqV7s4R1D6X0NpCf22PuFK2_uc8YapzYLE,2263
 kodit/indexing/__init__.py,sha256=cPyi2Iej3G1JFWlWr7X80_UrsMaTu5W5rBwgif1B3xo,75
 kodit/indexing/models.py,sha256=sZIhGwvL4Dw0QTWFxrjfWctSLkAoDT6fv5DlGz8-Fr8,1258
-kodit/indexing/repository.py,sha256=kvAlNfMSQYboF0TB1huw2qoBdLJ4UsEPiM7ZG-e6rrg,4300
-kodit/indexing/service.py,sha256=eopx_IZeaUjCI-5LeSqZq7W7m76JZsDcVvOFhLquHpI,5426
+kodit/indexing/repository.py,sha256=C020FGpIfTZmVZg7NH04kVuffWv7r7m-82Pdex8CItg,4388
+kodit/indexing/service.py,sha256=7vHqevve-PnKHP2pDfyrW5n3AXVOghABWNsNTw588KY,5499
 kodit/retreival/__init__.py,sha256=33PhJU-3gtsqYq6A1UkaLNKbev_Zee9Lq6dYC59-CsA,69
 kodit/retreival/repository.py,sha256=1lqGgJHsBmvMGMzEYa-hrdXg2q7rqtYPl1cvBb7jMRE,3119
 kodit/retreival/service.py,sha256=9wvURtPPJVvPUWNIC2waIrJMxcm1Ka1J_xDEOEedAFU,2007
@@ -33,8 +33,8 @@ kodit/sources/__init__.py,sha256=1NTZyPdjThVQpZO1Mp1ColVsS7sqYanOVLqnoqV9Ipo,83
 kodit/sources/models.py,sha256=xb42CaNDO1CUB8SIW-xXMrB6Ji8cFw-yeJ550xBEg9Q,2398
 kodit/sources/repository.py,sha256=mGJrHWH6Uo8YABdoojHFbzaf_jW-2ywJpAHIa1gnc3U,3401
 kodit/sources/service.py,sha256=cBCxnOQKwGNi2e13_3Vue8MylAaUxb9XG4IgM636la0,6712
-kodit-0.1.6.dist-info/METADATA,sha256=1HHbQJLVCV6-1Kim8-_pwXq_HmSYUgBLObA_Q3Ck2l8,2181
-kodit-0.1.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-kodit-0.1.6.dist-info/entry_points.txt,sha256=hoTn-1aKyTItjnY91fnO-rV5uaWQLQ-Vi7V5et2IbHY,40
-kodit-0.1.6.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-kodit-0.1.6.dist-info/RECORD,,
+kodit-0.1.7.dist-info/METADATA,sha256=8doJ6TfmVkn-OTLSbgRlDut6YxFbmBKs1L0rgZUlxUQ,2181
+kodit-0.1.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+kodit-0.1.7.dist-info/entry_points.txt,sha256=hoTn-1aKyTItjnY91fnO-rV5uaWQLQ-Vi7V5et2IbHY,40
+kodit-0.1.7.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+kodit-0.1.7.dist-info/RECORD,,