genai[refactoring]: Remove Pillow support, adjust dependencies, and c…

…lean up unused code (#603)

libs/genai/README.md

@@ -8,12 +8,6 @@ This package contains the LangChain integrations for Gemini through their genera
 pip install -U langchain-google-genai
 ```
 
-### Image utilities
-To use image utility methods, like loading images from GCS urls, install with extras group 'images':
-
-```bash
-pip install -e "langchain-google-genai[images]"
-```
 
 ## Chat Models
 
@@ -61,7 +55,6 @@ The value of `image_url` can be any of the following:
 - A public image URL
 - An accessible gcs file (e.g., "gcs://path/to/file.png")
 - A base64 encoded image (e.g., `data:image/png;base64,abcd124`)
-- A PIL image
 
 
 

libs/genai/langchain_google_genai/chat_models.py

@@ -5,7 +5,6 @@
 import logging
 import uuid
 import warnings
-from io import BytesIO
 from operator import itemgetter
 from typing import (
     Any,
@@ -22,7 +21,6 @@
     Union,
     cast,
 )
-from urllib.parse import urlparse
 
 import google.api_core
 
@@ -114,16 +112,6 @@
 
 from . import _genai_extension as genaix
 
-IMAGE_TYPES: Tuple = ()
-try:
-    import PIL
-    from PIL.Image import Image
-
-    IMAGE_TYPES = IMAGE_TYPES + (Image,)
-except ImportError:
-    PIL = None  # type: ignore
-    Image = None  # type: ignore
-
 logger = logging.getLogger(__name__)
 
 
@@ -245,46 +233,6 @@ def _is_openai_parts_format(part: dict) -> bool:
     return "type" in part
 
 
-def _is_vision_model(model: str) -> bool:
-    return "vision" in model
-
-
-def _is_url(s: str) -> bool:
-    try:
-        result = urlparse(s)
-        return all([result.scheme, result.netloc])
-    except Exception as e:
-        logger.debug(f"Unable to parse URL: {e}")
-        return False
-
-
-def _is_b64(s: str) -> bool:
-    return s.startswith("data:image")
-
-
-def _load_image_from_gcs(path: str, project: Optional[str] = None) -> Image:
-    try:
-        from google.cloud import storage  # type: ignore[attr-defined]
-    except ImportError:
-        raise ImportError(
-            "google-cloud-storage is required to load images from GCS."
-            " Install it with `pip install google-cloud-storage`"
-        )
-    if PIL is None:
-        raise ImportError(
-            "PIL is required to load images. Please install it "
-            "with `pip install pillow`"
-        )
-
-    gcs_client = storage.Client(project=project)
-    pieces = path.split("/")
-    blobs = list(gcs_client.list_blobs(pieces[2], prefix="/".join(pieces[3:])))
-    if len(blobs) > 1:
-        raise ValueError(f"Found more than one candidate for {path}!")
-    img_bytes = blobs[0].download_as_bytes()
-    return PIL.Image.open(BytesIO(img_bytes))
-
-
 def _convert_to_parts(
     raw_content: Union[str, Sequence[Union[str, dict]]],
 ) -> List[Part]:

libs/genai/pyproject.toml

@@ -14,12 +14,8 @@ license = "MIT"
 python = ">=3.9,<4.0"
 langchain-core = ">=0.3.15,<0.4"
 google-generativeai = "^0.8.0"
-pillow = { version = "^10.1.0", optional = true }
 pydantic = ">=2,<3"
 
-[tool.poetry.extras]
-images = ["pillow"]
-
 [tool.poetry.group.test]
 optional = true
 
@@ -48,7 +44,7 @@ codespell = "^2.2.0"
 optional = true
 
 [tool.poetry.group.test_integration.dependencies]
-pillow = "^10.1.0"
+pytest = "^7.3.0"
 
 
 [tool.poetry.group.lint]
@@ -62,7 +58,6 @@ ruff = "^0.1.5"
 mypy = "^1.10"
 types-requests = "^2.28.11.5"
 types-google-cloud-ndb = "^2.2.0.1"
-types-pillow = "^10.1.0.2"
 types-protobuf = "^4.24.0.20240302"
 numpy = "^1.26.2"
 
@@ -71,9 +66,7 @@ numpy = "^1.26.2"
 optional = true
 
 [tool.poetry.group.dev.dependencies]
-pillow = "^10.1.0"
 types-requests = "^2.31.0.10"
-types-pillow = "^10.1.0.2"
 types-google-cloud-ndb = "^2.2.0.1"
 
 [tool.ruff.lint]