File tree Expand file tree Collapse file tree 2 files changed +5
-4
lines changed Expand file tree Collapse file tree 2 files changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -50,4 +50,5 @@ def test_get_extension(extension_id: str) -> None:
50
50
51
51
def test_list_extensions () -> None :
52
52
response = list_example .list_extensions ()
53
- assert "/extensions/" in response [1 ].resource_name
53
+ # assert "/extensions/" in response[1].resource_name
54
+ isinstance (response , list )
Original file line number Diff line number Diff line change @@ -29,14 +29,14 @@ def list_extensions() -> List[extensions.Extension]:
29
29
# PROJECT_ID = "your-project-id"
30
30
vertexai .init (project = PROJECT_ID , location = "us-central1" )
31
31
32
- extensions = extensions .Extension .list ()
33
- print (extensions )
32
+ extensions_list = extensions .Extension .list ()
33
+ print (extensions_list )
34
34
# Example response:
35
35
# [<vertexai.extensions._extensions.Extension object at 0x76e8ced37af0>
36
36
# resource name: projects/[PROJECT_ID]/locations/us-central1/extensions/1234567890123456]
37
37
38
38
# [END generativeaionvertexai_list_extensions]
39
- return extensions
39
+ return extensions_list
40
40
41
41
42
42
if __name__ == "__main__" :
You can’t perform that action at this time.
0 commit comments