From ca696fbeac1f7be841bcc53f1f8aac3ec042979c Mon Sep 17 00:00:00 2001 From: mrishi Date: Tue, 12 May 2026 12:22:48 +0530 Subject: [PATCH 1/3] fix(feature_store): safely parse bq:// URIs using removeprefix The previous implementation used `.lstrip("bq://")`, which greedily dropped valid characters (like 'b' or 'q') from the beginning of Google Cloud Project IDs. Replaced `lstrip` with `removeprefix` to explicitly remove only the protocol string without mutating the project ID. --- vertexai/resources/preview/feature_store/offline_store.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vertexai/resources/preview/feature_store/offline_store.py b/vertexai/resources/preview/feature_store/offline_store.py index 3a785711e4..ad5bf193e7 100644 --- a/vertexai/resources/preview/feature_store/offline_store.py +++ b/vertexai/resources/preview/feature_store/offline_store.py @@ -118,7 +118,8 @@ def _feature_to_data_source( bq_uri = feature_group._gca_resource.big_query.big_query_source.input_uri assert bq_uri - fully_qualified_table = bq_uri.lstrip("bq://") + fully_qualified_table = bq_uri.removeprefix("bq://") + assert fully_qualified_table query = ( From 804ce2fa1cc164e13855432382044f0d06568d4d Mon Sep 17 00:00:00 2001 From: mrishi Date: Thu, 11 Jun 2026 06:55:29 +0530 Subject: [PATCH 2/3] fix(offline_store): safely parse bq:// URIs using removeprefix --- agentplatform/resources/preview/feature_store/offline_store.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/agentplatform/resources/preview/feature_store/offline_store.py b/agentplatform/resources/preview/feature_store/offline_store.py index 4ba369d168..1e55ae127c 100644 --- a/agentplatform/resources/preview/feature_store/offline_store.py +++ b/agentplatform/resources/preview/feature_store/offline_store.py @@ -116,7 +116,7 @@ def _feature_to_data_source( bq_uri = feature_group._gca_resource.big_query.big_query_source.input_uri assert bq_uri - fully_qualified_table = bq_uri.lstrip("bq://") + fully_qualified_table = bq_uri.removeprefix("bq://") assert fully_qualified_table query = ( From 9bb798fbea333b3ff4ac6f290320a0646e0c7c81 Mon Sep 17 00:00:00 2001 From: mrishi Date: Thu, 11 Jun 2026 06:55:54 +0530 Subject: [PATCH 3/3] fix(offline_store): safely parse bq:// URIs using removeprefix --- google/cloud/aiplatform/preview/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/google/cloud/aiplatform/preview/datasets.py b/google/cloud/aiplatform/preview/datasets.py index 1616abb991..5e0bde0340 100644 --- a/google/cloud/aiplatform/preview/datasets.py +++ b/google/cloud/aiplatform/preview/datasets.py @@ -1409,7 +1409,7 @@ def assemble( ) result = assemble_lro.result(timeout=None) _LOGGER.log_action_completed_against_resource("data", "assembled", self) - table_id = result.bigquery_destination.lstrip("bq://") + table_id = result.bigquery_destination.removeprefix("bq://") if load_dataframe: session_options = bigframes.BigQueryOptions( credentials=initializer.global_config.credentials,