Skip to content

Commit e0f7250

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
chore: Add delete log utils.
PiperOrigin-RevId: 611235350
1 parent 4739118 commit e0f7250

File tree

1 file changed

+33
-0
lines changed

1 file changed

+33
-0
lines changed

google/cloud/aiplatform/base.py

+33
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,39 @@ def log_create_complete_with_getter(
129129
self.info(f"To use this {cls.__name__} in another session:")
130130
self.info(f"{variable_name} = aiplatform.{cls.__name__}.get('{resource.name}')")
131131

132+
def log_delete_with_lro(
133+
self,
134+
resource: Type["VertexAiResourceNoun"],
135+
lro: Optional[operation.Operation] = None,
136+
):
137+
"""Logs delete event with LRO.
138+
139+
Args:
140+
resource: Vertex AI resource that will be deleted.
141+
lro: Backing LRO for creation.
142+
"""
143+
self.info(
144+
f"Deleting {resource.__class__.__name__} resource: {resource.resource_name}"
145+
)
146+
147+
if lro:
148+
self.info(
149+
f"Delete {resource.__class__.__name__} backing LRO: {lro.operation.name}"
150+
)
151+
152+
def log_delete_complete(
153+
self,
154+
resource: Type["VertexAiResourceNoun"],
155+
):
156+
"""Logs delete event is complete.
157+
158+
Args:
159+
resource: Vertex AI resource that was deleted.
160+
"""
161+
self.info(
162+
f"{resource.__class__.__name__} resource {resource.resource_name} deleted."
163+
)
164+
132165
def log_action_start_against_resource(
133166
self, action: str, noun: str, resource_noun_obj: "VertexAiResourceNoun"
134167
):

0 commit comments

Comments
 (0)