langchain-google-genai 2.0.6__tar.gz → 2.0.8__tar.gz

{langchain_google_genai-2.0.6 → langchain_google_genai-2.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-google-genai
-Version: 2.0.6
+Version: 2.0.8
 Summary: An integration package connecting Google's genai package and LangChain
 Home-page: https://github.com/langchain-ai/langchain-google
 License: MIT
@@ -13,7 +13,7 @@ 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: langchain-core (>=0.3.27,<0.4.0)
 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

{langchain_google_genai-2.0.6 → langchain_google_genai-2.0.8}/langchain_google_genai/_function_utils.py

@@ -315,10 +315,26 @@ def _get_properties_from_schema(schema: Dict) -> Dict[str, Any]:
         if properties_item.get("type_") == glm.Type.ARRAY and v.get("items"):
             properties_item["items"] = _get_items_from_schema_any(v.get("items"))
 
-        if properties_item.get("type_") == glm.Type.OBJECT and v.get("properties"):
-            properties_item["properties"] = _get_properties_from_schema_any(
-                v.get("properties")
-            )
+        if properties_item.get("type_") == glm.Type.OBJECT:
+            if (
+                v.get("anyOf")
+                and isinstance(v["anyOf"], list)
+                and isinstance(v["anyOf"][0], dict)
+            ):
+                v = v["anyOf"][0]
+            v_properties = v.get("properties")
+            if v_properties:
+                properties_item["properties"] = _get_properties_from_schema_any(
+                    v_properties
+                )
+                if isinstance(v_properties, dict):
+                    properties_item["required"] = [
+                        k for k, v in v_properties.items() if "default" not in v
+                    ]
+            else:
+                # Providing dummy type for object without properties
+                properties_item["type_"] = glm.Type.STRING
+
         if k == "title" and "description" not in properties_item:
             properties_item["description"] = k + " is " + str(v)
 
@@ -342,8 +358,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.6 → langchain_google_genai-2.0.8}/langchain_google_genai/chat_models.py

@@ -419,6 +419,9 @@ def _parse_response_candidate(
     for part in response_candidate.content.parts:
         try:
             text: Optional[str] = part.text
+            # Remove erroneous newline character if present
+            if text is not None:
+                text = text.rstrip("\n")
         except AttributeError:
             text = None
 
@@ -579,9 +582,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 +1377,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.6 → langchain_google_genai-2.0.8}/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.6 → langchain_google_genai-2.0.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langchain-google-genai"
-version = "2.0.6"
+version = "2.0.8"
 description = "An integration package connecting Google's genai package and LangChain"
 authors = []
 readme = "README.md"
@@ -12,7 +12,7 @@ license = "MIT"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<4.0"
-langchain-core = ">=0.3.15,<0.4"
+langchain-core = "^0.3.27"
 google-generativeai = "^0.8.0"
 pydantic = ">=2,<3"
 filetype = "^1.2.0"
@@ -34,6 +34,8 @@ langchain-tests = "0.3.1"
 ignore-words-list = "rouge"
 
 
+
+
 [tool.poetry.group.codespell]
 optional = true
 
@@ -41,6 +43,8 @@ optional = true
 codespell = "^2.2.0"
 
 
+
+
 [tool.poetry.group.test_integration]
 optional = true
 
@@ -48,6 +52,8 @@ optional = true
 pytest = "^7.3.0"
 
 
+
+
 [tool.poetry.group.lint]
 optional = true
 
@@ -55,6 +61,8 @@ optional = true
 ruff = "^0.1.5"
 
 
+
+
 [tool.poetry.group.typing.dependencies]
 mypy = "^1.10"
 types-requests = "^2.28.11.5"
@@ -63,6 +71,8 @@ types-protobuf = "^4.24.0.20240302"
 numpy = "^1.26.2"
 
 
+
+
 [tool.poetry.group.dev]
 optional = true