File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1919import prompt_list_version
2020# import prompt_restore_version
2121import prompt_template
22+ from vertexai .preview import prompts
2223
2324
2425def test_prompt_template () -> None :
@@ -29,6 +30,7 @@ def test_prompt_template() -> None:
2930def test_prompt_create () -> None :
3031 response = prompt_create .prompt_create ()
3132 assert response
33+ prompts .delete (prompt_id = response .prompt_id )
3234
3335
3436def test_prompt_list_prompts () -> None :
@@ -39,11 +41,14 @@ def test_prompt_list_prompts() -> None:
3941def 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
4447def 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
4954def test_prompt_delete () -> None :
You can’t perform that action at this time.
0 commit comments