kaizenkodo/ChatMastermind
2
0
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases Wiki Activity

fixed 'tag' and 'hist' commands

Browse Source
This commit is contained in:
juk0de 2023-08-12 08:20:00 +02:00
parent f90e7bcd47
commit 5a435c5f8f
2 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
chatmastermind/main.py
View File

@ -67,7 +67,7 @@ def tag_cmd(args: argparse.Namespace, config: dict) -> None:
Handler for the 'tag' command. Handler for the 'tag' command.
""" """
if args.list: 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: 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. Handler for the 'print' command.
""" """
fname = pathlib.Path(args.print) fname = pathlib.Path(args.file)
if fname.suffix == '.yaml': 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) data = yaml.load(f, Loader=yaml.FullLoader)
elif fname.suffix == '.txt': elif fname.suffix == '.txt':
data = read_file(fname) data = read_file(fname)
else: else:
print(f"Unknown file type: {args.print}") print(f"Unknown file type: {args.file}")
sys.exit(1) sys.exit(1)
if args.only_source_code: if args.only_source_code:
display_source_code(data['answer']) display_source_code(data['answer'])

5
chatmastermind/utils.py
View File

@ -78,9 +78,6 @@ def print_chat_hist(chat, dump=False, source_code=False) -> None:
print(f"{message['role'].upper()}: {message['content']}") print(f"{message['role'].upper()}: {message['content']}")
def print_tags_frequency(tags: List[str], dump=False) -> None: def print_tags_frequency(tags: List[str]) -> None:
if dump:
pp(tags)
return
for tag in sorted(set(tags)): for tag in sorted(set(tags)):
print(f"- {tag}: {tags.count(tag)}") print(f"- {tag}: {tags.count(tag)}")