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 12
12
# See the License for the specific language governing permissions and
13
13
# limitations under the License.
14
14
15
+ from vertexai .preview import prompts
16
+
15
17
import prompt_create
16
18
import prompt_delete
17
19
import prompt_get
@@ -29,6 +31,7 @@ def test_prompt_template() -> None:
29
31
def test_prompt_create () -> None :
30
32
response = prompt_create .prompt_create ()
31
33
assert response
34
+ prompts .delete (prompt_id = response .prompt_id )
32
35
33
36
34
37
def test_prompt_list_prompts () -> None :
@@ -39,11 +42,14 @@ def test_prompt_list_prompts() -> None:
39
42
def test_prompt_get () -> None :
40
43
get_prompt = prompt_get .get_prompt ()
41
44
assert get_prompt
45
+ prompts .delete (prompt_id = get_prompt .prompt_id )
42
46
43
47
44
48
def test_prompt_list_version () -> None :
45
49
list_versions = prompt_list_version .list_prompt_version ()
46
50
assert list_versions
51
+ for prompt in list_versions :
52
+ prompts .delete (prompt_id = prompt .prompt_id )
47
53
48
54
49
55
def test_prompt_delete () -> None :
You can’t perform that action at this time.
0 commit comments