fix: make token retrieval in _get_auth asynchronous to prevent blocki… (#23)

* fix: make token retrieval in _get_auth asynchronous to prevent blocking the event loop

* Updating to use the proper async type.

---------

Co-authored-by: Gerardo Lecaros <10088504+glecaros@users.noreply.github.com>

Author: nimanikoo

python/rtclient/__init__.py

@@ -8,7 +8,8 @@
 from typing import Literal, Optional
 
 from aiohttp import ClientSession, WSMsgType
-from azure.core.credentials import AzureKeyCredential, TokenCredential
+from azure.core.credentials import AzureKeyCredential
+from azure.core.credentials_async import AsyncTokenCredential
 
 from rtclient.models import (
     AssistantContentPart,
@@ -110,7 +111,7 @@ class RTLowLevelClient:
     def __init__(
         self,
         url: Optional[str] = None,
-        token_credential: Optional[TokenCredential] = None,
+        token_credential: Optional[AsyncTokenCredential] = None,
         key_credential: Optional[AzureKeyCredential] = None,
         model: Optional[str] = None,
         azure_deployment: Optional[str] = None,
@@ -138,10 +139,10 @@ def __init__(
     def _user_agent(self):
         return "ms-rtclient-0.4.3"
 
-    def _get_auth(self):
+    async def _get_auth(self):
         if self._token_credential:
-            scopes = ["https://cognitiveservices.azure.com/.default"]
-            token = self._token_credential.get_token(scopes)
+            scope = "https://cognitiveservices.azure.com/.default"
+            token = await self._token_credential.get_token(scope)
             return {"Authorization": f"Bearer {token.token}"}
         elif self._key_credential:
             return {"api-key": self._key_credential.key}
@@ -151,9 +152,9 @@ def _get_auth(self):
     async def connect(self):
         self.request_id = uuid.uuid4()
         if self._is_azure_openai:
-            auth_headers = self._get_auth()
-            headers = {"x-ms-client-request-id": str(self.request_id), "User-Agent": self._user_agent(), **auth_headers}
-            self.ws = await self._session.ws_connect(
+           auth_headers = await self._get_auth()
+           headers = {"x-ms-client-request-id": str(self.request_id), "User-Agent": self._user_agent(), **auth_headers}
+           self.ws = await self._session.ws_connect(
                 "/openai/realtime",
                 headers=headers,
                 params={"deployment": self._azure_deployment, "api-version": "2024-10-01-preview"},
@@ -376,7 +377,7 @@ class RTClient:
     def __init__(
         self,
         url: Optional[str] = None,
-        token_credential: Optional[TokenCredential] = None,
+        token_credential: Optional[AsyncTokenCredential] = None,
         key_credential: Optional[AzureKeyCredential] = None,
         model: Optional[str] = None,
         azure_deployment: Optional[str] = None,