feat(genai): Update bearer token code

Author: msampathkumar

genai/live/live_websocket_audiogen_with_txt.py

@@ -16,7 +16,7 @@
 import os
 
 
-def get_bearer_token():
+def get_bearer_token() -> str:
     import google.auth
     from google.auth.transport.requests import Request
 

genai/live/live_websocket_audiotranscript_with_txt.py

@@ -16,7 +16,7 @@
 import os
 
 
-def get_bearer_token():
+def get_bearer_token() -> str:
     import google.auth
     from google.auth.transport.requests import Request
 

genai/live/live_websocket_textgen_with_audio.py

@@ -16,7 +16,7 @@
 import os
 
 
-def get_bearer_token():
+def get_bearer_token() -> str:
     import google.auth
     from google.auth.transport.requests import Request
 
@@ -43,8 +43,8 @@ async def generate_content() -> str:
     from websockets.asyncio.client import connect
     from scipy.io import wavfile
 
-    def read_wavefile(filepath):
-        # Read the .wav file.
+    def read_wavefile(filepath: str) -> tuple[str, str]:
+        # Read the .wav file using scipy.io.wavfile.read
         rate, data = wavfile.read(filepath)
         # Convert the NumPy array of audio samples back to raw bytes
         raw_audio_bytes = data.tobytes()

genai/live/live_websocket_textgen_with_txt.py

@@ -16,7 +16,7 @@
 import os
 
 
-def get_bearer_token():
+def get_bearer_token() -> str:
     import google.auth
     from google.auth.transport.requests import Request
 

genai/live/test_live_examples.py

@@ -20,11 +20,11 @@
 
 import pytest
 
-import live_with_txt
-import live_websocket_textgen_with_audio
-import live_websocket_textgen_with_txt
 import live_websocket_audiogen_with_txt
 import live_websocket_audiotranscript_with_txt
+import live_websocket_textgen_with_audio
+import live_websocket_textgen_with_txt
+import live_with_txt
 
 os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
 os.environ["GOOGLE_CLOUD_LOCATION"] = "us-central1"