Skip to content

Commit 1158bb0

Browse files
authored
chore(ci): cleanup prompt resources (#13458)
1 parent 311a466 commit 1158bb0

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

generative_ai/prompts/test_prompt_template.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,8 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15+
from vertexai.preview import prompts
16+
1517
import prompt_create
1618
import prompt_delete
1719
import prompt_get
@@ -29,6 +31,7 @@ def test_prompt_template() -> None:
2931
def test_prompt_create() -> None:
3032
response = prompt_create.prompt_create()
3133
assert response
34+
prompts.delete(prompt_id=response.prompt_id)
3235

3336

3437
def test_prompt_list_prompts() -> None:
@@ -39,11 +42,14 @@ def test_prompt_list_prompts() -> None:
3942
def test_prompt_get() -> None:
4043
get_prompt = prompt_get.get_prompt()
4144
assert get_prompt
45+
prompts.delete(prompt_id=get_prompt.prompt_id)
4246

4347

4448
def test_prompt_list_version() -> None:
4549
list_versions = prompt_list_version.list_prompt_version()
4650
assert list_versions
51+
for prompt in list_versions:
52+
prompts.delete(prompt_id=prompt.prompt_id)
4753

4854

4955
def test_prompt_delete() -> None:

0 commit comments

Comments
 (0)