Skip to content

Commit 4740e9a

Browse files
author
Zhe Yu
committed
refactor(cli): use a simpler approach to build the config object.
1 parent fb3043d commit 4740e9a

1 file changed

Lines changed: 25 additions & 49 deletions

File tree

src/vectorcode/cli_utils.py

Lines changed: 25 additions & 49 deletions
Original file line numberDiff line numberDiff line change
@@ -338,61 +338,37 @@ async def parse_cli_args(args: Optional[Sequence[str]] = None):
338338
if main_args.action is None:
339339
main_args = main_parser.parse_args(["--help"])
340340

341-
files = []
342-
query = None
343-
recursive = False
344-
include_hidden = False
345-
number_of_result = 1
346-
force = False
347-
chunk_size = -1
348-
overlap_ratio = 0.2
349-
query_multiplier = -1
350-
query_exclude = []
351-
query_include = ["path", "document"]
352-
check_item = None
353-
absolute = False
341+
configs_items: dict[str, Any] = {
342+
"no_stderr": main_args.no_stderr,
343+
"action": CliAction(main_args.action),
344+
"project_root": main_args.project_root,
345+
"pipe": main_args.pipe,
346+
}
347+
354348
match main_args.action:
355349
case "vectorise":
356-
files = main_args.file_paths
357-
recursive = main_args.recursive
358-
include_hidden = main_args.include_hidden
359-
force = main_args.force
360-
chunk_size = main_args.chunk_size
361-
overlap_ratio = main_args.overlap
350+
configs_items["files"] = main_args.file_paths
351+
configs_items["recursive"] = main_args.recursive
352+
configs_items["include_hidden"] = main_args.include_hidden
353+
configs_items["force"] = main_args.force
354+
configs_items["chunk_size"] = main_args.chunk_size
355+
configs_items["overlap_ratio"] = main_args.overlap
362356
case "query":
363-
query = main_args.query
364-
number_of_result = main_args.number
365-
query_multiplier = main_args.multiplier
366-
query_exclude = main_args.exclude
367-
absolute = main_args.absolute
368-
query_include = main_args.include
357+
configs_items["query"] = main_args.query
358+
configs_items["n_result"] = main_args.number
359+
configs_items["query_multiplier"] = main_args.multiplier
360+
configs_items["query_exclude"] = main_args.exclude
361+
configs_items["use_absolute_path"] = main_args.absolute
362+
configs_items["include"] = [QueryInclude(i) for i in main_args.include]
369363
case "check":
370-
check_item = main_args.check_item
364+
configs_items["check_item"] = main_args.check_item
371365
case "init":
372-
force = main_args.force
366+
configs_items["force"] = main_args.force
373367
case "chunks":
374-
files = main_args.file_paths
375-
chunk_size = main_args.chunk_size
376-
overlap_ratio = main_args.overlap
377-
return Config(
378-
no_stderr=main_args.no_stderr,
379-
action=CliAction(main_args.action),
380-
files=files,
381-
project_root=main_args.project_root,
382-
query=query,
383-
recursive=recursive,
384-
include_hidden=include_hidden,
385-
n_result=number_of_result,
386-
pipe=main_args.pipe,
387-
force=force,
388-
chunk_size=chunk_size,
389-
overlap_ratio=overlap_ratio,
390-
query_multiplier=query_multiplier,
391-
query_exclude=query_exclude,
392-
check_item=check_item,
393-
use_absolute_path=absolute,
394-
include=[QueryInclude(i) for i in query_include],
395-
)
368+
configs_items["files"] = main_args.file_paths
369+
configs_items["chunk_size"] = main_args.chunk_size
370+
configs_items["overlap_ratio"] = main_args.overlap
371+
return Config(**configs_items)
396372

397373

398374
def expand_envs_in_dict(d: dict):

0 commit comments

Comments
 (0)