fix: move IPython import (#304)

* fix: move IPython import

* fix the mock

Author: anakin87

haystack_experimental/dataclasses/image_content.py

@@ -13,7 +13,6 @@
 from haystack.components.fetchers.link_content import LinkContentFetcher
 from haystack.lazy_imports import LazyImport
 from haystack.utils import is_in_jupyter
-from IPython.display import display
 
 from haystack_experimental.components.image_converters.image_utils import MIME_TO_FORMAT
 
@@ -85,6 +84,8 @@ def show(self) -> None:
         image = Image.open(image_bytes)
 
         if is_in_jupyter():
+            # ipython is not a core dependency so we cannot import it at the module level
+            from IPython.display import display
             display(image)
         else:
             image.show()

test/dataclasses/test_image_content.py

@@ -42,7 +42,7 @@ def test_image_content_show_in_jupyter(test_files_path):
     )
 
     with patch("haystack_experimental.dataclasses.image_content.is_in_jupyter", return_value=True), \
-        patch("haystack_experimental.dataclasses.image_content.display") as mock_display:
+        patch("IPython.display.display") as mock_display:
         image_content.show()
 
         mock_display.assert_called_once()