simplify the app using typer

chromy/cli.py

@@ -0,0 +1,159 @@
+from __future__ import annotations
+
+from typing import Annotated, Callable
+
+import typer
+from chromadb.errors import InternalError, NotFoundError
+
+from chromy.handlers.add_data import handle_add_data
+from chromy.handlers.count_collection import handle_count_collection
+from chromy.handlers.create_collection import handle_create_collection
+from chromy.handlers.delete_collection import (
+    handle_delete_collection,
+    handle_delete_records,
+)
+from chromy.handlers.list_collections import handle_list_collections
+from chromy.handlers.query import handle_query
+
+app = typer.Typer(help="Inspect local Chroma collections.")
+
+ExitCodeHandler = Callable[[], int]
+
+
+def _run(handler: ExitCodeHandler) -> None:
+    exit_code = handler()
+    if exit_code != 0:
+        raise typer.Exit(exit_code)
+
+
+def _fail(message: str) -> None:
+    typer.echo(message)
+    raise typer.Exit(1)
+
+
+@app.command("lc", help="List all collections stored in the local Chroma database.")
+@app.command(
+    "list-collections",
+    help="List all collections stored in the local Chroma database.",
+)
+def list_collections() -> None:
+    _run(handle_list_collections)
+
+
+@app.command("cc", help="Create a collection in the local Chroma database.")
+@app.command(
+    "create-collection",
+    help="Create a collection in the local Chroma database.",
+)
+def create_collection(
+    collection: Annotated[
+        str,
+        typer.Argument(help="Name of the collection to create."),
+    ],
+) -> None:
+    try:
+        _run(lambda: handle_create_collection(collection))
+    except InternalError:
+        _fail(f"Collection '{collection}' already exists.")
+
+
+@app.command("dc", help="Delete a collection from the local Chroma database.")
+@app.command(
+    "delete-collection",
+    help="Delete a collection from the local Chroma database.",
+)
+def delete_collection(
+    collection: Annotated[
+        str,
+        typer.Argument(help="Name of the collection to delete."),
+    ],
+) -> None:
+    try:
+        _run(lambda: handle_delete_collection(collection))
+    except NotFoundError:
+        _fail(f"Collection '{collection}' does not exist.")
+
+
+@app.command("co", help="Count records in a collection from the local Chroma database.")
+@app.command(
+    "count",
+    help="Count records in a collection from the local Chroma database.",
+)
+def count(
+    collection: Annotated[
+        str,
+        typer.Argument(help="Name of the collection to count."),
+    ],
+) -> None:
+    try:
+        _run(lambda: handle_count_collection(collection))
+    except NotFoundError:
+        _fail(f"Collection '{collection}' does not exist.")
+
+
+@app.command(
+    "ad",
+    help="Chunk, embed, and add a file to a collection in the local Chroma database.",
+)
+@app.command(
+    "add-data",
+    help="Chunk, embed, and add a file to a collection in the local Chroma database.",
+)
+def add_data(
+    collection: Annotated[
+        str,
+        typer.Argument(help="Name of the target collection."),
+    ],
+    file: Annotated[
+        str,
+        typer.Argument(help="Path to the file to chunk and add to the collection."),
+    ],
+) -> None:
+    try:
+        _run(lambda: handle_add_data(collection, file))
+    except NotFoundError:
+        _fail(f"Collection '{collection}' does not exist.")
+    except FileNotFoundError:
+        _fail(f"The file {file} was not found.")
+
+
+@app.command("q", help="Query a collection with the provided text.")
+@app.command("query", help="Query a collection with the provided text.")
+def query(
+    collection: Annotated[
+        str,
+        typer.Argument(help="Name of the target collection."),
+    ],
+    query_text: Annotated[
+        str,
+        typer.Argument(help="The text to query."),
+    ],
+) -> None:
+    try:
+        _run(lambda: handle_query(collection, query_text))
+    except NotFoundError:
+        _fail(f"Collection '{collection}' does not exist.")
+
+
+@app.command("del", help="Delete records from a collection using a metadata filter.")
+@app.command("delete", help="Delete records from a collection using a metadata filter.")
+def delete_records(
+    collection: Annotated[
+        str,
+        typer.Argument(help="Name of the target collection."),
+    ],
+    where: Annotated[
+        str,
+        typer.Option(
+            "--where",
+            help="Metadata filter in the format <condition>=<value>.",
+            metavar="CONDITION=VALUE",
+        ),
+    ],
+) -> None:
+    try:
+        _run(lambda: handle_delete_records(collection, where))
+    except NotFoundError:
+        _fail(f"Collection '{collection}' does not exist.")
+    except ValueError as exc:
+        _fail(str(exc))