langchain-google-genai 3.0.0rc1__tar.gz → 3.0.1__tar.gz

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.1
 Name: langchain-google-genai
-Version: 3.0.0rc1
+Version: 3.0.1
 Summary: An integration package connecting Google's genai package and LangChain
 License: MIT
 Project-URL: Source Code, https://github.com/langchain-ai/langchain-google/tree/main/libs/genai
 Project-URL: Release Notes, https://github.com/langchain-ai/langchain-google/releases
 Project-URL: repository, https://github.com/langchain-ai/langchain-google
 Requires-Python: <4.0.0,>=3.10.0
-Requires-Dist: langchain-core<2.0.0,>=1.0.0rc2
+Requires-Dist: langchain-core<2.0.0,>=1.0.0
 Requires-Dist: google-ai-generativelanguage<1.0.0,>=0.7.0
 Requires-Dist: pydantic<3.0.0,>=2.0.0
 Requires-Dist: filetype<2.0.0,>=1.2.0
@@ -131,7 +131,7 @@ Some Gemini models supports both text and inline image outputs.
 ```python
 from langchain_google_genai import ChatGoogleGenerativeAI
 
-llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash-image-preview")
+llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash-image")
 
 response = llm.invoke(
     "Generate an image of a cat and say meow",
@@ -179,7 +179,7 @@ from langchain_core.prompts import ChatPromptTemplate
 from langchain_google_genai import ChatGoogleGenerativeAI, Modality
 
 llm = ChatGoogleGenerativeAI(
-    model="gemini-2.5-flash-image-preview",
+    model="gemini-2.5-flash-image",
     response_modalities=[Modality.TEXT, Modality.IMAGE],
 )
 

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/README.md

@@ -116,7 +116,7 @@ Some Gemini models supports both text and inline image outputs.
 ```python
 from langchain_google_genai import ChatGoogleGenerativeAI
 
-llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash-image-preview")
+llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash-image")
 
 response = llm.invoke(
     "Generate an image of a cat and say meow",
@@ -164,7 +164,7 @@ from langchain_core.prompts import ChatPromptTemplate
 from langchain_google_genai import ChatGoogleGenerativeAI, Modality
 
 llm = ChatGoogleGenerativeAI(
-    model="gemini-2.5-flash-image-preview",
+    model="gemini-2.5-flash-image",
     response_modalities=[Modality.TEXT, Modality.IMAGE],
 )
 

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/langchain_google_genai/__init__.py

@@ -57,7 +57,12 @@ The package also supports creating embeddings with Google's models, useful for t
 
 """  # noqa: E501
 
-from langchain_google_genai._enums import HarmBlockThreshold, HarmCategory, Modality
+from langchain_google_genai._enums import (
+    HarmBlockThreshold,
+    HarmCategory,
+    MediaResolution,
+    Modality,
+)
 from langchain_google_genai.chat_models import ChatGoogleGenerativeAI
 from langchain_google_genai.embeddings import GoogleGenerativeAIEmbeddings
 from langchain_google_genai.genai_aqa import (
@@ -84,4 +89,5 @@ __all__ = [
     "HarmBlockThreshold",
     "HarmCategory",
     "Modality",
+    "MediaResolution",
 ]

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/langchain_google_genai/_common.py

@@ -6,7 +6,12 @@ from google.api_core.gapic_v1.client_info import ClientInfo
 from langchain_core.utils import secret_from_env
 from pydantic import BaseModel, Field, SecretStr
 
-from langchain_google_genai._enums import HarmBlockThreshold, HarmCategory, Modality
+from langchain_google_genai._enums import (
+    HarmBlockThreshold,
+    HarmCategory,
+    MediaResolution,
+    Modality,
+)
 
 _TELEMETRY_TAG = "remote_reasoning_engine"
 _TELEMETRY_ENV_VARIABLE_NAME = "GOOGLE_CLOUD_AGENT_ENGINE_ID"
@@ -19,83 +24,112 @@ class GoogleGenerativeAIError(Exception):
 class _BaseGoogleGenerativeAI(BaseModel):
     """Base class for Google Generative AI LLMs."""
 
-    model: str = Field(
-        ...,
-        description="""The name of the model to use.
-Examples:
-    - gemini-2.5-flash
-    - models/text-bison-001""",
-    )
+    model: str = Field(...)
     """Model name to use."""
+
     google_api_key: Optional[SecretStr] = Field(
         alias="api_key", default_factory=secret_from_env("GOOGLE_API_KEY", default=None)
     )
     """Google AI API key.
-    If not specified will be read from env var ``GOOGLE_API_KEY``."""
+    
+    If not specified will be read from env var å`GOOGLE_API_KEY`.
+    """
 
     credentials: Any = None
-    "The default custom credentials (google.auth.credentials.Credentials) to use "
-    "when making API calls. If not provided, credentials will be ascertained from "
-    "the GOOGLE_API_KEY envvar"
+    """The default custom credentials (`google.auth.credentials.Credentials`) to use 
+    when making API calls.
+    
+    If not provided, credentials will be ascertained from the `GOOGLE_API_KEY` env var.
+    """
 
     temperature: float = 0.7
-    """Run inference with this temperature. Must be within ``[0.0, 2.0]``. If unset,
-    will default to ``0.7``."""
+    """Run inference with this temperature. Must be within `[0.0, 2.0]`.
+    
+    If unset, will default to `0.7`.
+    """
 
     top_p: Optional[float] = None
     """Decode using nucleus sampling: consider the smallest set of tokens whose
-    probability sum is at least ``top_p``. Must be within ``[0.0, 1.0]``."""
+    probability sum is at least `top_p`.
+    
+    Must be within `[0.0, 1.0]`.
+    """
 
     top_k: Optional[int] = None
-    """Decode using top-k sampling: consider the set of ``top_k`` most probable tokens.
-    Must be positive."""
+    """Decode using top-k sampling: consider the set of `top_k` most probable tokens.
+    
+    Must be positive.
+    """
 
     max_output_tokens: Optional[int] = Field(default=None, alias="max_tokens")
     """Maximum number of tokens to include in a candidate. Must be greater than zero.
+    
     If unset, will use the model's default value, which varies by model.
-    See https://ai.google.dev/gemini-api/docs/models for model-specific limits."""
+    
+    See https://ai.google.dev/gemini-api/docs/models for model-specific limits.
+    """
 
     n: int = 1
-    """Number of chat completions to generate for each prompt. Note that the API may
-    not return the full ``n`` completions if duplicates are generated."""
+    """Number of chat completions to generate for each prompt.
+    
+    Note that the API may not return the full `n` completions if duplicates are
+    generated.
+    """
 
     max_retries: int = Field(default=6, alias="retries")
-    """The maximum number of retries to make when generating. If unset, will default
-    to ``6``."""
+    """The maximum number of retries to make when generating.
+    
+    If unset, will default to `6`.
+    """
 
     timeout: Optional[float] = Field(default=None, alias="request_timeout")
     """The maximum number of seconds to wait for a response."""
 
     client_options: Optional[Dict] = Field(
         default=None,
-        description=(
-            "A dictionary of client options to pass to the Google API client, "
-            "such as `api_endpoint`."
-        ),
     )
+    """"A dictionary of client options to pass to the Google API client, such as
+    `api_endpoint`.
+    """
+
+    base_url: Optional[str] = Field(
+        default=None,
+    )
+    """The base URL to use for the API client.
+    
+    Alias of `client_options['api_endpoint']`.
+    """
+
     transport: Optional[str] = Field(
         default=None,
-        description="A string, one of: [`rest`, `grpc`, `grpc_asyncio`].",
         alias="api_transport",
     )
+    """A string, one of: `[rest, grpc, grpc_asyncio]`."""
+
     additional_headers: Optional[Dict[str, str]] = Field(
         default=None,
-        description=(
-            "A key-value dictionary representing additional headers for the model call"
-        ),
     )
+    """"Key-value dictionary representing additional headers for the model call"""
+
     response_modalities: Optional[List[Modality]] = Field(
-        default=None, description=("A list of modalities of the response")
+        default=None,
     )
+    """A list of modalities of the response"""
 
     thinking_budget: Optional[int] = Field(
-        default=None, description="Indicates the thinking budget in tokens."
+        default=None,
+    )
+    """Indicates the thinking budget in tokens."""
+
+    media_resolution: Optional[MediaResolution] = Field(
+        default=None,
     )
+    """Media resolution for the input media."""
 
     include_thoughts: Optional[bool] = Field(
         default=None,
-        description="Indicates whether to include thoughts in the response.",
     )
+    """Indicates whether to include thoughts in the response."""
 
     safety_settings: Optional[Dict[HarmCategory, HarmBlockThreshold]] = None
     """The default safety settings to use for all generations.

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/langchain_google_genai/_compat.py

@@ -143,6 +143,10 @@ def _convert_from_v1_to_generativelanguage_v1beta(
         # TextContentBlock
         if block_dict["type"] == "text":
             new_block = {"text": block_dict.get("text", "")}
+            if (
+                thought_signature := (block_dict.get("extras") or {}).get("signature")  # type: ignore[attr-defined]
+            ) and model_provider == "google_genai":
+                new_block["thought_signature"] = thought_signature
             new_content.append(new_block)
             # Citations are only handled on output. Can't pass them back :/
 

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/langchain_google_genai/_enums.py

@@ -3,5 +3,6 @@ import google.ai.generativelanguage_v1beta as genai
 HarmBlockThreshold = genai.SafetySetting.HarmBlockThreshold
 HarmCategory = genai.HarmCategory
 Modality = genai.GenerationConfig.Modality
+MediaResolution = genai.GenerationConfig.MediaResolution
 
-__all__ = ["HarmBlockThreshold", "HarmCategory", "Modality"]
+__all__ = ["HarmBlockThreshold", "HarmCategory", "Modality", "MediaResolution"]

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/langchain_google_genai/chat_models.py

@@ -199,7 +199,7 @@ def _chat_with_retry(generation_method: Callable, **kwargs: Any) -> Any:
 
     Args:
         generation_method (Callable): The chat generation method to be executed.
-        **kwargs (Any): Additional keyword arguments to pass to the generation method.
+        **kwargs: Additional keyword arguments to pass to the generation method.
 
     Returns:
         Any: The result from the chat generation method.
@@ -256,7 +256,7 @@ async def _achat_with_retry(generation_method: Callable, **kwargs: Any) -> Any:
 
     Args:
         generation_method (Callable): The chat generation method to be executed.
-        **kwargs (Any): Additional keyword arguments to pass to the generation method.
+        **kwargs: Additional keyword arguments to pass to the generation method.
 
     Returns:
         Any: The result from the chat generation method.
@@ -360,7 +360,7 @@ def _convert_to_parts(
                     if "mime_type" in part:
                         inline_data["mime_type"] = part["mime_type"]
                     else:
-                        # Guess mime type based on data field if not provided
+                        # Guess MIME type based on data field if not provided
                         source = cast(
                             "str",
                             part.get("url") or part.get("base64") or part.get("data"),
@@ -1912,11 +1912,17 @@ class ChatGoogleGenerativeAI(_BaseGoogleGenerativeAI, BaseChatModel):
             else:
                 google_api_key = self.google_api_key
         transport: Optional[str] = self.transport
+
+        # Merge base_url into client_options if provided
+        client_options = self.client_options or {}
+        if self.base_url and "api_endpoint" not in client_options:
+            client_options = {**client_options, "api_endpoint": self.base_url}
+
         self.client = genaix.build_generative_service(
             credentials=self.credentials,
             api_key=google_api_key,
             client_info=client_info,
-            client_options=self.client_options,
+            client_options=client_options,
             transport=transport,
         )
         self.async_client_running = None
@@ -1941,11 +1947,17 @@ class ChatGoogleGenerativeAI(_BaseGoogleGenerativeAI, BaseChatModel):
             transport = self.transport
             if transport == "rest":
                 transport = "grpc_asyncio"
+
+            # Merge base_url into client_options if provided
+            client_options = self.client_options or {}
+            if self.base_url and "api_endpoint" not in client_options:
+                client_options = {**client_options, "api_endpoint": self.base_url}
+
             self.async_client_running = genaix.build_generative_async_service(
                 credentials=self.credentials,
                 api_key=google_api_key,
                 client_info=get_client_info(f"ChatGoogleGenerativeAI:{self.model}"),
-                client_options=self.client_options,
+                client_options=client_options,
                 transport=transport,
             )
         return self.async_client_running
@@ -1960,6 +1972,7 @@ class ChatGoogleGenerativeAI(_BaseGoogleGenerativeAI, BaseChatModel):
             "n": self.n,
             "safety_settings": self.safety_settings,
             "response_modalities": self.response_modalities,
+            "media_resolution": self.media_resolution,
             "thinking_budget": self.thinking_budget,
             "include_thoughts": self.include_thoughts,
         }
@@ -2077,6 +2090,11 @@ class ChatGoogleGenerativeAI(_BaseGoogleGenerativeAI, BaseChatModel):
             gapic_response_schema = _dict_to_gapic_schema(response_schema)
             if gapic_response_schema is not None:
                 gen_config["response_schema"] = gapic_response_schema
+
+        media_resolution = kwargs.get("media_resolution", self.media_resolution)
+        if media_resolution is not None:
+            gen_config["media_resolution"] = media_resolution
+
         return GenerationConfig(**gen_config)
 
     def _generate(

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/langchain_google_genai/embeddings.py

@@ -87,6 +87,14 @@ class GoogleGenerativeAIEmbeddings(BaseModel, Embeddings):
             "such as `api_endpoint`."
         ),
     )
+    base_url: Optional[str] = Field(
+        default=None,
+    )
+    """The base URL to use for the API client.
+    
+    Alias of `client_options['api_endpoint']`.
+    """
+
     transport: Optional[str] = Field(
         default=None,
         description="A string, one of: [``'rest'``, ``'grpc'``, ``'grpc_asyncio'``].",
@@ -109,11 +117,16 @@ class GoogleGenerativeAIEmbeddings(BaseModel, Embeddings):
         if not any(self.model.startswith(prefix) for prefix in ("models/",)):
             self.model = f"models/{self.model}"
 
+        # Merge base_url into client_options if provided
+        client_options = self.client_options or {}
+        if self.base_url and "api_endpoint" not in client_options:
+            client_options = {**client_options, "api_endpoint": self.base_url}
+
         self.client = build_generative_service(
             credentials=self.credentials,
             api_key=google_api_key,
             client_info=client_info,
-            client_options=self.client_options,
+            client_options=client_options,
             transport=self.transport,
         )
         # Always defer async client initialization to first async call.
@@ -137,11 +150,16 @@ class GoogleGenerativeAIEmbeddings(BaseModel, Embeddings):
             if transport == "rest":
                 transport = "grpc_asyncio"
 
+            # Merge base_url into client_options if provided
+            client_options = self.client_options or {}
+            if self.base_url and "api_endpoint" not in client_options:
+                client_options = {**client_options, "api_endpoint": self.base_url}
+
             self.async_client = build_generative_async_service(
                 credentials=self.credentials,
                 api_key=google_api_key,
                 client_info=client_info,
-                client_options=self.client_options,
+                client_options=client_options,
                 transport=transport,
             )
         return self.async_client
@@ -242,7 +260,7 @@ class GoogleGenerativeAIEmbeddings(BaseModel, Embeddings):
             batch_size: [int] The batch size of embeddings to send to the model
             task_type: `task_type <https://ai.google.dev/api/embeddings#tasktype>`__
             titles: An optional list of titles for texts provided.
-              Only applicable when TaskType is ``'RETRIEVAL_DOCUMENT'``.
+                Only applicable when TaskType is ``'RETRIEVAL_DOCUMENT'``.
             output_dimensionality: Optional `reduced dimension for the output embedding <https://ai.google.dev/api/embeddings#EmbedContentRequest>`__.
 
         Returns:
@@ -293,7 +311,7 @@ class GoogleGenerativeAIEmbeddings(BaseModel, Embeddings):
             text: The text to embed.
             task_type: `task_type <https://ai.google.dev/api/embeddings#tasktype>`__
             title: An optional title for the text.
-              Only applicable when TaskType is ``'RETRIEVAL_DOCUMENT'``.
+                Only applicable when TaskType is ``'RETRIEVAL_DOCUMENT'``.
             output_dimensionality: Optional `reduced dimension for the output embedding <https://ai.google.dev/api/embeddings#EmbedContentRequest>`__.
 
         Returns:

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/langchain_google_genai/genai_aqa.py

@@ -103,19 +103,30 @@ class GenAIAqa(RunnableSerializable[AqaInput, AqaOutput]):
     # google.generativeai installed.
     answer_style: int = 1
 
-    def __init__(self, **kwargs: Any) -> None:
+    def __init__(
+        self,
+        *,
+        answer_style: int = genai.GenerateAnswerRequest.AnswerStyle.ABSTRACTIVE,
+        safety_settings: Optional[List[genai.SafetySetting]] = None,
+        temperature: Optional[float] = None,
+        **kwargs: Any,
+    ) -> None:
         """Construct a Google Generative AI AQA model.
 
         All arguments are optional.
 
         Args:
             answer_style: See
-              `google.ai.generativelanguage.GenerateAnswerRequest.AnswerStyle`.
+                `google.ai.generativelanguage.GenerateAnswerRequest.AnswerStyle`.
             safety_settings: See `google.ai.generativelanguage.SafetySetting`.
             temperature: 0.0 to 1.0.
         """
         super().__init__(**kwargs)
-        self._client = _AqaModel(**kwargs)
+        self._client = _AqaModel(
+            answer_style=answer_style,
+            safety_settings=safety_settings,
+            temperature=temperature,
+        )
 
     def invoke(
         self, input: AqaInput, config: Optional[RunnableConfig] = None, **kwargs: Any

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/langchain_google_genai/google_vector_store.py

@@ -262,9 +262,9 @@ class GoogleVectorStore(VectorStore):
         document.
 
         Raises:
-            DoesNotExistsException if the IDs do not match to anything on Google
-                server. In this case, consider using `create_corpus` or
-                `create_document` to create one.
+            DoesNotExistsException: If the IDs do not match to anything on Google
+                server. In this case, consider using ``create_corpus`` or
+                ``create_document`` to create one.
         """
         super().__init__(**kwargs)
         self._retriever = _SemanticRetriever.from_ids(corpus_id, document_id)
@@ -352,7 +352,7 @@ class GoogleVectorStore(VectorStore):
             Document.
 
         Raises:
-            DoesNotExistsException if the IDs do not match to anything at
+            DoesNotExistsException: If the IDs do not match to anything at
                 Google server.
         """
         if corpus_id is None or document_id is None:
@@ -463,16 +463,22 @@ class GoogleVectorStore(VectorStore):
         """
         return lambda score: score
 
-    def as_aqa(self, **kwargs: Any) -> Runnable[str, AqaOutput]:
+    def as_aqa(
+        self,
+        *,
+        answer_style: int = 1,
+        safety_settings: Optional[List[Any]] = None,
+        temperature: Optional[float] = None,
+    ) -> Runnable[str, AqaOutput]:
         """Construct a Google Generative AI AQA engine.
 
         All arguments are optional.
 
         Args:
             answer_style: See
-              `google.ai.generativelanguage.GenerateAnswerRequest.AnswerStyle`.
-            safety_settings: See `google.ai.generativelanguage.SafetySetting`.
-            temperature: 0.0 to 1.0.
+                ``google.ai.generativelanguage.GenerateAnswerRequest.AnswerStyle``.
+            safety_settings: See ``google.ai.generativelanguage.SafetySetting``.
+            temperature: Value between 0.0 and 1.0 controlling randomness.
         """
         return (
             RunnablePassthrough[str]()
@@ -481,7 +487,11 @@ class GoogleVectorStore(VectorStore):
                 "passages": self.as_retriever(),
             }
             | RunnableLambda(_toAqaInput)
-            | GenAIAqa(**kwargs)
+            | GenAIAqa(
+                answer_style=answer_style,
+                safety_settings=safety_settings,
+                temperature=temperature,
+            )
         )
 
 

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/langchain_google_genai/llms.py

@@ -77,6 +77,7 @@ class GoogleGenerativeAI(_BaseGoogleGenerativeAI, BaseLLM):
             timeout=self.timeout,
             model=self.model,
             client_options=self.client_options,
+            base_url=self.base_url,
             transport=self.transport,
             additional_headers=self.additional_headers,
             safety_settings=self.safety_settings,

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/pyproject.toml

@@ -6,14 +6,14 @@ build-backend = "pdm.backend"
 
 [project]
 name = "langchain-google-genai"
-version = "3.0.0rc1"
+version = "3.0.1"
 description = "An integration package connecting Google's genai package and LangChain"
 authors = []
 requires-python = ">=3.10.0,<4.0.0"
 readme = "README.md"
 repository = "https://github.com/langchain-ai/langchain-google"
 dependencies = [
-    "langchain-core>=1.0.0rc2,<2.0.0",
+    "langchain-core>=1.0.0,<2.0.0",
     "google-ai-generativelanguage>=0.7.0,<1.0.0",
     "pydantic>=2.0.0,<3.0.0",
     "filetype>=1.2.0,<2.0.0",
@@ -49,7 +49,7 @@ test = [
     "pytest-socket>=0.7.0,<1.0.0",
     "numpy>=1.26.4; python_version<'3.13'",
     "numpy>=2.1.0; python_version>='3.13'",
-    "langchain-tests>=1.0.0rc1,<2.0.0",
+    "langchain-tests>=1.0.0,<2.0.0",
 ]
 test_integration = [
     "pytest>=8.4.0,<9.0.0",

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/tests/integration_tests/test_chat_models.py

@@ -23,6 +23,7 @@ from langchain_google_genai import (
     ChatGoogleGenerativeAI,
     HarmBlockThreshold,
     HarmCategory,
+    MediaResolution,
     Modality,
 )
 
@@ -286,6 +287,7 @@ def _check_thinking_output(content: list, output_version: str) -> None:
         assert isinstance(block[thinking_key], str)
 
 
+@pytest.mark.flaky(retries=3, delay=1)
 @pytest.mark.parametrize("output_version", ["v0", "v1"])
 def test_chat_google_genai_invoke_thinking_include_thoughts(
     output_version: str,
@@ -557,6 +559,42 @@ def test_chat_google_genai_multimodal(
         assert len(response.content.strip()) > 0
 
 
+@pytest.mark.parametrize(
+    "message",
+    [
+        HumanMessage(
+            content=[
+                {
+                    "type": "text",
+                    "text": "Guess what's in this picture! You have 3 guesses.",
+                },
+                {
+                    "type": "image_url",
+                    "image_url": "https://picsum.photos/seed/picsum/200/300",
+                },
+            ]
+        ),
+    ],
+)
+def test_chat_google_genai_invoke_media_resolution(message: BaseMessage) -> None:
+    """Test invoke vision model with `media_resolution` set to low and without."""
+    llm = ChatGoogleGenerativeAI(model=_VISION_MODEL)
+    result = llm.invoke([message])
+    result_low_res = llm.invoke(
+        [message], media_resolution=MediaResolution.MEDIA_RESOLUTION_LOW
+    )
+
+    assert isinstance(result_low_res, AIMessage)
+    _check_usage_metadata(result_low_res)
+
+    assert result.usage_metadata is not None
+    assert result_low_res.usage_metadata is not None
+    assert (
+        result_low_res.usage_metadata["input_tokens"]
+        < result.usage_metadata["input_tokens"] / 3
+    )
+
+
 def test_chat_google_genai_single_call_with_history() -> None:
     model = ChatGoogleGenerativeAI(model=_MODEL)
     text_question1, text_answer1 = "How much is 2+2?", "4"
@@ -570,21 +608,19 @@ def test_chat_google_genai_single_call_with_history() -> None:
 
 
 @pytest.mark.parametrize(
-    ("model_name", "convert_system_message_to_human"),
-    [(_MODEL, True), ("models/gemini-2.5-pro", False)],
+    "model_name",
+    [_MODEL, "models/gemini-2.5-pro"],
 )
 def test_chat_google_genai_system_message(
-    model_name: str, convert_system_message_to_human: bool
+    model_name: str,
 ) -> None:
     """Test system message handling in ChatGoogleGenerativeAI.
 
-    Parameterized to test different models and system message conversion settings.
-
-    Useful since I think some models (e.g. Gemini Pro) do not like system messages?
+    Tests that system messages are properly converted to system instructions
+    for different models.
     """
     model = ChatGoogleGenerativeAI(
         model=model_name,
-        convert_system_message_to_human=convert_system_message_to_human,
     )
     text_question1, text_answer1 = "How much is 2+2?", "4"
     text_question2 = "How much is 3+3?"

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/tests/integration_tests/test_standard.py

@@ -1,12 +1,10 @@
 """Standard LangChain interface tests."""
 
-import base64
+import os
 from typing import Literal
 
-import httpx
 import pytest
 from langchain_core.language_models import BaseChatModel
-from langchain_core.messages import HumanMessage
 from langchain_core.rate_limiters import InMemoryRateLimiter
 from langchain_core.tools import BaseTool
 from langchain_tests.integration_tests import ChatModelIntegrationTests
@@ -16,6 +14,14 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 rate_limiter = InMemoryRateLimiter(requests_per_second=0.25)
 
 
+def _has_multimodal_secrets() -> bool:
+    """Check if integration test secrets are available.
+
+    Returns `True` if running in an environment with access to secrets.
+    """
+    return bool(os.environ.get("LANGCHAIN_TESTS_USER_AGENT"))
+
+
 class TestGeminiFlashStandard(ChatModelIntegrationTests):
     @property
     def chat_model_class(self) -> type[BaseChatModel]:
@@ -48,55 +54,29 @@ class TestGeminiFlashStandard(ChatModelIntegrationTests):
     def supports_audio_inputs(self) -> bool:
         return True
 
-    @pytest.mark.xfail(reason="Override parent method to use a reliable image URL")
-    def test_image_inputs(self, model: BaseChatModel) -> None:
-        """Override parent method to use a reliable image URL."""
-        if not self.supports_image_inputs:
-            pytest.skip("Model does not support image message.")
-
-        # Use a reliable image URL that works with requests
-        image_url = "https://picsum.photos/seed/picsum/200/300"
-        image_data = base64.b64encode(httpx.get(image_url).content).decode("utf-8")
-
-        # OpenAI format, base64 data
-        message = HumanMessage(
-            content=[
-                {"type": "text", "text": "describe the weather in this image"},
-                {
-                    "type": "image_url",
-                    "image_url": {"url": f"data:image/jpeg;base64,{image_data}"},
-                },
-            ],
-        )
-        _ = model.invoke([message])
-
-        # Standard format, base64 data
-        message = HumanMessage(
-            content=[
-                {"type": "text", "text": "describe the weather in this image"},
-                {
-                    "type": "image",
-                    "source_type": "base64",
-                    "mime_type": "image/jpeg",
-                    "data": image_data,
-                },
-            ],
-        )
-        _ = model.invoke([message])
-
-        # Standard format, URL
-        if self.supports_image_urls:
-            message = HumanMessage(
-                content=[
-                    {"type": "text", "text": "describe the weather in this image"},
-                    {
-                        "type": "image",
-                        "source_type": "url",
-                        "url": image_url,
-                    },
-                ],
-            )
-            _ = model.invoke([message])
+    @pytest.mark.xfail(
+        not _has_multimodal_secrets(),
+        reason=(
+            "Multimodal tests require integration secrets (user agent to fetch "
+            "external resources)"
+        ),
+        run=False,
+    )
+    def test_audio_inputs(self, model: BaseChatModel) -> None:
+        """Skip audio tests in PR context - requires external resource fetching."""
+        super().test_audio_inputs(model)
+
+    @pytest.mark.xfail(
+        not _has_multimodal_secrets(),
+        reason=(
+            "Multimodal tests require integration secrets (user agent to fetch "
+            "external resources)"
+        ),
+        run=False,
+    )
+    def test_pdf_inputs(self, model: BaseChatModel) -> None:
+        """Skip PDF tests in PR context - requires external resource fetching."""
+        super().test_pdf_inputs(model)
 
 
 class TestGeminiProStandard(ChatModelIntegrationTests):

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/tests/unit_tests/test_chat_models.py

@@ -27,6 +27,7 @@ from langchain_core.messages import (
     SystemMessage,
     ToolMessage,
 )
+from langchain_core.messages import content as types
 from langchain_core.messages.block_translators.google_genai import (
     _convert_to_v1_from_genai,
 )
@@ -36,6 +37,9 @@ from pydantic import SecretStr
 from pydantic_core._pydantic_core import ValidationError
 
 from langchain_google_genai import HarmBlockThreshold, HarmCategory, Modality
+from langchain_google_genai._compat import (
+    _convert_from_v1_to_generativelanguage_v1beta,
+)
 from langchain_google_genai.chat_models import (
     ChatGoogleGenerativeAI,
     _chat_with_retry,
@@ -431,6 +435,46 @@ def test_additional_headers_support(headers: Optional[dict[str, str]]) -> None:
     assert "ChatGoogleGenerativeAI" in call_client_info.user_agent
 
 
+def test_base_url_support() -> None:
+    """Test that `base_url` is properly merged into `client_options`."""
+    mock_client = Mock()
+    mock_generate_content = Mock()
+    mock_generate_content.return_value = GenerateContentResponse(
+        candidates=[Candidate(content=Content(parts=[Part(text="test response")]))]
+    )
+    mock_client.return_value.generate_content = mock_generate_content
+    base_url = "https://example.com"
+    param_api_key = "[secret]"
+    param_secret_api_key = SecretStr(param_api_key)
+    param_transport = "rest"
+
+    with patch(
+        "langchain_google_genai._genai_extension.v1betaGenerativeServiceClient",
+        mock_client,
+    ):
+        chat = ChatGoogleGenerativeAI(
+            model=MODEL_NAME,
+            google_api_key=param_secret_api_key,
+            base_url=base_url,
+            transport=param_transport,
+        )
+
+    response = chat.invoke("test")
+    assert response.content == "test response"
+
+    mock_client.assert_called_once_with(
+        transport=param_transport,
+        client_options=ANY,
+        client_info=ANY,
+    )
+    call_client_options = mock_client.call_args_list[0].kwargs["client_options"]
+    assert call_client_options.api_key == param_api_key
+    assert call_client_options.api_endpoint == base_url
+    call_client_info = mock_client.call_args_list[0].kwargs["client_info"]
+    assert "langchain-google-genai" in call_client_info.user_agent
+    assert "ChatGoogleGenerativeAI" in call_client_info.user_agent
+
+
 def test_default_metadata_field_alias() -> None:
     """Test 'default_metadata' and 'default_metadata_input' fields work correctly."""
     # Test with default_metadata_input field name (alias) - should accept None without
@@ -1859,3 +1903,15 @@ def test_chat_google_genai_invoke_with_audio_mocked() -> None:
     assert audio_block["type"] == "audio"
     assert "base64" in audio_block
     assert audio_block["base64"] == base64.b64encode(wav_bytes).decode()
+
+
+def test_compat() -> None:
+    block: types.TextContentBlock = {"type": "text", "text": "foo"}
+    result = _convert_from_v1_to_generativelanguage_v1beta([block], "google_genai")
+    expected = [{"text": "foo"}]
+    assert result == expected
+
+    block = {"type": "text", "text": "foo", "extras": {"signature": "bar"}}
+    result = _convert_from_v1_to_generativelanguage_v1beta([block], "google_genai")
+    expected = [{"text": "foo", "thought_signature": "bar"}]
+    assert result == expected

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/tests/unit_tests/test_embeddings.py

@@ -1,6 +1,6 @@
 """Test embeddings model integration."""
 
-from unittest.mock import MagicMock, patch
+from unittest.mock import ANY, MagicMock, patch
 
 import pytest
 from google.ai.generativelanguage_v1beta.types import (
@@ -160,3 +160,33 @@ def test_embed_documents_with_numerous_texts() -> None:
         )
         mock_embed.assert_called_with(request)
         assert mock_embed.call_count == test_corpus_size / test_batch_size
+
+
+def test_base_url_support() -> None:
+    """Test that base_url is properly merged into client_options."""
+    base_url = "https://example.com"
+    param_api_key = "[secret]"
+    param_secret_api_key = SecretStr(param_api_key)
+    param_transport = "rest"
+
+    with patch(
+        "langchain_google_genai._genai_extension.v1betaGenerativeServiceClient"
+    ) as mock_client:
+        _ = GoogleGenerativeAIEmbeddings(
+            model=f"models/{MODEL_NAME}",
+            google_api_key=param_secret_api_key,
+            base_url=base_url,
+            transport=param_transport,
+        )
+
+    mock_client.assert_called_once_with(
+        transport=param_transport,
+        client_options=ANY,
+        client_info=ANY,
+    )
+    call_client_options = mock_client.call_args_list[0].kwargs["client_options"]
+    assert call_client_options.api_key == param_api_key
+    assert call_client_options.api_endpoint == base_url
+    call_client_info = mock_client.call_args_list[0].kwargs["client_info"]
+    assert "langchain-google-genai" in call_client_info.user_agent
+    assert "GoogleGenerativeAIEmbeddings" in call_client_info.user_agent

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/tests/unit_tests/test_imports.py

@@ -12,6 +12,7 @@ EXPECTED_ALL = [
     "HarmBlockThreshold",
     "HarmCategory",
     "Modality",
+    "MediaResolution",
     "DoesNotExistsException",
 ]
 

{langchain_google_genai-3.0.0rc1 → langchain_google_genai-3.0.1}/tests/unit_tests/test_llms.py

@@ -1,4 +1,12 @@
-from unittest.mock import patch
+from unittest.mock import ANY, Mock, patch
+
+from google.ai.generativelanguage_v1beta.types import (
+    Candidate,
+    Content,
+    GenerateContentResponse,
+    Part,
+)
+from pydantic import SecretStr
 
 from langchain_google_genai.llms import GoogleGenerativeAI
 
@@ -47,3 +55,40 @@ def test_tracing_params() -> None:
         call_args = mock_warning.call_args[0][0]
         assert "Unexpected argument 'safety_setting'" in call_args
         assert "Did you mean: 'safety_settings'?" in call_args
+
+
+def test_base_url_support() -> None:
+    """Test that base_url is properly passed through to ChatGoogleGenerativeAI."""
+    mock_client = Mock()
+    mock_generate_content = Mock()
+    mock_generate_content.return_value = GenerateContentResponse(
+        candidates=[Candidate(content=Content(parts=[Part(text="test response")]))]
+    )
+    mock_client.return_value.generate_content = mock_generate_content
+    base_url = "https://example.com"
+    param_api_key = "[secret]"
+    param_secret_api_key = SecretStr(param_api_key)
+    param_transport = "rest"
+
+    with patch(
+        "langchain_google_genai._genai_extension.v1betaGenerativeServiceClient",
+        mock_client,
+    ):
+        llm = GoogleGenerativeAI(
+            model=MODEL_NAME,
+            google_api_key=param_secret_api_key,
+            base_url=base_url,
+            transport=param_transport,
+        )
+
+    response = llm.invoke("test")
+    assert response == "test response"
+
+    mock_client.assert_called_once_with(
+        transport=param_transport,
+        client_options=ANY,
+        client_info=ANY,
+    )
+    call_client_options = mock_client.call_args_list[0].kwargs["client_options"]
+    assert call_client_options.api_key == param_api_key
+    assert call_client_options.api_endpoint == base_url