File tree Expand file tree Collapse file tree 1 file changed +6
-0
lines changed
Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change 1212# See the License for the specific language governing permissions and
1313# limitations under the License.
1414
15+ from vertexai .preview import prompts
16+
1517import prompt_create
1618import prompt_delete
1719import prompt_get
@@ -29,6 +31,7 @@ def test_prompt_template() -> None:
2931def test_prompt_create () -> None :
3032 response = prompt_create .prompt_create ()
3133 assert response
34+ prompts .delete (prompt_id = response .prompt_id )
3235
3336
3437def test_prompt_list_prompts () -> None :
@@ -39,11 +42,14 @@ def test_prompt_list_prompts() -> None:
3942def 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
4448def 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
4955def test_prompt_delete () -> None :
You can’t perform that action at this time.
0 commit comments