Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
25 changes: 19 additions & 6 deletions src/vectorcode/cli_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -172,22 +172,27 @@ async def merge_from(self, other: "Config") -> "Config":


def get_cli_parser():
__default_config = Config()
shared_parser = argparse.ArgumentParser(add_help=False)
chunking_parser = argparse.ArgumentParser(add_help=False)
chunking_parser.add_argument(
"--overlap", "-o", type=float, help="Ratio of overlaps between chunks."
"--overlap",
"-o",
type=float,
default=__default_config.overlap_ratio,
help="Ratio of overlaps between chunks.",
)
chunking_parser.add_argument(
"-c",
"--chunk_size",
type=int,
default=-1,
default=__default_config.chunk_size,
help="Size of chunks (-1 for no chunking).",
)
chunking_parser.add_argument(
"--encoding",
type=str,
default="utf8",
default=__default_config.encoding,
help="Encoding used by the files. See https://docs.python.org/3/library/codecs.html#standard-encodings for supported encodings. Use `_auto` for automatic encoding detection.",
)
shared_parser.add_argument(
Expand Down Expand Up @@ -262,10 +267,18 @@ def get_cli_parser():
)
query_parser.add_argument("query", nargs="+", help="Query keywords.")
query_parser.add_argument(
"--multiplier", "-m", type=int, default=-1, help="Query multiplier."
"--multiplier",
"-m",
type=int,
default=__default_config.query_multiplier,
help="Query multiplier.",
)
query_parser.add_argument(
"-n", "--number", type=int, default=1, help="Number of results to retrieve."
"-n",
"--number",
type=int,
default=__default_config.n_result,
help="Number of results to retrieve.",
)
query_parser.add_argument(
"--exclude", nargs="*", help="Files to exclude from query results."
Expand All @@ -281,7 +294,7 @@ def get_cli_parser():
choices=list(i.value for i in QueryInclude),
nargs="+",
help="What to include in the final output.",
default=["path", "document"],
default=__default_config.include,
)

subparsers.add_parser("drop", parents=[shared_parser], help="Remove a collection.")
Expand Down
6 changes: 6 additions & 0 deletions tests/test_cli_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -489,6 +489,12 @@ async def test_parse_cli_args_chunks():
assert config.overlap_ratio == 0.5
assert config.chunk_size == 100

with patch("sys.argv", ["vectorcode", "chunks", "file.py"]):
config = await parse_cli_args()
assert config.action == CliAction.chunks
assert config.overlap_ratio == Config().overlap_ratio
assert config.chunk_size == Config().chunk_size


@pytest.mark.asyncio
async def test_config_import_from_hnsw():
Expand Down