@@ -385,12 +385,9 @@ def test_create_client_user_agent(self):
385385 client_class = utils .ModelClientWithOverride
386386 )
387387
388- for wrapped_method in client ._transport ._wrapped_methods .values ():
389- # wrapped_method._metadata looks like:
390- # [('x-goog-api-client', 'model-builder/0.3.1 gl-python/3.7.6 grpc/1.30.0 gax/1.22.2 gapic/0.3.1')]
391- user_agent = wrapped_method ._metadata [0 ][1 ]
392- assert "model-builder/" in user_agent
393- assert "google.cloud.aiplatform" in user_agent
388+ user_agent = client ._client_info .to_user_agent ()
389+ assert "model-builder/" in user_agent
390+ assert "google.cloud.aiplatform" in user_agent
394391
395392 def test_create_client_user_agent_top_level_method (self ):
396393 initializer .global_config .init (project = _TEST_PROJECT , location = _TEST_LOCATION )
@@ -404,14 +401,11 @@ def __init__(self):
404401 client_class = utils .ModelClientWithOverride
405402 )
406403
407- for wrapped_method in SomeClass ()._client ._transport ._wrapped_methods .values ():
408- # wrapped_method._metadata looks like:
409- # [('x-goog-api-client', 'model-builder/0.3.1 gl-python/3.7.6 grpc/1.30.0 gax/1.22.2 gapic/0.3.1')]
410- user_agent = wrapped_method ._metadata [0 ][1 ]
411- assert (
412- f"+{ initializer ._TOP_GOOGLE_CONSTRUCTOR_METHOD_TAG } +vertexai.SomeClass.__init__"
413- in user_agent
414- )
404+ user_agent = SomeClass ()._client ._client_info .to_user_agent ()
405+ assert (
406+ f"+{ initializer ._TOP_GOOGLE_CONSTRUCTOR_METHOD_TAG } +vertexai.SomeClass.__init__"
407+ in user_agent
408+ )
415409
416410 def test_create_client_appended_user_agent (self ):
417411 appended_user_agent = ["fake_user_agent" , "another_fake_user_agent" ]
@@ -421,12 +415,9 @@ def test_create_client_appended_user_agent(self):
421415 appended_user_agent = appended_user_agent ,
422416 )
423417
424- for wrapped_method in client ._transport ._wrapped_methods .values ():
425- # wrapped_method._metadata looks like:
426- # [('x-goog-api-client', 'model-builder/0.3.1 gl-python/3.7.6 grpc/1.30.0 gax/1.22.2 gapic/0.3.1')]
427- user_agent = wrapped_method ._metadata [0 ][1 ]
428- assert " " + appended_user_agent [0 ] in user_agent
429- assert " " + appended_user_agent [1 ] in user_agent
418+ user_agent = client ._client_info .to_user_agent ()
419+ assert " " + appended_user_agent [0 ] in user_agent
420+ assert " " + appended_user_agent [1 ] in user_agent
430421
431422 def test_set_api_endpoint (self ):
432423 initializer .global_config .init (
0 commit comments