ara-cli 0.1.10.1__py3-none-any.whl → 0.1.10.5__py3-none-any.whl

tests/test_prompt_handler.py

@@ -698,16 +698,24 @@ class TestArtefactAndTemplateHandling:
 
         mock_collect.return_value = ("### GIVENS\ncontent", [{"type": "image_url"}])
         
-        final_message_list = [{'role': 'user', 'content': ['### GIVENS\ncontent', {'type': 'image_url'}]}]
-        mock_append_images.return_value = final_message_list
+        # append_images_to_message returns a single dict, not a list of dicts.
+        returned_message_dict = {'role': 'user', 'content': ['### GIVENS\ncontent', {'type': 'image_url'}]}
+        mock_append_images.return_value = returned_message_dict
 
         mock_send.return_value = iter([MagicMock(choices=[MagicMock(delta=MagicMock(content="llm response"))])])
 
         prompt_handler.create_and_send_custom_prompt(self.mock_classifier, self.mock_param)
         
         mock_collect.assert_called_once()
-        mock_append_images.assert_called_once_with([{'role': 'user', 'content': '### GIVENS\ncontent'}], [{'type': 'image_url'}])
-        mock_send.assert_called_once_with(final_message_list)
+
+        # Assert that append_images_to_message was called with a single dict (the bug fix)
+        mock_append_images.assert_called_once_with(
+            {'role': 'user', 'content': '### GIVENS\ncontent'},
+            [{'type': 'image_url'}]
+        )
+
+        # Assert that send_prompt was called with a list containing the dict returned from append_images_to_message
+        mock_send.assert_called_once_with([returned_message_dict])
 
         log_file = self.root / "ara" / self.mock_classifier / f"{self.mock_param}.data" / f"{self.mock_classifier}.prompt_log.md"
         assert "llm response" in log_file.read_text()