Skip to content

Commit d735816

Browse files
committed
chore(azure): fix lint issues
1 parent 47d2fab commit d735816

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

src/openai/lib/azure.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from .._types import NOT_GIVEN, Omit, Timeout, NotGiven
1111
from .._utils import is_given, is_mapping
1212
from .._client import OpenAI, AsyncOpenAI
13+
from .._compat import model_copy
1314
from .._models import FinalRequestOptions
1415
from .._streaming import Stream, AsyncStream
1516
from .._exceptions import OpenAIError
@@ -281,8 +282,10 @@ def _get_azure_ad_token(self) -> str | None:
281282
return None
282283

283284
@override
284-
def _prepare_options(self, options: FinalRequestOptions) -> None:
285+
def _prepare_options(self, options: FinalRequestOptions) -> FinalRequestOptions:
285286
headers: dict[str, str | Omit] = {**options.headers} if is_given(options.headers) else {}
287+
288+
options = model_copy(options)
286289
options.headers = headers
287290

288291
azure_ad_token = self._get_azure_ad_token()
@@ -296,7 +299,7 @@ def _prepare_options(self, options: FinalRequestOptions) -> None:
296299
# should never be hit
297300
raise ValueError("Unable to handle auth")
298301

299-
return super()._prepare_options(options)
302+
return options
300303

301304

302305
class AsyncAzureOpenAI(BaseAzureClient[httpx.AsyncClient, AsyncStream[Any]], AsyncOpenAI):
@@ -524,8 +527,10 @@ async def _get_azure_ad_token(self) -> str | None:
524527
return None
525528

526529
@override
527-
async def _prepare_options(self, options: FinalRequestOptions) -> None:
530+
async def _prepare_options(self, options: FinalRequestOptions) -> FinalRequestOptions:
528531
headers: dict[str, str | Omit] = {**options.headers} if is_given(options.headers) else {}
532+
533+
options = model_copy(options)
529534
options.headers = headers
530535

531536
azure_ad_token = await self._get_azure_ad_token()
@@ -539,4 +544,4 @@ async def _prepare_options(self, options: FinalRequestOptions) -> None:
539544
# should never be hit
540545
raise ValueError("Unable to handle auth")
541546

542-
return await super()._prepare_options(options)
547+
return options

0 commit comments

Comments
 (0)