[gcp][feat] Add Vertex AI collection (#2211)

Co-authored-by: Matthias Veit <aquamatthias@users.noreply.github.com>

Author: 1101-1

plugins/gcp/fix_plugin_gcp/collector.py

@@ -4,7 +4,7 @@
 
 from fix_plugin_gcp.config import GcpConfig
 from fix_plugin_gcp.gcp_client import GcpApiSpec
-from fix_plugin_gcp.resources import compute, container, billing, sqladmin, storage
+from fix_plugin_gcp.resources import compute, container, billing, sqladmin, storage, aiplatform
 from fix_plugin_gcp.resources.base import GcpResource, GcpProject, ExecutorQueue, GraphBuilder, GcpRegion, GcpZone
 from fix_plugin_gcp.utils import Credentials
 from fixlib.baseresources import Cloud
@@ -13,7 +13,12 @@
 
 log = logging.getLogger("fix.plugins.gcp")
 all_resources: List[Type[GcpResource]] = (
-    compute.resources + container.resources + billing.resources + sqladmin.resources + storage.resources
+    compute.resources
+    + container.resources
+    + billing.resources
+    + sqladmin.resources
+    + storage.resources
+    + aiplatform.resources
 )
 
 

plugins/gcp/fix_plugin_gcp/gcp_client.py

@@ -30,6 +30,7 @@ class GcpApiSpec:
     response_path: str
     response_regional_sub_path: Optional[str] = None
     set_label_identifier: str = "resource"
+    service_with_region_prefix: bool = False
     get_identifier: Optional[str] = None
     delete_identifier: Optional[str] = None
     required_iam_permissions: Optional[List[str]] = None
@@ -178,6 +179,8 @@ def next_responses(request: Any) -> None:
             ):
                 return next_responses(nxt_req)
 
+        if api_spec.service_with_region_prefix and isinstance(executor._baseUrl, str):
+            executor._baseUrl = executor._baseUrl.replace(api_spec.service, f"{self.region}-{api_spec.service}", 1)
         next_responses(getattr(executor, api_spec.action)(**params))
         return result