question_cmd: fixed AI and model arguments when repeating messages

chatmastermind/commands/question.py

@@ -105,11 +105,21 @@ def make_request(ai: AI, chat: ChatDB, message: Message, args: argparse.Namespac
         print(response.tokens)
 
 
-def repeat_messages(messages: list[Message], ai: AI, chat: ChatDB, args: argparse.Namespace) -> None:
+def repeat_messages(messages: list[Message], chat: ChatDB, args: argparse.Namespace, config: Config) -> None:
     """
     Repeat the given messages using the given arguments.
     """
+    ai: AI
     for msg in messages:
+        ai_args = args
+        # if AI or model have not been specified, use those from the original message
+        if args.AI is None or args.model is None:
+            ai_args = args.copy()
+            if args.AI is None and msg.ai is not None:
+                ai_args.AI = msg.ai
+            if args.model is None and msg.model is not None:
+                ai_args.model = msg.model
+        ai = create_ai(ai_args, config)
         print(f"--------- Repeating message '{msg.msg_id()}': ---------")
         # overwrite the latest message if requested or empty
         # -> but not if it's in the DB!
@@ -139,11 +149,10 @@ def question_cmd(args: argparse.Namespace, config: Config) -> None:
         message = create_message(chat, args)
     if args.create:
         return
-    # create the correct AI instance
-    ai: AI = create_ai(args, config)
 
     # === ASK ===
     if args.ask:
+        ai: AI = create_ai(args, config)
         make_request(ai, chat, message, args)
     # === REPEAT ===
     elif args.repeat is not None:
@@ -158,7 +167,7 @@ def question_cmd(args: argparse.Namespace, config: Config) -> None:
         # repeat given message(s)
         else:
             repeat_msgs = chat.msg_find(args.repeat, loc='disk')
-        repeat_messages(repeat_msgs, ai, chat, args)
+        repeat_messages(repeat_msgs, chat, args, config)
     # === PROCESS ===
     elif args.process is not None:
         # TODO: process either all questions without an

tests/test_question_cmd.py

@@ -282,6 +282,9 @@ class TestQuestionCmd(TestQuestionCmdBase):
         # exclude '.next'
         return sorted([f for f in Path(tmp_dir.name).glob('*.[ty]*')])
 
+
+class TestQuestionCmdAsk(TestQuestionCmd):
+
     @mock.patch('chatmastermind.commands.question.create_ai')
     def test_ask_single_answer(self, mock_create_ai: MagicMock) -> None:
         """
@@ -370,6 +373,9 @@ class TestQuestionCmd(TestQuestionCmdBase):
         self.assertEqual(len(self.message_list(self.cache_dir)), 1)
         self.assert_messages_equal(cached_msg, [expected_question])
 
+
+class TestQuestionCmdRepeat(TestQuestionCmd):
+
     @mock.patch('chatmastermind.commands.question.create_ai')
     def test_repeat_single_question(self, mock_create_ai: MagicMock) -> None:
         """
@@ -511,3 +517,44 @@ class TestQuestionCmd(TestQuestionCmdBase):
         cached_msg = chat.msg_gather(loc='cache')
         self.assertEqual(len(self.message_list(self.cache_dir)), 2)
         self.assert_messages_equal(cached_msg, expected_responses)
+
+    @mock.patch('chatmastermind.commands.question.create_ai')
+    def test_repeat_multiple_questions(self, mock_create_ai: MagicMock) -> None:
+        """
+        Repeat multiple  questions.
+        """
+        # 1. create some questions / messages
+        # cached message without an answer
+        message1 = Message(Question('Question 1'),
+                           ai='foo',
+                           model='bla',
+                           file_path=Path(self.cache_dir.name) / '0001.txt')
+        # cached message with an answer
+        message2 = Message(Question('Question 2'),
+                           Answer('Answer 2'),
+                           ai='openai',
+                           model='gpt-3.5-turbo',
+                           file_path=Path(self.cache_dir.name) / '0002.txt')
+        # DB message without an answer
+        message3 = Message(Question('Question 3'),
+                           ai='openai',
+                           model='gpt-3.5-turbo',
+                           file_path=Path(self.db_dir.name) / '0003.txt')
+        message1.to_file()
+        message2.to_file()
+        message3.to_file()
+        # chat = ChatDB.from_dir(Path(self.cache_dir.name),
+        #                        Path(self.db_dir.name))
+
+        # 2. repeat all three questions (without overwriting)
+        self.args.ask = None
+        self.args.repeat = ['0001', '0002', '0003']
+        self.args.overwrite = False
+        question_cmd(self.args, self.config)
+        # two new files should be in the cache directory
+        # * the repeated cached message with answer
+        # * the repeated DB message
+        # -> the cached message wihtout answer should be overwritten
+        self.assertEqual(len(self.message_list(self.cache_dir)), 4)
+        self.assertEqual(len(self.message_list(self.db_dir)), 1)
+        # FIXME: also compare actual content!