From 17de0b99678381fa7e0fac9285d71bc26c649a67 Mon Sep 17 00:00:00 2001 From: Oleksandr Kozachuk Date: Mon, 11 Sep 2023 13:17:59 +0200 Subject: [PATCH] Remove old code. --- chatmastermind/main.py | 105 +---------------------------------------- 1 file changed, 1 insertion(+), 104 deletions(-) diff --git a/chatmastermind/main.py b/chatmastermind/main.py index 1a375d0..99aca09 100755 --- a/chatmastermind/main.py +++ b/chatmastermind/main.py @@ -18,110 +18,7 @@ from .commands.print import print_cmd def tags_completer(prefix: str, parsed_args: Any, **kwargs: Any) -> list[str]: config = Config.from_file(parsed_args.config) - return get_tags_unique(config, 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) - if args.source_code_only: - display_source_code(data['answer']) - elif args.answer: - print(data['answer'].strip()) - elif args.question: - print(data['question'].strip()) - else: - print(dump_data(data).strip()) + return list(Message.tags_from_dir(Path(config.db), prefix=prefix)) def create_parser() -> argparse.ArgumentParser: