diff --git a/chatmastermind/commands/config.py b/chatmastermind/commands/config.py new file mode 100644 index 0000000..262164c --- /dev/null +++ b/chatmastermind/commands/config.py @@ -0,0 +1,11 @@ +import argparse +from pathlib import Path +from ..configuration import Config + + +def config_cmd(args: argparse.Namespace) -> None: + """ + Handler for the 'config' command. + """ + if args.create: + Config.create_default(Path(args.create)) diff --git a/chatmastermind/commands/hist.py b/chatmastermind/commands/hist.py new file mode 100644 index 0000000..88ed3be --- /dev/null +++ b/chatmastermind/commands/hist.py @@ -0,0 +1,23 @@ +import argparse +from pathlib import Path +from ..configuration import Config +from ..chat import ChatDB +from ..message import MessageFilter + + +def hist_cmd(args: argparse.Namespace, config: Config) -> None: + """ + Handler for the 'hist' command. + """ + + mfilter = MessageFilter(tags_or=args.or_tags, + tags_and=args.and_tags, + tags_not=args.exclude_tags, + question_contains=args.question, + answer_contains=args.answer) + chat = ChatDB.from_dir(Path('.'), + Path(config.db), + mfilter=mfilter) + chat.print(args.source_code_only, + args.with_tags, + args.with_files) diff --git a/chatmastermind/commands/print.py b/chatmastermind/commands/print.py new file mode 100644 index 0000000..51e76f8 --- /dev/null +++ b/chatmastermind/commands/print.py @@ -0,0 +1,19 @@ +import sys +import argparse +from pathlib import Path +from ..configuration import Config +from ..message import Message, MessageError + + +def print_cmd(args: argparse.Namespace, config: Config) -> None: + """ + Handler for the 'print' command. + """ + fname = Path(args.file) + try: + message = Message.from_file(fname) + if message: + print(message.to_str(source_code_only=args.source_code_only)) + except MessageError: + print(f"File is not a valid message: {args.file}") + sys.exit(1) diff --git a/chatmastermind/commands/question.py b/chatmastermind/commands/question.py new file mode 100644 index 0000000..39af6c0 --- /dev/null +++ b/chatmastermind/commands/question.py @@ -0,0 +1,49 @@ +import argparse +from pathlib import Path +from ..configuration import Config +from ..chat import ChatDB +from ..message import Message, Question +from ..ai_factory import create_ai +from ..ai import AI, AIResponse + + +def question_cmd(args: argparse.Namespace, config: Config) -> None: + """ + Handler for the 'question' command. + """ + chat = ChatDB.from_dir(cache_path=Path('.'), + db_path=Path(config.db)) + # if it's a new question, create and store it immediately + if args.ask or args.create: + # FIXME: add sources to the question + message = Message(question=Question(args.question), + tags=args.ouput_tags, # FIXME + ai=args.ai, + model=args.model) + chat.add_to_cache([message]) + if args.create: + return + + # create the correct AI instance + ai: AI = create_ai(args, config) + if args.ask: + response: AIResponse = ai.request(message, + chat, + args.num_answers, # FIXME + args.otags) # FIXME + assert response + # TODO: + # * add answer to the message above (and create + # more messages for any additional answers) + pass + elif args.repeat: + lmessage = chat.latest_message() + assert lmessage + # TODO: repeat either the last question or the + # one(s) given in 'args.repeat' (overwrite + # existing ones if 'args.overwrite' is True) + pass + elif args.process: + # TODO: process either all questions without an + # answer or the one(s) given in 'args.process' + pass diff --git a/chatmastermind/commands/tags.py b/chatmastermind/commands/tags.py new file mode 100644 index 0000000..2906a5b --- /dev/null +++ b/chatmastermind/commands/tags.py @@ -0,0 +1,17 @@ +import argparse +from pathlib import Path +from ..configuration import Config +from ..chat import ChatDB + + +def tags_cmd(args: argparse.Namespace, config: Config) -> None: + """ + Handler for the 'tags' command. + """ + chat = ChatDB.from_dir(cache_path=Path('.'), + db_path=Path(config.db)) + if args.list: + tags_freq = chat.tags_frequency(args.prefix, args.contain) + for tag, freq in tags_freq.items(): + print(f"- {tag}: {freq}") + # TODO: add renaming diff --git a/chatmastermind/main.py b/chatmastermind/main.py index 857bb5a..5c3a458 100755 --- a/chatmastermind/main.py +++ b/chatmastermind/main.py @@ -6,12 +6,14 @@ import sys import argcomplete import argparse from pathlib import Path -from .configuration import Config, default_config_path -from .chat import ChatDB -from .message import Message, MessageFilter, MessageError, Question -from .ai_factory import create_ai -from .ai import AI, AIResponse from typing import Any +from .configuration import Config, default_config_path +from .message import Message +from .commands.question import question_cmd +from .commands.tags import tags_cmd +from .commands.config import config_cmd +from .commands.hist import hist_cmd +from .commands.print import print_cmd def tags_completer(prefix: str, parsed_args: Any, **kwargs: Any) -> list[str]: @@ -19,101 +21,6 @@ def tags_completer(prefix: str, parsed_args: Any, **kwargs: Any) -> list[str]: return list(Message.tags_from_dir(Path(config.db), prefix=prefix)) -def tags_cmd(args: argparse.Namespace, config: Config) -> None: - """ - Handler for the 'tags' command. - """ - chat = ChatDB.from_dir(cache_path=Path('.'), - db_path=Path(config.db)) - if args.list: - tags_freq = chat.tags_frequency(args.prefix, args.contain) - for tag, freq in tags_freq.items(): - print(f"- {tag}: {freq}") - # TODO: add renaming - - -def config_cmd(args: argparse.Namespace) -> None: - """ - Handler for the 'config' command. - """ - if args.create: - Config.create_default(Path(args.create)) - - -def question_cmd(args: argparse.Namespace, config: Config) -> None: - """ - Handler for the 'question' command. - """ - chat = ChatDB.from_dir(cache_path=Path('.'), - db_path=Path(config.db)) - # if it's a new question, create and store it immediately - if args.ask or args.create: - # FIXME: add sources to the question - message = Message(question=Question(args.question), - tags=args.ouput_tags, # FIXME - ai=args.ai, - model=args.model) - chat.add_to_cache([message]) - if args.create: - return - - # create the correct AI instance - ai: AI = create_ai(args, config) - if args.ask: - response: AIResponse = ai.request(message, - chat, - args.num_answers, # FIXME - args.otags) # FIXME - assert response - # TODO: - # * add answer to the message above (and create - # more messages for any additional answers) - pass - elif args.repeat: - lmessage = chat.latest_message() - assert lmessage - # TODO: repeat either the last question or the - # one(s) given in 'args.repeat' (overwrite - # existing ones if 'args.overwrite' is True) - pass - elif args.process: - # TODO: process either all questions without an - # answer or the one(s) given in 'args.process' - pass - - -def hist_cmd(args: argparse.Namespace, config: Config) -> None: - """ - Handler for the 'hist' command. - """ - - mfilter = MessageFilter(tags_or=args.or_tags, - tags_and=args.and_tags, - tags_not=args.exclude_tags, - question_contains=args.question, - answer_contains=args.answer) - chat = ChatDB.from_dir(Path('.'), - Path(config.db), - mfilter=mfilter) - chat.print(args.source_code_only, - args.with_tags, - args.with_files) - - -def print_cmd(args: argparse.Namespace, config: Config) -> None: - """ - Handler for the 'print' command. - """ - fname = Path(args.file) - try: - message = Message.from_file(fname) - if message: - print(message.to_str(source_code_only=args.source_code_only)) - except MessageError: - print(f"File is not a valid message: {args.file}") - sys.exit(1) - - def create_parser() -> argparse.ArgumentParser: parser = argparse.ArgumentParser( description="ChatMastermind is a Python application that automates conversation with AI") diff --git a/setup.py b/setup.py index 8484629..a311605 100644 --- a/setup.py +++ b/setup.py @@ -12,7 +12,7 @@ setup( long_description=long_description, long_description_content_type="text/markdown", url="https://github.com/ok2/ChatMastermind", - packages=find_packages() + ["chatmastermind.ais"], + packages=find_packages() + ["chatmastermind.ais", "chatmastermind.commands"], classifiers=[ "Development Status :: 3 - Alpha", "Environment :: Console",