diff --git a/README.md b/README.md index 56e4525..dd38a71 100644 --- a/README.md +++ b/README.md @@ -28,7 +28,7 @@ pip install . ## Usage ```bash -cmm [-h] [-p PRINT | -q QUESTION | -D | -d | -l] [-c CONFIG] [-m MAX_TOKENS] [-T TEMPERATURE] [-M MODEL] [-n NUMBER] [-t [TAGS [TAGS ...]]] [-e [EXTAGS [EXTAGS ...]]] [-o [OTAGS [OTAGS ...]]] [-w] [-W] +cmm [-h] [-p PRINT | -q QUESTION | -D | -d | -l] [-c CONFIG] [-m MAX_TOKENS] [-T TEMPERATURE] [-M MODEL] [-n NUMBER] [-t [TAGS [TAGS ...]]] [-e [EXTAGS [EXTAGS ...]]] [-o [OTAGS [OTAGS ...]]] [-a] [-w] [-W] ``` ### Arguments @@ -37,6 +37,7 @@ cmm [-h] [-p PRINT | -q QUESTION | -D | -d | -l] [-c CONFIG] [-m MAX_TOKENS] [-T - `-q`, `--question`: Question to ask. - `-D`, `--chat-dump`: Print chat history as a Python structure. - `-d`, `--chat`: Print chat history as readable text. +- `-a`, `--match-all-tags`: All given tags must match when selecting chat history entries. - `-w`, `--with-tags`: Print chat history with tags. - `-W`, `--with-tags`: Print chat history with filenames. - `-l`, `--list-tags`: List all tags and their frequency. diff --git a/chatmastermind/main.py b/chatmastermind/main.py index 5dad76a..6fe4984 100755 --- a/chatmastermind/main.py +++ b/chatmastermind/main.py @@ -56,7 +56,8 @@ def process_and_display_chat(args: argparse.Namespace, full_question = '\n\n'.join(question_parts) chat = create_chat(full_question, tags, extags, config, - args.with_tags, args.with_file) + args.match_all_tags, args.with_tags, + args.with_file) display_chat(chat, dump, args.only_source_code) return chat, full_question, tags @@ -104,6 +105,7 @@ def create_parser() -> argparse.ArgumentParser: parser.add_argument('-S', '--only-source-code', help='Print only source code', action='store_true') parser.add_argument('-w', '--with-tags', help="Print chat history with tags.", action='store_true') parser.add_argument('-W', '--with-file', help="Print chat history with filename.", action='store_true') + parser.add_argument('-a', '--match-all-tags', help="All given tags must match when selecting chat history entries.", action='store_true') tags_arg = parser.add_argument('-t', '--tags', nargs='*', help='List of tag names', metavar='TAGS') tags_arg.completer = tags_completer # type: ignore extags_arg = parser.add_argument('-e', '--extags', nargs='*', help='List of tag names to exclude', metavar='EXTAGS') diff --git a/chatmastermind/storage.py b/chatmastermind/storage.py index 0839d61..ee1443f 100644 --- a/chatmastermind/storage.py +++ b/chatmastermind/storage.py @@ -64,6 +64,7 @@ def create_chat(question: Optional[str], tags: Optional[List[str]], extags: Optional[List[str]], config: Dict[str, Any], + match_all_tags: bool = False, with_tags: bool = False, with_file: bool = False ) -> List[Dict[str, str]]: @@ -78,8 +79,11 @@ def create_chat(question: Optional[str], else: continue data_tags = set(data.get('tags', [])) - tags_match = \ - not tags or data_tags.intersection(tags) + tags_match: bool + if match_all_tags: + tags_match = not tags or set(tags).issubset(data_tags) + else: + tags_match = not tags or bool(data_tags.intersection(tags)) extags_do_not_match = \ not extags or not data_tags.intersection(extags) if tags_match and extags_do_not_match: