google-cloud-bigquery-storage 2.40.0__tar.gz → 2.41.0__tar.gz

{google_cloud_bigquery_storage-2.40.0/google_cloud_bigquery_storage.egg-info → google_cloud_bigquery_storage-2.41.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-bigquery-storage
-Version: 2.40.0
+Version: 2.41.0
 Summary: Google Cloud Bigquery Storage API client library
 Home-page: https://github.com/googleapis/google-cloud-python/tree/main/packages/google-cloud-bigquery-storage
 Author: Google LLC
@@ -28,9 +28,11 @@ Requires-Dist: grpcio<2.0.0,>=1.75.1; python_version >= "3.14"
 Requires-Dist: proto-plus<2.0.0,>=1.26.1
 Requires-Dist: protobuf<8.0.0,>=6.33.5
 Provides-Extra: pandas
-Requires-Dist: pandas>=1.1.3; extra == "pandas"
+Requires-Dist: pandas<3.0.0,>=1.1.3; extra == "pandas"
+Requires-Dist: pyarrow>=3.0.0; extra == "pandas"
+Requires-Dist: pandas-gbq<2.0.0,>=0.35.1; extra == "pandas"
 Provides-Extra: fastavro
-Requires-Dist: fastavro>=1.1.0; extra == "fastavro"
+Requires-Dist: fastavro<2.0.0,>=1.1.0; extra == "fastavro"
 Provides-Extra: pyarrow
 Requires-Dist: pyarrow>=3.0.0; extra == "pyarrow"
 Dynamic: author

{google_cloud_bigquery_storage-2.40.0 → google_cloud_bigquery_storage-2.41.0}/google/cloud/bigquery_storage/__init__.py

@@ -43,6 +43,7 @@ from google.cloud.bigquery_storage_v1.types.storage import (
     AppendRowsResponse,
     BatchCommitWriteStreamsRequest,
     BatchCommitWriteStreamsResponse,
+    ClientStats,
     CreateReadSessionRequest,
     CreateWriteStreamRequest,
     FinalizeWriteStreamRequest,
@@ -87,6 +88,7 @@ __all__ = (
     "AppendRowsResponse",
     "BatchCommitWriteStreamsRequest",
     "BatchCommitWriteStreamsResponse",
+    "ClientStats",
     "CreateReadSessionRequest",
     "CreateWriteStreamRequest",
     "FinalizeWriteStreamRequest",

{google_cloud_bigquery_storage-2.40.0 → google_cloud_bigquery_storage-2.41.0}/google/cloud/bigquery_storage/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "2.40.0"  # {x-release-please-version}
+__version__ = "2.41.0"  # {x-release-please-version}

{google_cloud_bigquery_storage-2.40.0/google/cloud/bigquery_storage_v1alpha → google_cloud_bigquery_storage-2.41.0/google/cloud/bigquery_storage_v1}/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "2.40.0"  # {x-release-please-version}
+__version__ = "2.41.0"  # {x-release-please-version}

{google_cloud_bigquery_storage-2.40.0 → google_cloud_bigquery_storage-2.41.0}/google/cloud/bigquery_storage_v1/reader.py

@@ -617,6 +617,12 @@ class ReadRowsPage(object):
             pandas.DataFrame:
                 A data frame of all rows in the stream.
         """
+        warnings.warn(
+            "Retrieving DataFrames directly via google-cloud-bigquery-storage is deprecated. "
+            "Please use 'pandas_gbq.read_gbq()' or 'pandas_gbq.pandas.read_bigquery_stream_batches()'.",
+            PendingDeprecationWarning,
+            stacklevel=2,
+        )
         if pandas is None:
             raise ImportError(_PANDAS_REQUIRED)
 

{google_cloud_bigquery_storage-2.40.0 → google_cloud_bigquery_storage-2.41.0}/google/cloud/bigquery_storage_v1/services/big_query_write/async_client.py

@@ -524,7 +524,7 @@ class BigQueryWriteAsyncClient:
                 within the same connection for the default stream.
 
                 The size of a single AppendRowsRequest must be less than
-                10 MB in size. Requests larger than this return an
+                20 MB in size. Requests larger than this return an
                 error, typically ``INVALID_ARGUMENT``.
             retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if any,
                 should be retried.

{google_cloud_bigquery_storage-2.40.0 → google_cloud_bigquery_storage-2.41.0}/google/cloud/bigquery_storage_v1/services/big_query_write/client.py

@@ -974,7 +974,7 @@ class BigQueryWriteClient(metaclass=BigQueryWriteClientMeta):
                 within the same connection for the default stream.
 
                 The size of a single AppendRowsRequest must be less than
-                10 MB in size. Requests larger than this return an
+                20 MB in size. Requests larger than this return an
                 error, typically ``INVALID_ARGUMENT``.
             retry (google.api_core.retry.Retry): Designation of what errors, if any,
                 should be retried.

{google_cloud_bigquery_storage-2.40.0 → google_cloud_bigquery_storage-2.41.0}/google/cloud/bigquery_storage_v1/types/__init__.py

@@ -32,6 +32,7 @@ from .storage import (
     AppendRowsResponse,
     BatchCommitWriteStreamsRequest,
     BatchCommitWriteStreamsResponse,
+    ClientStats,
     CreateReadSessionRequest,
     CreateWriteStreamRequest,
     FinalizeWriteStreamRequest,
@@ -73,6 +74,7 @@ __all__ = (
     "AppendRowsResponse",
     "BatchCommitWriteStreamsRequest",
     "BatchCommitWriteStreamsResponse",
+    "ClientStats",
     "CreateReadSessionRequest",
     "CreateWriteStreamRequest",
     "FinalizeWriteStreamRequest",

{google_cloud_bigquery_storage-2.40.0 → google_cloud_bigquery_storage-2.41.0}/google/cloud/bigquery_storage_v1/types/storage.py

@@ -46,6 +46,7 @@ __protobuf__ = proto.module(
         "FlushRowsResponse",
         "StorageError",
         "RowError",
+        "ClientStats",
     },
 )
 
@@ -108,6 +109,8 @@ class CreateReadSessionRequest(proto.Message):
 class ReadRowsRequest(proto.Message):
     r"""Request message for ``ReadRows``.
 
+    .. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
+
     Attributes:
         read_stream (str):
             Required. Stream to read rows from.
@@ -116,6 +119,12 @@ class ReadRowsRequest(proto.Message):
             last row read from Read. Requesting a larger
             offset is undefined. If not specified, start
             reading from offset zero.
+        arrow_serialization_options (google.cloud.bigquery_storage_v1.types.ArrowSerializationOptions):
+            Optional. Options specific to the Apache
+            Arrow output format.
+            This feature is not yet available.
+
+            This field is a member of `oneof`_ ``output_format_serialization_options``.
     """
 
     read_stream: str = proto.Field(
@@ -126,6 +135,12 @@ class ReadRowsRequest(proto.Message):
         proto.INT64,
         number=2,
     )
+    arrow_serialization_options: arrow.ArrowSerializationOptions = proto.Field(
+        proto.MESSAGE,
+        number=5,
+        oneof="output_format_serialization_options",
+        message=arrow.ArrowSerializationOptions,
+    )
 
 
 class ThrottleState(proto.Message):
@@ -249,6 +264,14 @@ class ReadRowsResponse(proto.Message):
             client should skip decompression.
 
             This field is a member of `oneof`_ ``_uncompressed_byte_size``.
+        total_estimated_row_count (int):
+            Output only. The total estimated number of
+            rows in the query results. Only populated when
+            reading data from a BigQuery job.
+
+            This feature is not yet available.
+
+            This field is a member of `oneof`_ ``_total_estimated_row_count``.
     """
 
     avro_rows: avro.AvroRows = proto.Field(
@@ -294,6 +317,11 @@ class ReadRowsResponse(proto.Message):
         number=9,
         optional=True,
     )
+    total_estimated_row_count: int = proto.Field(
+        proto.INT64,
+        number=10,
+        optional=True,
+    )
 
 
 class SplitReadStreamRequest(proto.Message):
@@ -385,7 +413,7 @@ class AppendRowsRequest(proto.Message):
     table destinations within the same connection for the default
     stream.
 
-    The size of a single AppendRowsRequest must be less than 10 MB in
+    The size of a single AppendRowsRequest must be less than 20 MB in
     size. Requests larger than this return an error, typically
     ``INVALID_ARGUMENT``.
 
@@ -481,6 +509,9 @@ class AppendRowsRequest(proto.Message):
             ``DEFAULT_VALUE`` and at the same time, set
             ``missing_value_interpretations`` to ``NULL_VALUE`` on those
             columns.
+        client_stats (google.cloud.bigquery_storage_v1.types.ClientStats):
+            Optional. Stats and telemetry data gathered
+            on the client side.
     """
 
     class MissingValueInterpretation(proto.Enum):
@@ -599,6 +630,11 @@ class AppendRowsRequest(proto.Message):
         number=8,
         enum=MissingValueInterpretation,
     )
+    client_stats: "ClientStats" = proto.Field(
+        proto.MESSAGE,
+        number=9,
+        message="ClientStats",
+    )
 
 
 class AppendRowsResponse(proto.Message):
@@ -981,4 +1017,157 @@ class RowError(proto.Message):
     )
 
 
+class ClientStats(proto.Message):
+    r"""Stats and telemetry data gathered on the client side about
+    requests being sent to the BigQuery Storage service, for
+    internal use only.
+
+    Attributes:
+        request_stats (google.cloud.bigquery_storage_v1.types.ClientStats.RequestStats):
+            Optional. Per-request stats.
+        window_stats (google.cloud.bigquery_storage_v1.types.ClientStats.WindowStats):
+            Optional. Windowed stats.
+    """
+
+    class RequestStats(proto.Message):
+        r"""Stats and telemetry data gathered on the client side about a
+        single request.
+
+
+        .. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
+
+        Attributes:
+            send_time_millis (int):
+                Optional. Timestamp indicating when the
+                request was sent over the network, expressed in
+                epoch milliseconds.
+
+                This field is a member of `oneof`_ ``_send_time_millis``.
+            queued_requests_count (int):
+                Optional. Number of pending requests at the
+                moment this request was sent. This includes
+                requests waiting to be sent, and those that are
+                inflight.
+
+                This field is a member of `oneof`_ ``_queued_requests_count``.
+        """
+
+        send_time_millis: int = proto.Field(
+            proto.INT64,
+            number=1,
+            optional=True,
+        )
+        queued_requests_count: int = proto.Field(
+            proto.INT64,
+            number=2,
+            optional=True,
+        )
+
+    class WindowStats(proto.Message):
+        r"""Aggregate connection metrics over a window interval.
+
+        .. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
+
+        Attributes:
+            max_response_latency_millis (int):
+                Optional. The maximum response latency
+                observed in the window, expressed in
+                milliseconds.
+
+                This field is a member of `oneof`_ ``_max_response_latency_millis``.
+            avg_response_latency_millis (int):
+                Optional. The average response latency
+                observed in the window, expressed in
+                milliseconds.
+
+                This field is a member of `oneof`_ ``_avg_response_latency_millis``.
+            longest_wait_no_response_millis (int):
+                Optional. The longest time spent waiting without receiving a
+                response in the window. This could exceed
+                max_response_latency_millis because the latter is evaluated
+                only when a response is received. Expressed in milliseconds.
+
+                This field is a member of `oneof`_ ``_longest_wait_no_response_millis``.
+            requests_sent_count (int):
+                Optional. How many requests were sent in the
+                window.
+
+                This field is a member of `oneof`_ ``_requests_sent_count``.
+            responses_received_count (int):
+                Optional. How many responses were received in
+                the window.
+
+                This field is a member of `oneof`_ ``_responses_received_count``.
+            bytes_sent_count (int):
+                Optional. How many bytes were sent in the
+                window.
+
+                This field is a member of `oneof`_ ``_bytes_sent_count``.
+            window_start_time_epoch_millis (int):
+                Optional. Start time of the window interval
+                for which these stats are aggregated, expressed
+                in epoch milliseconds.
+
+                This field is a member of `oneof`_ ``_window_start_time_epoch_millis``.
+            window_millis (int):
+                Optional. Duration of the window interval for
+                which these stats are aggregated, expressed in
+                milliseconds.
+
+                This field is a member of `oneof`_ ``_window_millis``.
+        """
+
+        max_response_latency_millis: int = proto.Field(
+            proto.INT64,
+            number=1,
+            optional=True,
+        )
+        avg_response_latency_millis: int = proto.Field(
+            proto.INT64,
+            number=2,
+            optional=True,
+        )
+        longest_wait_no_response_millis: int = proto.Field(
+            proto.INT64,
+            number=3,
+            optional=True,
+        )
+        requests_sent_count: int = proto.Field(
+            proto.INT64,
+            number=4,
+            optional=True,
+        )
+        responses_received_count: int = proto.Field(
+            proto.INT64,
+            number=5,
+            optional=True,
+        )
+        bytes_sent_count: int = proto.Field(
+            proto.INT64,
+            number=6,
+            optional=True,
+        )
+        window_start_time_epoch_millis: int = proto.Field(
+            proto.INT64,
+            number=7,
+            optional=True,
+        )
+        window_millis: int = proto.Field(
+            proto.INT64,
+            number=8,
+            optional=True,
+        )
+
+    request_stats: RequestStats = proto.Field(
+        proto.MESSAGE,
+        number=1,
+        message=RequestStats,
+    )
+    window_stats: WindowStats = proto.Field(
+        proto.MESSAGE,
+        number=2,
+        message=WindowStats,
+    )
+
+
 __all__ = tuple(sorted(__protobuf__.manifest))

{google_cloud_bigquery_storage-2.40.0/google/cloud/bigquery_storage_v1beta → google_cloud_bigquery_storage-2.41.0/google/cloud/bigquery_storage_v1alpha}/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "2.40.0"  # {x-release-please-version}
+__version__ = "2.41.0"  # {x-release-please-version}

{google_cloud_bigquery_storage-2.40.0/google/cloud/bigquery_storage_v1 → google_cloud_bigquery_storage-2.41.0/google/cloud/bigquery_storage_v1beta}/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "2.40.0"  # {x-release-please-version}
+__version__ = "2.41.0"  # {x-release-please-version}

google_cloud_bigquery_storage-2.41.0/google/cloud/bigquery_storage_v1beta2/gapic_version.py

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+__version__ = "2.41.0"  # {x-release-please-version}

{google_cloud_bigquery_storage-2.40.0 → google_cloud_bigquery_storage-2.41.0/google_cloud_bigquery_storage.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-bigquery-storage
-Version: 2.40.0
+Version: 2.41.0
 Summary: Google Cloud Bigquery Storage API client library
 Home-page: https://github.com/googleapis/google-cloud-python/tree/main/packages/google-cloud-bigquery-storage
 Author: Google LLC
@@ -28,9 +28,11 @@ Requires-Dist: grpcio<2.0.0,>=1.75.1; python_version >= "3.14"
 Requires-Dist: proto-plus<2.0.0,>=1.26.1
 Requires-Dist: protobuf<8.0.0,>=6.33.5
 Provides-Extra: pandas
-Requires-Dist: pandas>=1.1.3; extra == "pandas"
+Requires-Dist: pandas<3.0.0,>=1.1.3; extra == "pandas"
+Requires-Dist: pyarrow>=3.0.0; extra == "pandas"
+Requires-Dist: pandas-gbq<2.0.0,>=0.35.1; extra == "pandas"
 Provides-Extra: fastavro
-Requires-Dist: fastavro>=1.1.0; extra == "fastavro"
+Requires-Dist: fastavro<2.0.0,>=1.1.0; extra == "fastavro"
 Provides-Extra: pyarrow
 Requires-Dist: pyarrow>=3.0.0; extra == "pyarrow"
 Dynamic: author

{google_cloud_bigquery_storage-2.40.0 → google_cloud_bigquery_storage-2.41.0}/google_cloud_bigquery_storage.egg-info/SOURCES.txt

@@ -120,6 +120,7 @@ tests/unit/__init__.py
 tests/unit/helpers.py
 tests/unit/test_packaging.py
 tests/unit/test_read_client_v1.py
+tests/unit/test_reader_pandas_gbq.py
 tests/unit/test_reader_v1.py
 tests/unit/test_reader_v1_arrow.py
 tests/unit/test_writer_v1.py

{google_cloud_bigquery_storage-2.40.0 → google_cloud_bigquery_storage-2.41.0}/google_cloud_bigquery_storage.egg-info/requires.txt

@@ -8,10 +8,12 @@ protobuf<8.0.0,>=6.33.5
 grpcio<2.0.0,>=1.75.1
 
 [fastavro]
-fastavro>=1.1.0
+fastavro<2.0.0,>=1.1.0
 
 [pandas]
-pandas>=1.1.3
+pandas<3.0.0,>=1.1.3
+pyarrow>=3.0.0
+pandas-gbq<2.0.0,>=0.35.1
 
 [pyarrow]
 pyarrow>=3.0.0

{google_cloud_bigquery_storage-2.40.0 → google_cloud_bigquery_storage-2.41.0}/setup.py

@@ -54,9 +54,13 @@ dependencies = [
     "protobuf >= 6.33.5, < 8.0.0",
 ]
 extras = {
-    "pandas": ["pandas>=1.1.3"],
-    "fastavro": ["fastavro>=1.1.0"],
-    "pyarrow": ["pyarrow>=3.0.0"],
+    "pandas": [
+        "pandas >= 1.1.3, < 3.0.0",
+        "pyarrow >= 3.0.0",
+        "pandas-gbq >= 0.35.1, < 2.0.0",
+    ],
+    "fastavro": ["fastavro >= 1.1.0, < 2.0.0"],
+    "pyarrow": ["pyarrow >= 3.0.0"],
 }
 url = "https://github.com/googleapis/google-cloud-python/tree/main/packages/google-cloud-bigquery-storage"
 

{google_cloud_bigquery_storage-2.40.0 → google_cloud_bigquery_storage-2.41.0}/tests/unit/test_read_client_v1.py

@@ -158,24 +158,15 @@ def test_read_rows(mock_transport, client_under_test):
     ["google.cloud.bigquery_storage_v1", "google.cloud.bigquery_storage_v1beta2"],
 )
 def test_init_default_client_info(module_under_test):
-    from google.api_core.gapic_v1.client_info import METRICS_METADATA_KEY
-
     mut = importlib.import_module(module_under_test)
-
     creds = mock.Mock(spec=credentials.Credentials)
-    client = mut.BigQueryWriteClient(credentials=creds)
+    expected_client_info = f"gccl/{mut.__version__}"
 
-    installed_version = mut.__version__
-    expected_client_info = f"gccl/{installed_version}"
+    with mock.patch("google.api_core.gapic_v1.method.wrap_method") as mock_wrap:
+        mut.BigQueryWriteClient(credentials=creds)
 
-    for wrapped_method in client.transport._wrapped_methods.values():
-        user_agent = next(
-            (
-                header_value
-                for header, header_value in wrapped_method._metadata
-                if header == METRICS_METADATA_KEY
-            ),
-            None,
-        )
-        assert user_agent is not None
-        assert expected_client_info in user_agent
+        assert mock_wrap.call_count > 0
+        for call in mock_wrap.call_args_list:
+            client_info = call.kwargs.get("client_info")
+            assert client_info is not None
+            assert expected_client_info in client_info.to_user_agent()