File tree 1 file changed +5
-7
lines changed
1 file changed +5
-7
lines changed Original file line number Diff line number Diff line change @@ -414,19 +414,17 @@ def test_generate_content_from_text_and_remote_video(
414
414
def test_generate_content_from_text_and_remote_audio (
415
415
self , api_endpoint_env_name , api_transport
416
416
):
417
- vision_model = generative_models .GenerativeModel (GEMINI_15_PRO_2_MODEL_NAME )
417
+ pro_model = generative_models .GenerativeModel (GEMINI_15_PRO_2_MODEL_NAME )
418
418
audio_part = generative_models .Part .from_uri (
419
- uri = "gs://cloud-samples-data/audio/speech_16k.wav " ,
420
- mime_type = "audio/wav " ,
419
+ uri = "gs://cloud-samples-data/generative-ai/ audio/pixel.mp3 " ,
420
+ mime_type = "audio/mp3 " ,
421
421
)
422
- response = vision_model .generate_content (
422
+ response = pro_model .generate_content (
423
423
contents = ["What is in the audio?" , audio_part ],
424
424
generation_config = generative_models .GenerationConfig (audio_timestamp = True ),
425
425
)
426
426
assert response .text
427
- assert api_transport in get_client_api_transport (
428
- vision_model ._prediction_client
429
- )
427
+ assert api_transport in get_client_api_transport (pro_model ._prediction_client )
430
428
431
429
def test_grounding_google_search_retriever (self , api_endpoint_env_name ):
432
430
model = preview_generative_models .GenerativeModel (GEMINI_MODEL_NAME )
You can’t perform that action at this time.
0 commit comments