fix: updates timeout/retry code to respect hanging server

[chalmerlowe]

Description

This PR fixes a crash when handling _InactiveRpcError during retry logic and ensures proper timeout propagation in RowIterator.to_dataframe.

Fixes

Retry Logic Crash: Addressed an issue in google/cloud/bigquery/retry.py where _should_retry would raise a TypeError when inspecting unstructured gRPC errors (like _InactiveRpcError). The fix adds robust error inspection to fallback gracefully when exc.errors is not subscriptable.

Timeout Propagation: Added the missing timeout parameter to RowIterator.to_dataframe in google/cloud/bigquery/table.py. This ensures that the user-specified timeout is correctly passed down to the underlying to_arrow call, preventing the client from hanging indefinitely when the Storage API is unresponsive.

Changes

Modified google/cloud/bigquery/retry.py: Updated _should_retry to handle TypeError and KeyError when accessing exc.errors.
Modified google/cloud/bigquery/table.py: Updated RowIterator.to_dataframe signature and implementation to accept and pass the timeout parameter.

The first half of this work was completed in PR #2354

[gemini-code-assist]

Summary of Changes

Hello @chalmerlowe, I'm Gemini Code Assist¹! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request significantly improves the stability and predictability of the BigQuery client library by addressing critical issues in error handling and timeout management. It ensures that the retry mechanism can robustly handle various gRPC error formats without crashing and guarantees that data retrieval operations from the BigQuery Storage API will respect specified timeouts, preventing unresponsive client states.

Highlights

• Robustified Retry Logic: The _should_retry function in google/cloud/bigquery/retry.py has been enhanced with a try-except block to gracefully handle unstructured gRPC errors, such as _InactiveRpcError. This prevents TypeError or KeyError crashes when exc.errors is missing, empty, or malformed, ensuring more resilient error inspection.
• Improved Timeout Propagation: The _download_table_bqstorage function in google/cloud/bigquery/_pandas_helpers.py now correctly propagates the timeout parameter to the bqstorage_client.create_read_session call. This ensures that user-specified timeouts are respected during BigQuery Storage API reads, preventing indefinite hangs.
• New Unit Test for Timeout: A new unit test, test_download_table_bqstorage_passes_timeout_to_create_read_session, has been added to verify that the timeout parameter and a corresponding retry policy with the correct deadline are properly passed to the BigQuery Storage API client during data download operations.

Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature	Command	Description
Code Review	/gemini review	Performs a code review for the current pull request in its current state.
Pull Request Summary	/gemini summary	Provides a summary of the current pull request in its current state.
Comment	@gemini-code-assist	Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help	/gemini help	Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for Github and other Google products, sign up here.

You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension.

Footnotes

1. Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution. ↩

[gemini-code-assist]

Code Review

This pull request addresses a crash in retry logic and ensures proper timeout propagation. The changes in google/cloud/bigquery/retry.py add robust error inspection to handle unstructured gRPC errors, while the updates in google/cloud/bigquery/table.py and google/cloud/bigquery/_pandas_helpers.py ensure that the user-specified timeout is correctly passed down to the underlying to_arrow and create_read_session calls. A new test case is added to verify the timeout propagation.

[gemini-code-assist]

The try...except block catches multiple exception types. Consider logging the specific exception that occurred for better debugging and monitoring.