Merge branch 'main' into shuowei-gbq-delegation-telemetry-mvp

Author: shuoweil

.github/workflows/import-profiler.yml

@@ -32,5 +32,7 @@ jobs:
         TARGET_BRANCH: ${{ github.base_ref || github.event.merge_group.base_ref }}
         TEST_TYPE: import_profile
         PY_VERSION: "3.15"
+        # Workaround: Allows libcst to compile on Python 3.15+ while PyO3 catches up
+        PYO3_USE_ABI3_FORWARD_COMPATIBILITY: "1"
       run: |
         ci/run_conditional_tests.sh

librarian.yaml

@@ -16,8 +16,8 @@ version: v0.26.0
 repo: googleapis/google-cloud-python
 sources:
   googleapis:
-    commit: 99f54e6513f09d8df1707a6c553b0a3c6ef9b5fb
-    sha256: 7cee15fe1ae76a16aaf70425c4380a3a0fcbfeb4ea8e30c934c43f5cd54b9657
+    commit: fe04226e3d76435acbfd3ffb51f6771740e1fb5b
+    sha256: e99c641a9c5525394de43c762b9db50c3ada1f219d6dbdba594ae992fbe1d838
 default:
   output: packages
   tag_format: '{name}-v{version}'

packages/bigframes/tests/unit/display/test_anywidget.py

@@ -45,11 +45,13 @@ def test_navigation_to_invalid_page_resets_to_valid_page_without_deadlock():
     mock_df._block = mock_block
 
     # We mock _initial_load to avoid complex setup
-    with mock.patch.object(TableWidget, "_initial_load"):
-        with bigframes.option_context(
+    with (
+        mock.patch.object(TableWidget, "_initial_load"),
+        bigframes.option_context(
             "display.render_mode", "anywidget", "display.max_rows", 10
-        ):
-            widget = TableWidget(mock_df)
+        ),
+    ):
+        widget = TableWidget(mock_df)
 
     # Simulate "loaded data but unknown total rows" state
     widget.page_size = 10
@@ -179,11 +181,11 @@ def test_page_size_change_resets_sort(mock_df):
 
 def test_cell_execution_count_propagation(mock_df):
     """Test that the captured cell_execution_count is propagated to to_pandas_batches."""
-    with mock.patch(
-        "bigframes.core.utils.get_ipython_execution_count", return_value=42
+    with (
+        mock.patch("bigframes.core.utils.get_ipython_execution_count", return_value=42),
+        bigframes.option_context("display.render_mode", "anywidget"),
     ):
-        with bigframes.option_context("display.render_mode", "anywidget"):
-            widget = TableWidget(mock_df)
+        widget = TableWidget(mock_df)
 
     assert widget._cell_execution_count == 42
 
@@ -274,19 +276,23 @@ def __init__(self):
 
 
 def test_init_raises_if_anywidget_not_installed():
-    with mock.patch("bigframes.display.anywidget._ANYWIDGET_INSTALLED", False):
-        with pytest.raises(ImportError):
-            from bigframes.display.anywidget import TableWidget
+    with (
+        mock.patch("bigframes.display.anywidget._ANYWIDGET_INSTALLED", False),
+        pytest.raises(ImportError),
+    ):
+        from bigframes.display.anywidget import TableWidget
 
-            TableWidget(mock.Mock())
+        TableWidget(mock.Mock())
 
 
 def test_init_initializes_attributes(mock_df_deferred):
     from bigframes.display.anywidget import TableWidget
 
-    with bigframes.option_context("display.render_mode", "anywidget"):
-        with mock.patch.object(TableWidget, "_initial_load"):
-            widget = TableWidget(mock_df_deferred)
+    with (
+        bigframes.option_context("display.render_mode", "anywidget"),
+        mock.patch.object(TableWidget, "_initial_load"),
+    ):
+        widget = TableWidget(mock_df_deferred)
 
     assert widget._dataframe is mock_df_deferred
     assert widget.page == 0
@@ -328,19 +334,21 @@ def test_validate_page_clamping(mock_df_deferred):
 def test_validate_page_size(mock_df_deferred):
     from bigframes.display.anywidget import TableWidget
 
-    with bigframes.option_context("display.render_mode", "anywidget"):
-        with mock.patch.object(TableWidget, "_initial_load"):
-            widget = TableWidget(mock_df_deferred)
+    with (
+        bigframes.option_context("display.render_mode", "anywidget"),
+        mock.patch.object(TableWidget, "_initial_load"),
+    ):
+        widget = TableWidget(mock_df_deferred)
 
-            widget.page_size = 50
-            assert widget.page_size == 50
+        widget.page_size = 50
+        assert widget.page_size == 50
 
-            original_size = widget.page_size
-            widget.page_size = -5
-            assert widget.page_size == original_size
+        original_size = widget.page_size
+        widget.page_size = -5
+        assert widget.page_size == original_size
 
-            widget.page_size = 10000
-            assert widget.page_size == 1000
+        widget.page_size = 10000
+        assert widget.page_size == 1000
 
 
 def test_page_size_change_resets_page_and_sort(mock_df_deferred):
@@ -373,14 +381,16 @@ def test_page_size_change_resets_batches(mock_df_deferred):
 def test_sort_change_resets_batches(mock_df_deferred):
     from bigframes.display.anywidget import TableWidget
 
-    with bigframes.option_context("display.render_mode", "anywidget"):
-        with mock.patch.object(TableWidget, "_initial_load"):
-            widget = TableWidget(mock_df_deferred)
-            widget._initial_load_complete = True
+    with (
+        bigframes.option_context("display.render_mode", "anywidget"),
+        mock.patch.object(TableWidget, "_initial_load"),
+    ):
+        widget = TableWidget(mock_df_deferred)
+        widget._initial_load_complete = True
 
-            mock_df_deferred.to_pandas_batches.reset_mock()
+        mock_df_deferred.to_pandas_batches.reset_mock()
 
-            widget.sort_context = [{"column": "B", "ascending": False}]
+        widget.sort_context = [{"column": "B", "ascending": False}]
 
     assert mock_df_deferred.to_pandas_batches.call_count >= 1
 
@@ -510,22 +520,24 @@ def test_deferred_mode_execution_in_colab(mock_deferred_df, mock_df_deferred):
     batches.total_rows = 1
     mock_df_deferred.to_pandas_batches.return_value = batches
 
-    with mock.patch.dict(sys.modules, {"google.colab": mock.MagicMock()}):
-        with bigframes.option_context("display.render_mode", "anywidget"):
-            widget = TableWidget(mock_deferred_df)
-            widget.is_deferred_mode = True
+    with (
+        mock.patch.dict(sys.modules, {"google.colab": mock.MagicMock()}),
+        bigframes.option_context("display.render_mode", "anywidget"),
+    ):
+        widget = TableWidget(mock_deferred_df)
+        widget.is_deferred_mode = True
 
-            widget.start_execution = True
+        widget.start_execution = True
 
-            thread = getattr(widget, "_execution_thread", None)
-            if thread is not None:
-                thread.join(timeout=5)
+        thread = getattr(widget, "_execution_thread", None)
+        if thread is not None:
+            thread.join(timeout=5)
 
-            assert widget.is_deferred_mode is True
-            assert widget.table_html == ""
+        assert widget.is_deferred_mode is True
+        assert widget.table_html == ""
 
-            # Simulate frontend ping callback
-            widget.ping = 1
+        # Simulate frontend ping callback
+        widget.ping = 1
 
-            assert widget.is_deferred_mode is False
-            assert widget.table_html != ""
+        assert widget.is_deferred_mode is False
+        assert widget.table_html != ""

packages/bigframes/tests/unit/display/test_html.py

@@ -203,19 +203,20 @@ def test_repr_mimebundle_head():
     )
 
     # Mock _get_obj_metadata
-    with patch("bigframes.display.html._get_obj_metadata", return_value=(False, False)):
-        # Mock create_html_representation and create_text_representation
-        with patch(
+    with (
+        patch("bigframes.display.html._get_obj_metadata", return_value=(False, False)),
+        patch(
             "bigframes.display.html.create_html_representation", return_value="<html>"
-        ) as mock_create_html:
-            with patch(
-                "bigframes.display.plaintext.create_text_representation",
-                return_value="text",
-            ) as mock_create_text:
-                bundle = bf_html.repr_mimebundle_head(mock_df)
-
-                assert bundle == {"text/html": "<html>", "text/plain": "text"}
-                mock_df._prepare_display_df.assert_called_once()
-                mock_df._block.retrieve_repr_request_results.assert_called_once()
-                mock_create_html.assert_called_once()
-                mock_create_text.assert_called_once()
+        ) as mock_create_html,
+        patch(
+            "bigframes.display.plaintext.create_text_representation",
+            return_value="text",
+        ) as mock_create_text,
+    ):
+        bundle = bf_html.repr_mimebundle_head(mock_df)
+
+        assert bundle == {"text/html": "<html>", "text/plain": "text"}
+        mock_df._prepare_display_df.assert_called_once()
+        mock_df._block.retrieve_repr_request_results.assert_called_once()
+        mock_create_html.assert_called_once()
+        mock_create_text.assert_called_once()

packages/bigframes/tests/unit/display/test_render_mode.py

@@ -14,6 +14,8 @@
 
 import unittest.mock as mock
 
+import pytest
+
 import bigframes.display.html as bf_html
 import bigframes.pandas as bpd
 
@@ -62,7 +64,10 @@ def test_repr_mimebundle_selection_logic():
 
         # Test fallback to static deferred repr when anywidget fails
         mock_anywidget.side_effect = Exception("Anywidget failed")
-        with bpd.option_context("display.repr_mode", "deferred"):
+        with (
+            bpd.option_context("display.repr_mode", "deferred"),
+            pytest.warns(UserWarning, match="Anywidget mode is not available"),
+        ):
             bundle = bf_html.repr_mimebundle(mock_obj)
             assert bundle == {"text/plain": "deferred"}
             mock_deferred.assert_called_once()

packages/google-api-core/google/api_core/operations_v1/__init__.py

@@ -14,27 +14,68 @@
 
 """Package for interacting with the google.longrunning.operations meta-API."""
 
-from google.api_core.operations_v1.abstract_operations_client import AbstractOperationsClient
-from google.api_core.operations_v1.operations_async_client import OperationsAsyncClient
-from google.api_core.operations_v1.operations_client import OperationsClient
-from google.api_core.operations_v1.transports.rest import OperationsRestTransport
+import importlib.util
+from typing import Set
+
+try:
+    _has_async_rest = (
+        importlib.util.find_spec("google.auth.aio.transport.sessions") is not None
+    )
+except ModuleNotFoundError:
+    _has_async_rest = False
+
+# PEP 0810: Explicit Lazy Imports
+# Python 3.15+ natively intercepts and defers these imports.
+# Developers can disable this behavior and force eager imports.
+# For more information, see:
+# https://docs.python.org/3.15/library/sys.html#sys.set_lazy_imports_filter
+# Older Python versions safely ignore this variable.
+# NOTE: We statically define all modules here (including async ones) to ensure
+# static analysis tools (mypy, pyright, Ruff) can easily parse them. If async
+# support is not present, the imports are ignored, making their presence safe.
+__lazy_modules__: Set[str] = {
+    "google.api_core.operations_v1.abstract_operations_client",
+    "google.api_core.operations_v1.operations_async_client",
+    "google.api_core.operations_v1.operations_client",
+    "google.api_core.operations_v1.transports.rest",
+    "google.api_core.operations_v1.transports.rest_asyncio",
+    "google.api_core.operations_v1.operations_rest_client_async",
+}
 
 __all__ = [
     "AbstractOperationsClient",
     "OperationsAsyncClient",
     "OperationsClient",
-    "OperationsRestTransport"
+    "OperationsRestTransport",
 ]
 
-try:
-    from google.api_core.operations_v1.transports.rest_asyncio import (
-        AsyncOperationsRestTransport,
-    )
-    from google.api_core.operations_v1.operations_rest_client_async import AsyncOperationsRestClient
-
-    __all__ += ["AsyncOperationsRestClient", "AsyncOperationsRestTransport"]
-except ImportError:
-    # This import requires the `async_rest` extra.
-    # Don't raise an exception if `AsyncOperationsRestTransport` cannot be imported
-    # as other transports are still available.
-    pass
+
+from google.api_core.operations_v1.abstract_operations_client import (  # noqa: E402
+    AbstractOperationsClient,
+)
+from google.api_core.operations_v1.operations_async_client import (  # noqa: E402
+    OperationsAsyncClient,
+)
+from google.api_core.operations_v1.operations_client import (  # noqa: E402
+    OperationsClient,
+)
+from google.api_core.operations_v1.transports.rest import (  # noqa: E402
+    OperationsRestTransport,
+)
+
+if _has_async_rest:
+    try:
+        # On Python 3.15+, PEP 0810 lazy loading means these imports will succeed
+        # instantly (returning a lazy proxy). Any actual ImportErrors (e.g., due to
+        # missing aiohttp/auth dependencies) are deferred until the proxies are accessed.
+        from google.api_core.operations_v1.transports.rest_asyncio import (  # noqa: E402, F401
+            AsyncOperationsRestTransport,
+        )
+        from google.api_core.operations_v1.operations_rest_client_async import (  # noqa: E402, F401
+            AsyncOperationsRestClient,
+        )
+
+        __all__.extend(["AsyncOperationsRestClient", "AsyncOperationsRestTransport"])
+    except ImportError:
+        # Fallback for older python/environments when importlib find_spec succeeds but actual import fails
+        pass

packages/google-apps-chat/google/apps/chat/__init__.py

@@ -43,6 +43,7 @@
     UploadAttachmentRequest,
     UploadAttachmentResponse,
 )
+from google.apps.chat_v1.types.audience import Audience
 from google.apps.chat_v1.types.availability import (
     Availability,
     CustomStatus,
@@ -199,6 +200,7 @@
     "GetAttachmentRequest",
     "UploadAttachmentRequest",
     "UploadAttachmentResponse",
+    "Audience",
     "Availability",
     "CustomStatus",
     "DoNotDisturbMetadata",

packages/google-apps-chat/google/apps/chat_v1/__init__.py

@@ -45,6 +45,7 @@
     UploadAttachmentRequest,
     UploadAttachmentResponse,
 )
+from .types.audience import Audience
 from .types.availability import (
     Availability,
     CustomStatus,
@@ -271,6 +272,7 @@ def _get_version(dependency_name):
     "AttachedGif",
     "Attachment",
     "AttachmentDataRef",
+    "Audience",
     "Availability",
     "CalendarEventLinkData",
     "CardWithId",

packages/google-apps-chat/google/apps/chat_v1/services/chat_service/async_client.py

@@ -2535,6 +2535,27 @@ async def sample_update_space():
                 ``access_settings.audience`` is not supported with
                 ``useAdminAccess``.
 
+                ``access_settings.access_permission_settings``: Updates
+                the `access permission
+                settings <https://support.google.com/chat/answer/11971020>`__
+                of who can discover and join the space where
+                ``spaceType`` field is ``SPACE``. Principals allowed to
+                join the space must also be allowed to discover it. To
+                update access permission settings for a space, the
+                authenticating user must be a space manager or assistant
+                manager and omit all other field masks in the request.
+                You can't update this field if the space is in `import
+                mode <https://developers.google.com/workspace/chat/import-data-overview>`__.
+                To learn more, see `Make a space discoverable to
+                specific
+                users <https://developers.google.com/workspace/chat/space-target-audience>`__.
+                ``access_settings.access_permission_settings`` is not
+                supported with ``useAdminAccess``. The supported field
+                masks include:
+
+                - ``access_settings.access_permission_settings.discoverSpaceSetting``
+                - ``access_settings.access_permission_settings.joinSpaceSetting``
+
                 ``permission_settings``: Supports changing the
                 `permission
                 settings <https://support.google.com/chat/answer/13340792>`__