From 5a435c5f8f701e14a7a82b675b581aaa34fe10fc Mon Sep 17 00:00:00 2001 From: juk0de Date: Sat, 12 Aug 2023 08:20:00 +0200 Subject: [PATCH] fixed 'tag' and 'hist' commands --- chatmastermind/main.py | 8 ++++---- chatmastermind/utils.py | 5 +---- 2 files changed, 5 insertions(+), 8 deletions(-) diff --git a/chatmastermind/main.py b/chatmastermind/main.py index 32622da..f80c26c 100755 --- a/chatmastermind/main.py +++ b/chatmastermind/main.py @@ -67,7 +67,7 @@ def tag_cmd(args: argparse.Namespace, config: dict) -> None: Handler for the 'tag' command. """ if args.list: - print_tags_frequency(get_tags(config, None), args.dump) + print_tags_frequency(get_tags(config, None)) def model_cmd(args: argparse.Namespace, config: dict) -> None: @@ -115,14 +115,14 @@ def print_cmd(args: argparse.Namespace, config: dict) -> None: """ Handler for the 'print' command. """ - fname = pathlib.Path(args.print) + fname = pathlib.Path(args.file) if fname.suffix == '.yaml': - with open(args.print, 'r') as f: + with open(args.file, 'r') as f: data = yaml.load(f, Loader=yaml.FullLoader) elif fname.suffix == '.txt': data = read_file(fname) else: - print(f"Unknown file type: {args.print}") + print(f"Unknown file type: {args.file}") sys.exit(1) if args.only_source_code: display_source_code(data['answer']) diff --git a/chatmastermind/utils.py b/chatmastermind/utils.py index cdd0e60..78440fa 100644 --- a/chatmastermind/utils.py +++ b/chatmastermind/utils.py @@ -78,9 +78,6 @@ def print_chat_hist(chat, dump=False, source_code=False) -> None: print(f"{message['role'].upper()}: {message['content']}") -def print_tags_frequency(tags: List[str], dump=False) -> None: - if dump: - pp(tags) - return +def print_tags_frequency(tags: List[str]) -> None: for tag in sorted(set(tags)): print(f"- {tag}: {tags.count(tag)}")