Copybara import of the project:

--
ca696fb by mrishi <rishisapps@gmail.com>:

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.

--
804ce2f by mrishi <rishisapps@gmail.com>:

fix(offline_store): safely parse bq:// URIs using removeprefix

--
9bb798f by mrishi <rishisapps@gmail.com>:

fix(offline_store): safely parse bq:// URIs using removeprefix

--
f7e9447 by mrishi <rishisapps@gmail.com>:

fix(aiplatform): safely parse bq:// URIs using removeprefix

--
8f7aa97 by mrishi <rishisapps@gmail.com>:

fix(ai_platform): Black check

COPYBARA_INTEGRATE_REVIEW=#6771 from mrishi:fix-bq-uri-parsing c922bbb
PiperOrigin-RevId: 936106848

Author: mrishi

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 = (

google/cloud/aiplatform/preview/datasets.py

@@ -39,7 +39,6 @@
 from google.protobuf import struct_pb2
 from google.protobuf import json_format
 
-
 _MULTIMODAL_METADATA_SCHEMA_URI = (
     "gs://google-cloud-aiplatform/schema/dataset/metadata/multimodal_1.0.0.yaml"
 )
@@ -1122,7 +1121,9 @@ def to_bigframes(self) -> "bigframes.pandas.DataFrame":  # type: ignore # noqa:
             A BigFrames dataframe.
         """
         bigframes = _try_import_bigframes()
-        return bigframes.pandas.read_gbq_table(self.bigquery_table.lstrip("bq://"))
+        return bigframes.pandas.read_gbq_table(
+            self.bigquery_table.removeprefix("bq://")
+        )
 
     @classmethod
     @base.optional_sync()
@@ -1409,7 +1410,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,

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 = (