langchain-google-genai 2.0.3__py3-none-any.whl → 2.0.5__py3-none-any.whl

langchain_google_genai/_image_utils.py

@@ -25,7 +25,6 @@ class ImageBytesLoader:
 
     Currently supported:
         - B64 Encoded image string
-        - Local file path
         - URL
     """
 
@@ -35,7 +34,6 @@ class ImageBytesLoader:
         Args:
             image_string: Can be either:
                     - B64 Encoded image string
-                    - Local file path
                     - URL
 
         Returns:
@@ -51,12 +49,17 @@ class ImageBytesLoader:
             return self._bytes_from_url(image_string)
 
         if route == Route.LOCAL_FILE:
+            raise ValueError(
+                "Loading from local files is no longer supported for security reasons. "
+                "Please pass in images as Google Cloud Storage URI, "
+                "b64 encoded image string (data:image/...), or valid image url."
+            )
             return self._bytes_from_file(image_string)
 
         raise ValueError(
             "Image string must be one of: Google Cloud Storage URI, "
-            "b64 encoded image string (data:image/...), valid image url, "
-            f"or existing local image file. Instead got '{image_string}'."
+            "b64 encoded image string (data:image/...), or valid image url."
+            f"Instead got '{image_string}'."
         )
 
     def load_part(self, image_string: str) -> Part:
@@ -65,7 +68,6 @@ class ImageBytesLoader:
         Args:
             image_string: Can be either:
                     - B64 Encoded image string
-                    - Local file path
                     - URL
         """
         route = self._route(image_string)
@@ -77,7 +79,12 @@ class ImageBytesLoader:
             bytes_ = self._bytes_from_url(image_string)
 
         if route == Route.LOCAL_FILE:
-            bytes_ = self._bytes_from_file(image_string)
+            msg = (
+                "Loading from local files is no longer supported for security reasons. "
+                "Please specify images as Google Cloud Storage URI, "
+                "b64 encoded image string (data:image/...), or valid image url."
+            )
+            raise ValueError(msg)
 
         inline_data: Dict[str, Any] = {"data": bytes_}
         mime_type, _ = mimetypes.guess_type(image_string)
@@ -98,8 +105,8 @@ class ImageBytesLoader:
 
         raise ValueError(
             "Image string must be one of: "
-            "b64 encoded image string (data:image/...), valid image url, "
-            f"or existing local image file. Instead got '{image_string}'."
+            "b64 encoded image string (data:image/...) or valid image url."
+            f" Instead got '{image_string}'."
         )
 
     def _bytes_from_b64(self, base64_image: str) -> bytes:
@@ -121,19 +128,6 @@ class ImageBytesLoader:
 
         raise ValueError(f"Error in b64 encoded image. Must follow pattern: {pattern}")
 
-    def _bytes_from_file(self, file_path: str) -> bytes:
-        """Gets image bytes from a local file path.
-
-        Args:
-            file_path: Existing file path.
-
-        Returns:
-            Image bytes
-        """
-        with open(file_path, "rb") as image_file:
-            image_bytes = image_file.read()
-        return image_bytes
-
     def _bytes_from_url(self, url: str) -> bytes:
         """Gets image bytes from a public url.
 

langchain_google_genai/chat_models.py

@@ -1,13 +1,10 @@
 from __future__ import annotations
 
 import asyncio
-import base64
 import json
 import logging
-import os
 import uuid
 import warnings
-from io import BytesIO
 from operator import itemgetter
 from typing import (
     Any,
@@ -24,13 +21,11 @@ from typing import (
     Union,
     cast,
 )
-from urllib.parse import urlparse
 
 import google.api_core
 
 # TODO: remove ignore once the google package is published with types
 import proto  # type: ignore[import]
-import requests
 from google.ai.generativelanguage_v1beta import (
     GenerativeServiceAsyncClient as v1betaGenerativeServiceAsyncClient,
 )
@@ -117,16 +112,6 @@ from langchain_google_genai.llms import _BaseGoogleGenerativeAI
 
 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__)
 
 
@@ -248,75 +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 _url_to_pil(image_source: str) -> Image:
-    if PIL is None:
-        raise ImportError(
-            "PIL is required to load images. Please install it "
-            "with `pip install pillow`"
-        )
-    try:
-        if isinstance(image_source, IMAGE_TYPES):
-            return image_source  # type: ignore[return-value]
-        elif _is_url(image_source):
-            if image_source.startswith("gs://"):
-                return _load_image_from_gcs(image_source)
-            response = requests.get(image_source)
-            response.raise_for_status()
-            return PIL.Image.open(BytesIO(response.content))
-        elif _is_b64(image_source):
-            _, encoded = image_source.split(",", 1)
-            data = base64.b64decode(encoded)
-            return PIL.Image.open(BytesIO(data))
-        elif os.path.exists(image_source):
-            return PIL.Image.open(image_source)
-        else:
-            raise ValueError(
-                "The provided string is not a valid URL, base64, or file path."
-            )
-    except Exception as e:
-        raise ValueError(f"Unable to process the provided image source: {e}")
-
-
 def _convert_to_parts(
     raw_content: Union[str, Sequence[Union[str, dict]]],
 ) -> List[Part]:
@@ -400,8 +316,17 @@ def _parse_chat_history(
             continue
         elif isinstance(message, AIMessage):
             role = "model"
-            raw_function_call = message.additional_kwargs.get("function_call")
-            if raw_function_call:
+            if message.tool_calls:
+                parts = []
+                for tool_call in message.tool_calls:
+                    function_call = FunctionCall(
+                        {
+                            "name": tool_call["name"],
+                            "args": tool_call["args"],
+                        }
+                    )
+                    parts.append(Part(function_call=function_call))
+            elif raw_function_call := message.additional_kwargs.get("function_call"):
                 function_call = FunctionCall(
                     {
                         "name": raw_function_call["name"],

langchain_google_genai/llms.py

@@ -310,10 +310,21 @@ class GoogleGenerativeAI(_BaseGoogleGenerativeAI, BaseLLM):
                     generation_config=generation_config,
                     safety_settings=kwargs.pop("safety_settings", None),
                 )
+                generation_info = None
+                if res.usage_metadata is not None:
+                    generation_info = {
+                        "usage_metadata": res.to_dict().get("usage_metadata")
+                    }
+
                 candidates = [
                     "".join([p.text for p in c.content.parts]) for c in res.candidates
                 ]
-                generations.append([Generation(text=c) for c in candidates])
+                generations.append(
+                    [
+                        Generation(text=c, generation_info=generation_info)
+                        for c in candidates
+                    ]
+                )
             else:
                 res = _completion_with_retry(
                     self,

{langchain_google_genai-2.0.3.dist-info → langchain_google_genai-2.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-google-genai
-Version: 2.0.3
+Version: 2.0.5
 Summary: An integration package connecting Google's genai package and LangChain
 Home-page: https://github.com/langchain-ai/langchain-google
 License: MIT
@@ -11,10 +11,8 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Provides-Extra: images
 Requires-Dist: google-generativeai (>=0.8.0,<0.9.0)
-Requires-Dist: langchain-core (>=0.3.13,<0.4)
-Requires-Dist: pillow (>=10.1.0,<11.0.0) ; extra == "images"
+Requires-Dist: langchain-core (>=0.3.15,<0.4)
 Requires-Dist: pydantic (>=2,<3)
 Project-URL: Repository, https://github.com/langchain-ai/langchain-google
 Project-URL: Source Code, https://github.com/langchain-ai/langchain-google/tree/main/libs/genai
@@ -30,12 +28,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
 
@@ -82,9 +74,7 @@ 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 local file path
 - A base64 encoded image (e.g., `data:image/png;base64,abcd124`)
-- A PIL image
 
 
 

{langchain_google_genai-2.0.3.dist-info → langchain_google_genai-2.0.5.dist-info}/RECORD

@@ -3,14 +3,14 @@ langchain_google_genai/_common.py,sha256=ASlwE8hEbvOm55BVF_D4rf2nl7RYsnpsi5xbM6D
 langchain_google_genai/_enums.py,sha256=KLPmxS1K83K4HjBIXFaXoL_sFEOv8Hq-2B2PDMKyDgo,197
 langchain_google_genai/_function_utils.py,sha256=ksKgLoKPhKzd3QLLgYojuq4tOegoEL-oHPncARPds3w,16891
 langchain_google_genai/_genai_extension.py,sha256=81a4ly5ZHlqMf37uJfdB8K41qE6J5ujLnbUypIfFf2o,20775
-langchain_google_genai/_image_utils.py,sha256=-0XgCMdYkvrIktFvUpy-2GPbFgfSVKZICawB2hiJzus,4999
-langchain_google_genai/chat_models.py,sha256=pObFpTYKh-of52BrAbpVv1UIDp_qZ8XRTDRhPYgNZ2M,56506
+langchain_google_genai/_image_utils.py,sha256=_3knaOcAlI8AxHO8NyqNIqGbtldWRBU5wm5trdyhTzY,5033
+langchain_google_genai/chat_models.py,sha256=X5CHWmRQjoV1l6AN0VbVX5cSYyW5FAoyS9jWFBqhig8,54232
 langchain_google_genai/embeddings.py,sha256=PBJlcRExfegD2V9hnKQHW99fJJGsbPbmMW1TfQtPmqU,10132
 langchain_google_genai/genai_aqa.py,sha256=qB6h3-BSXqe0YLR3eeVllYzmNKK6ofI6xJLdBahUVZo,4300
 langchain_google_genai/google_vector_store.py,sha256=4wvhIiOmc3Fo046FyafPmT9NBCLek-9bgluvuTfrbpQ,16148
-langchain_google_genai/llms.py,sha256=lXh0Se-kbBJlHrw0kR1DRsZw_bet1thKZ2HskvT8itU,14221
+langchain_google_genai/llms.py,sha256=EPUgkz5aqKOyKbztT7br8w60Uo5D_X_bF5qP-zd6iLs,14593
 langchain_google_genai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langchain_google_genai-2.0.3.dist-info/LICENSE,sha256=DppmdYJVSc1jd0aio6ptnMUn5tIHrdAhQ12SclEBfBg,1072
-langchain_google_genai-2.0.3.dist-info/METADATA,sha256=OvxxVXvYGo3Qa7L8y1o06VX8JCR31WQSjOmpclTEhKA,3851
-langchain_google_genai-2.0.3.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-langchain_google_genai-2.0.3.dist-info/RECORD,,
+langchain_google_genai-2.0.5.dist-info/LICENSE,sha256=DppmdYJVSc1jd0aio6ptnMUn5tIHrdAhQ12SclEBfBg,1072
+langchain_google_genai-2.0.5.dist-info/METADATA,sha256=mE8t2CijamvFdZk3CKWgZW5SNvaB2LU8rl9jO-7K_nU,3551
+langchain_google_genai-2.0.5.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+langchain_google_genai-2.0.5.dist-info/RECORD,,