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..9c56ced --- /dev/null +++ b/chatmastermind/commands/question.py @@ -0,0 +1,57 @@ +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 create_message(chat: ChatDB, args: argparse.Namespace) -> Message: + """ + Creates (and writes) a new message from the given arguments. + """ + # FIXME: add sources to the question + message = Message(question=Question(args.question), + tags=args.output_tags, # FIXME + ai=args.ai, + model=args.model) + chat.add_to_cache([message]) + return message + + +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: + message = create_message(chat, args) + 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 58ce9ed..02cdffd 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]: @@ -136,20 +138,28 @@ def create_parser() -> argparse.ArgumentParser: # a parent parser for all commands that support tag selection tag_parser = argparse.ArgumentParser(add_help=False) tag_arg = tag_parser.add_argument('-t', '--or-tags', nargs='+', - help='List of tag names (one must match)', metavar='OTAGS') + help='List of tags (one must match)', metavar='OTAGS') tag_arg.completer = tags_completer # type: ignore atag_arg = tag_parser.add_argument('-k', '--and-tags', nargs='+', - help='List of tag names (all must match)', metavar='ATAGS') + help='List of tags (all must match)', metavar='ATAGS') atag_arg.completer = tags_completer # type: ignore etag_arg = tag_parser.add_argument('-x', '--exclude-tags', nargs='+', - help='List of tag names to exclude', metavar='XTAGS') + help='List of tags to exclude', metavar='XTAGS') etag_arg.completer = tags_completer # type: ignore otag_arg = tag_parser.add_argument('-o', '--output-tags', nargs='+', - help='List of output tag names, default is input', metavar='OUTTAGS') + help='List of output tags (default: use input tags)', metavar='OUTTAGS') otag_arg.completer = tags_completer # type: ignore + # a parent parser for all commands that support AI configuration + ai_parser = argparse.ArgumentParser(add_help=False) + ai_parser.add_argument('-A', '--AI', help='AI ID to use') + ai_parser.add_argument('-M', '--model', help='Model to use') + ai_parser.add_argument('-n', '--num-answers', help='Number of answers to request', type=int, default=1) + ai_parser.add_argument('-m', '--max-tokens', help='Max. nr. of tokens', type=int) + ai_parser.add_argument('-T', '--temperature', help='Temperature value', type=float) + # 'question' command parser - question_cmd_parser = cmdparser.add_parser('question', parents=[tag_parser], + question_cmd_parser = cmdparser.add_parser('question', parents=[tag_parser, ai_parser], help="ask, create and process questions.", aliases=['q']) question_cmd_parser.set_defaults(func=question_cmd) @@ -160,12 +170,6 @@ def create_parser() -> argparse.ArgumentParser: question_group.add_argument('-p', '--process', nargs='*', help='Process existing questions') question_cmd_parser.add_argument('-O', '--overwrite', help='Overwrite existing messages when repeating them', action='store_true') - question_cmd_parser.add_argument('-m', '--max-tokens', help='Max tokens to use', type=int) - question_cmd_parser.add_argument('-T', '--temperature', help='Temperature to use', type=float) - question_cmd_parser.add_argument('-A', '--AI', help='AI to use') - question_cmd_parser.add_argument('-M', '--model', help='Model to use') - question_cmd_parser.add_argument('-n', '--num-answers', help='Number of answers to produce', type=int, - default=1) question_cmd_parser.add_argument('-s', '--source', nargs='+', help='Source add content of a file to the query') question_cmd_parser.add_argument('-S', '--source-code-only', help='Add pure source code to the chat history', action='store_true') @@ -213,18 +217,10 @@ def create_parser() -> argparse.ArgumentParser: aliases=['p']) print_cmd_parser.set_defaults(func=print_cmd) print_cmd_parser.add_argument('-f', '--file', help='File to print', required=True) -<<<<<<< HEAD print_cmd_modes = print_cmd_parser.add_mutually_exclusive_group() print_cmd_modes.add_argument('-q', '--question', help='Print only question', action='store_true') print_cmd_modes.add_argument('-a', '--answer', help='Print only answer', action='store_true') print_cmd_modes.add_argument('-S', '--only-source-code', help='Print only source code', action='store_true') -||||||| parent of bf1cbff (cmm: the 'print' command now uses 'Message.from_file()') - print_cmd_parser.add_argument('-S', '--source-code-only', help='Print only source code', - action='store_true') -======= - print_cmd_parser.add_argument('-S', '--source-code-only', help='Print source code only (from the answer, if available)', - action='store_true') ->>>>>>> bf1cbff (cmm: the 'print' command now uses 'Message.from_file()') argcomplete.autocomplete(parser) return parser 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", diff --git a/tests/test_ai_factory.py b/tests/test_ai_factory.py new file mode 100644 index 0000000..d63970e --- /dev/null +++ b/tests/test_ai_factory.py @@ -0,0 +1,48 @@ +import argparse +import unittest +from unittest.mock import MagicMock +from chatmastermind.ai_factory import create_ai +from chatmastermind.configuration import Config +from chatmastermind.ai import AIError +from chatmastermind.ais.openai import OpenAI + + +class TestCreateAI(unittest.TestCase): + def setUp(self) -> None: + self.args = MagicMock(spec=argparse.Namespace) + self.args.ai = 'default' + self.args.model = None + self.args.max_tokens = None + self.args.temperature = None + + def test_create_ai_from_args(self) -> None: + # Create an AI with the default configuration + config = Config() + self.args.ai = 'default' + ai = create_ai(self.args, config) + self.assertIsInstance(ai, OpenAI) + + def test_create_ai_from_default(self) -> None: + self.args.ai = None + # Create an AI with the default configuration + config = Config() + ai = create_ai(self.args, config) + self.assertIsInstance(ai, OpenAI) + + def test_create_empty_ai_error(self) -> None: + self.args.ai = None + # Create Config with empty AIs + config = Config() + config.ais = {} + # Call create_ai function and assert that it raises AIError + with self.assertRaises(AIError): + create_ai(self.args, config) + + def test_create_unsupported_ai_error(self) -> None: + # Mock argparse.Namespace with ai='invalid_ai' + self.args.ai = 'invalid_ai' + # Create default Config + config = Config() + # Call create_ai function and assert that it raises AIError + with self.assertRaises(AIError): + create_ai(self.args, config)