langchain-google-genai 2.0.5__tar.gz → 2.0.7__tar.gz

{langchain_google_genai-2.0.5 → langchain_google_genai-2.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-google-genai
-Version: 2.0.5
+Version: 2.0.7
 Summary: An integration package connecting Google's genai package and LangChain
 Home-page: https://github.com/langchain-ai/langchain-google
 License: MIT
@@ -11,6 +11,7 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: filetype (>=1.2.0,<2.0.0)
 Requires-Dist: google-generativeai (>=0.8.0,<0.9.0)
 Requires-Dist: langchain-core (>=0.3.15,<0.4)
 Requires-Dist: pydantic (>=2,<3)

{langchain_google_genai-2.0.5 → langchain_google_genai-2.0.7}/langchain_google_genai/_function_utils.py

@@ -342,8 +342,6 @@ def _get_items_from_schema(schema: Union[Dict, List, str]) -> Dict[str, Any]:
         items["type_"] = _get_type_from_schema(schema)
         if items["type_"] == glm.Type.OBJECT and "properties" in schema:
             items["properties"] = _get_properties_from_schema_any(schema["properties"])
-        if "title" in schema:
-            items["title"] = schema
         if "title" in schema or "description" in schema:
             items["description"] = (
                 schema.get("description") or schema.get("title") or ""

{langchain_google_genai-2.0.5 → langchain_google_genai-2.0.7}/langchain_google_genai/_image_utils.py

@@ -8,6 +8,7 @@ from enum import Enum
 from typing import Any, Dict
 from urllib.parse import urlparse
 
+import filetype  # type: ignore[import]
 import requests
 from google.ai.generativelanguage_v1beta.types import Part
 
@@ -87,7 +88,13 @@ class ImageBytesLoader:
             raise ValueError(msg)
 
         inline_data: Dict[str, Any] = {"data": bytes_}
+
         mime_type, _ = mimetypes.guess_type(image_string)
+        if not mime_type:
+            kind = filetype.guess(bytes_)
+            if kind:
+                mime_type = kind.mime
+
         if mime_type:
             inline_data["mime_type"] = mime_type
 

{langchain_google_genai-2.0.5 → langchain_google_genai-2.0.7}/langchain_google_genai/chat_models.py

@@ -579,9 +579,9 @@ class ChatGoogleGenerativeAI(_BaseGoogleGenerativeAI, BaseChatModel):
     Instantiation:
         To use, you must have either:
 
-            1. The ``GOOGLE_API_KEY``` environment variable set with your API key, or
-            2. Pass your API key using the google_api_key kwarg to the ChatGoogle
-               constructor.
+            1. The ``GOOGLE_API_KEY`` environment variable set with your API key, or
+            2. Pass your API key using the google_api_key kwarg
+            to the ChatGoogleGenerativeAI constructor.
 
         .. code-block:: python
 
@@ -1374,7 +1374,11 @@ class ChatGoogleGenerativeAI(_BaseGoogleGenerativeAI, BaseChatModel):
 
     @property
     def _supports_tool_choice(self) -> bool:
-        return "gemini-1.5-pro" in self.model or "gemini-1.5-flash" in self.model
+        return (
+            "gemini-1.5-pro" in self.model
+            or "gemini-1.5-flash" in self.model
+            or "gemini-2" in self.model
+        )
 
 
 def _get_tool_name(

{langchain_google_genai-2.0.5 → langchain_google_genai-2.0.7}/langchain_google_genai/embeddings.py

@@ -27,9 +27,9 @@ class GoogleGenerativeAIEmbeddings(BaseModel, Embeddings):
 
     To use, you must have either:
 
-        1. The ``GOOGLE_API_KEY``` environment variable set with your API key, or
-        2. Pass your API key using the google_api_key kwarg to the ChatGoogle
-           constructor.
+        1. The ``GOOGLE_API_KEY`` environment variable set with your API key, or
+        2. Pass your API key using the google_api_key kwarg
+        to the GoogleGenerativeAIEmbeddings constructor.
 
     Example:
         .. code-block:: python

{langchain_google_genai-2.0.5 → langchain_google_genai-2.0.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langchain-google-genai"
-version = "2.0.5"
+version = "2.0.7"
 description = "An integration package connecting Google's genai package and LangChain"
 authors = []
 readme = "README.md"
@@ -15,6 +15,7 @@ python = ">=3.9,<4.0"
 langchain-core = ">=0.3.15,<0.4"
 google-generativeai = "^0.8.0"
 pydantic = ">=2,<3"
+filetype = "^1.2.0"
 
 [tool.poetry.group.test]
 optional = true