Skip to content

Commit 73878c6

Browse files
committed
chore(ci): cleanup prompt resources
1 parent c549fc7 commit 73878c6

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

generative_ai/prompts/test_prompt_template.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
import prompt_list_version
2020
# import prompt_restore_version
2121
import prompt_template
22+
from vertexai.preview import prompts
2223

2324

2425
def test_prompt_template() -> None:
@@ -29,6 +30,7 @@ def test_prompt_template() -> None:
2930
def test_prompt_create() -> None:
3031
response = prompt_create.prompt_create()
3132
assert response
33+
prompts.delete(prompt_id = response.prompt_id)
3234

3335

3436
def test_prompt_list_prompts() -> None:
@@ -39,11 +41,14 @@ def test_prompt_list_prompts() -> None:
3941
def test_prompt_get() -> None:
4042
get_prompt = prompt_get.get_prompt()
4143
assert get_prompt
44+
prompts.delete(prompt_id = get_prompt.prompt_id)
4245

4346

4447
def test_prompt_list_version() -> None:
4548
list_versions = prompt_list_version.list_prompt_version()
4649
assert list_versions
50+
for prompt in list_versions:
51+
prompts.delete(prompt_id = prompt.prompt_id)
4752

4853

4954
def test_prompt_delete() -> None:

0 commit comments

Comments
 (0)