Skip to content
This repository was archived by the owner on Sep 2, 2025. It is now read-only.
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 7 additions & 3 deletions dbt/adapters/bigquery/connections.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
from concurrent.futures import TimeoutError
import json
import re
from contextlib import contextmanager
Expand Down Expand Up @@ -726,9 +727,12 @@ def _query_and_results(
logger.debug(
self._bq_job_link(query_job.location, query_job.project, query_job.job_id)
)

iterator = query_job.result(max_results=limit, timeout=job_execution_timeout)
return query_job, iterator
try:
iterator = query_job.result(max_results=limit, timeout=job_execution_timeout)
return query_job, iterator
except TimeoutError:
exc = f"Operation did not complete within the designated timeout of {job_execution_timeout} seconds."
raise TimeoutError(exc)

def _retry_and_handle(self, msg, conn, fn):
"""retry a function call within the context of exception_handler."""
Expand Down