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.
"""
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'])

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']}")
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)}")