langchain-google-genai 1.0.9__py3-none-any.whl → 1.0.10__py3-none-any.whl

langchain_google_genai/_function_utils.py

@@ -107,6 +107,13 @@ def _format_json_schema_to_gapic(schema: Dict[str, Any]) -> Dict[str, Any]:
                     pvalue
                 )
             continue
+        elif key == "allOf":
+            if len(value) > 1:
+                logger.warning(
+                    "Only first value for 'allOf' key is supported. "
+                    f"Got {len(value)}, ignoring other than first value!"
+                )
+            return _format_json_schema_to_gapic(value[0])
         elif key in ["type", "_type"]:
             converted_schema["type"] = str(value).upper()
         elif key not in _ALLOWED_SCHEMA_FIELDS_SET:

langchain_google_genai/llms.py

@@ -9,7 +9,7 @@ from langchain_core.callbacks import (
     AsyncCallbackManagerForLLMRun,
     CallbackManagerForLLMRun,
 )
-from langchain_core.language_models import LanguageModelInput
+from langchain_core.language_models import LangSmithParams, LanguageModelInput
 from langchain_core.language_models.llms import BaseLLM, create_base_retry_decorator
 from langchain_core.outputs import Generation, GenerationChunk, LLMResult
 from langchain_core.pydantic_v1 import BaseModel, Field, SecretStr, root_validator
@@ -268,6 +268,16 @@ class GoogleGenerativeAI(_BaseGoogleGenerativeAI, BaseLLM):
 
         return values
 
+    def _get_ls_params(
+        self, stop: Optional[List[str]] = None, **kwargs: Any
+    ) -> LangSmithParams:
+        """Get standard params for tracing."""
+        ls_params = super()._get_ls_params(stop=stop, **kwargs)
+        ls_params["ls_provider"] = "google_genai"
+        if ls_max_tokens := kwargs.get("max_output_tokens", self.max_output_tokens):
+            ls_params["ls_max_tokens"] = ls_max_tokens
+        return ls_params
+
     def _generate(
         self,
         prompts: List[str],

{langchain_google_genai-1.0.9.dist-info → langchain_google_genai-1.0.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-google-genai
-Version: 1.0.9
+Version: 1.0.10
 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
 Provides-Extra: images
 Requires-Dist: google-generativeai (>=0.7.0,<0.8.0)
-Requires-Dist: langchain-core (>=0.2.32,<0.3)
+Requires-Dist: langchain-core (>=0.2.33,<0.3)
 Requires-Dist: pillow (>=10.1.0,<11.0.0) ; extra == "images"
 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-1.0.9.dist-info → langchain_google_genai-1.0.10.dist-info}/RECORD

@@ -1,16 +1,16 @@
 langchain_google_genai/__init__.py,sha256=Oji-S2KYWrku1wyQEskY84IOfY8MfRhujjJ4d7hbsk4,2758
 langchain_google_genai/_common.py,sha256=ASlwE8hEbvOm55BVF_D4rf2nl7RYsnpsi5xbM6DW3Cc,1576
 langchain_google_genai/_enums.py,sha256=KLPmxS1K83K4HjBIXFaXoL_sFEOv8Hq-2B2PDMKyDgo,197
-langchain_google_genai/_function_utils.py,sha256=mlCbVghOCu7B0g9sP0lqo_UT7awqAKbxK_eWrfWVU9E,11319
+langchain_google_genai/_function_utils.py,sha256=gcyKqUyMKMkoLjRGy5NgeeXKwAYhyMhR_HPGlctlAUk,11631
 langchain_google_genai/_genai_extension.py,sha256=ZwNwLV22RSf9LB7FOCLsoHzLlQDF-EQmRNYM1an2uSw,20769
 langchain_google_genai/_image_utils.py,sha256=-0XgCMdYkvrIktFvUpy-2GPbFgfSVKZICawB2hiJzus,4999
 langchain_google_genai/chat_models.py,sha256=NNAtdHOKCv4E1RnQMwc-gI9GYJQfp-Ccaz6oPtmkd5g,47149
 langchain_google_genai/embeddings.py,sha256=FWouioVYXV4Hpikl7C-EkzL9DwUzqa_bQkfJxRY7JZg,10089
 langchain_google_genai/genai_aqa.py,sha256=zcC5cdFYtqLK7DGPhYGvWNeHHeU-CQKA9KhewmsA5lw,4303
 langchain_google_genai/google_vector_store.py,sha256=PPIk-4FmD5UUdmYA2u7VcEhGsiztvRVN59QoGLXdfoA,16139
-langchain_google_genai/llms.py,sha256=W8tvdKVSfVTz6Vr3RSj_LJ9j-GGE3zdF3fLLrL7xxmc,13694
+langchain_google_genai/llms.py,sha256=FzlpRrCebuTR4ye0nnKuNy0MX_2lZXItL_oqNMQemaA,14150
 langchain_google_genai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langchain_google_genai-1.0.9.dist-info/LICENSE,sha256=DppmdYJVSc1jd0aio6ptnMUn5tIHrdAhQ12SclEBfBg,1072
-langchain_google_genai-1.0.9.dist-info/METADATA,sha256=92xOfix0vXV6_43UPM0RE1FVcz-mujJXv25dwXCkwcA,3818
-langchain_google_genai-1.0.9.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-langchain_google_genai-1.0.9.dist-info/RECORD,,
+langchain_google_genai-1.0.10.dist-info/LICENSE,sha256=DppmdYJVSc1jd0aio6ptnMUn5tIHrdAhQ12SclEBfBg,1072
+langchain_google_genai-1.0.10.dist-info/METADATA,sha256=WfG7dVpeATITKrwtj-VAzVKnQacbqrxxltzqTxNH3Ck,3819
+langchain_google_genai-1.0.10.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+langchain_google_genai-1.0.10.dist-info/RECORD,,