Add PngImagePlugin to imports. (#438)

google-gemini · Jul 8, 2024 · fb033db · fb033db
1 parent 85ee8c1
commit fb033db
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/google/generativeai/types/content_types.py b/google/generativeai/types/content_types.py
@@ -30,13 +30,15 @@
 
 if typing.TYPE_CHECKING:
     import PIL.Image
+    import PIL.PngImagePlugin
     import IPython.display
 
     IMAGE_TYPES = (PIL.Image.Image, IPython.display.Image)
 else:
     IMAGE_TYPES = ()
     try:
         import PIL.Image
+        import PIL.PngImagePlugin
 
         IMAGE_TYPES = IMAGE_TYPES + (PIL.Image.Image,)
     except ImportError: