31 lines
927 B
Python
31 lines
927 B
Python
|
import argparse
|
||
|
|
|
||
|
|
|
||
|
|
def build_parser() -> argparse.ArgumentParser:
|
||
|
|
parser = argparse.ArgumentParser(description="Inspect local Chroma collections.")
|
||
|
|
subparsers = parser.add_subparsers(dest="command")
|
||
|
|
|
||
|
|
subparsers.add_parser(
|
||
|
|
"list-collections",
|
||
|
|
aliases=["lc"],
|
||
|
|
help="List all collections stored in the local Chroma database.",
|
||
|
|
)
|
||
|
|
|
||
|
|
create_parser = subparsers.add_parser(
|
||
|
|
"create-collection",
|
||
|
|
aliases=["cc"],
|
||
|
|
help="Create a collection in the local Chroma database.",
|
||
|
|
)
|
||
|
|
create_parser.add_argument("name", help="Name of the collection to create.")
|
||
|
|
|
||
|
|
delete_parser = subparsers.add_parser(
|
||
|
|
"delete-collection",
|
||
|
|
aliases=["dc"],
|
||
|
|
help="Delete a collection from the local Chroma database.",
|
||
|
|
)
|
||
|
|
delete_parser.add_argument("name", help="Name of the collection to delete.")
|
||
|
|
|
||
|
|
# TODO add "count <name>" command.
|
||
|
|
|
||
|
|
return parser
|