lint

Author: kristapratico

src/openai/_utils/__init__.py

@@ -25,13 +25,15 @@
     coerce_integer as coerce_integer,
     file_from_path as file_from_path,
     parse_datetime as parse_datetime,
+    is_azure_client as is_azure_client,
     strip_not_given as strip_not_given,
     deepcopy_minimal as deepcopy_minimal,
     get_async_library as get_async_library,
     maybe_coerce_float as maybe_coerce_float,
     get_required_header as get_required_header,
     maybe_coerce_boolean as maybe_coerce_boolean,
     maybe_coerce_integer as maybe_coerce_integer,
+    is_async_azure_client as is_async_azure_client,
 )
 from ._typing import (
     is_list_type as is_list_type,

src/openai/_utils/_utils.py

@@ -5,6 +5,7 @@
 import inspect
 import functools
 from typing import (
+    TYPE_CHECKING,
     Any,
     Tuple,
     Mapping,
@@ -30,6 +31,9 @@
 _SequenceT = TypeVar("_SequenceT", bound=Sequence[object])
 CallableT = TypeVar("CallableT", bound=Callable[..., Any])
 
+if TYPE_CHECKING:
+    from ..lib.azure import AzureOpenAI, AsyncAzureOpenAI
+
 
 def flatten(t: Iterable[Iterable[_T]]) -> list[_T]:
     return [item for sublist in t for item in sublist]
@@ -412,3 +416,11 @@ def json_safe(data: object) -> object:
         return data.isoformat()
 
     return data
+
+
+def is_azure_client(client: object) -> TypeGuard[AzureOpenAI]:
+    return hasattr(client, "_azure_ad_token_provider")
+
+
+def is_async_azure_client(client: object) -> TypeGuard[AsyncAzureOpenAI]:
+    return hasattr(client, "_azure_ad_token_provider")

src/openai/resources/beta/realtime/realtime.py

@@ -21,9 +21,11 @@
 )
 from ...._types import NOT_GIVEN, Query, Headers, NotGiven
 from ...._utils import (
+    is_azure_client,
     maybe_transform,
     strip_not_given,
     async_maybe_transform,
+    is_async_azure_client,
 )
 from ...._compat import cached_property
 from ...._models import construct_type_unchecked
@@ -321,11 +323,11 @@ async def __aenter__(self) -> AsyncRealtimeConnection:
 
         auth_headers = self.__client.auth_headers
         extra_query = self.__extra_query
-        if self.__client.__class__.__name__ == "AsyncAzureOpenAI":
+        if is_async_azure_client(self.__client):
             extra_query = {
                 **self.__extra_query,
                 "api-version": self.__client._api_version,
-                "deployment": self.__client._azure_deployment or self.__model
+                "deployment": self.__client._azure_deployment or self.__model,
             }
             if self.__client.api_key != "<missing API key>":
                 auth_headers = {"api-key": self.__client.api_key}
@@ -513,11 +515,11 @@ def __enter__(self) -> RealtimeConnection:
 
         auth_headers = self.__client.auth_headers
         extra_query = self.__extra_query
-        if self.__client.__class__.__name__ == "AzureOpenAI":
+        if is_azure_client(self.__client):
             extra_query = {
                 **self.__extra_query,
                 "api-version": self.__client._api_version,
-                "deployment": self.__client._azure_deployment or self.__model
+                "deployment": self.__client._azure_deployment or self.__model,
             }
             if self.__client.api_key != "<missing API key>":
                 auth_headers = {"api-key": self.__client.api_key}