Skip to content

Commit b74c798

Browse files
stainless-app[bot]stainless-bot
authored andcommitted
chore(internal): minor options / compat functions updates (#431)
1 parent 9186c5f commit b74c798

File tree

2 files changed

+9
-9
lines changed

2 files changed

+9
-9
lines changed

src/finch/_base_client.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -880,9 +880,9 @@ def __exit__(
880880
def _prepare_options(
881881
self,
882882
options: FinalRequestOptions, # noqa: ARG002
883-
) -> None:
883+
) -> FinalRequestOptions:
884884
"""Hook for mutating the given options"""
885-
return None
885+
return options
886886

887887
def _prepare_request(
888888
self,
@@ -962,7 +962,7 @@ def _request(
962962
input_options = model_copy(options)
963963

964964
cast_to = self._maybe_override_cast_to(cast_to, options)
965-
self._prepare_options(options)
965+
options = self._prepare_options(options)
966966

967967
retries = self._remaining_retries(remaining_retries, options)
968968
request = self._build_request(options)
@@ -1456,9 +1456,9 @@ async def __aexit__(
14561456
async def _prepare_options(
14571457
self,
14581458
options: FinalRequestOptions, # noqa: ARG002
1459-
) -> None:
1459+
) -> FinalRequestOptions:
14601460
"""Hook for mutating the given options"""
1461-
return None
1461+
return options
14621462

14631463
async def _prepare_request(
14641464
self,
@@ -1543,7 +1543,7 @@ async def _request(
15431543
input_options = model_copy(options)
15441544

15451545
cast_to = self._maybe_override_cast_to(cast_to, options)
1546-
await self._prepare_options(options)
1546+
options = await self._prepare_options(options)
15471547

15481548
retries = self._remaining_retries(remaining_retries, options)
15491549
request = self._build_request(options)

src/finch/_compat.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -118,10 +118,10 @@ def get_model_fields(model: type[pydantic.BaseModel]) -> dict[str, FieldInfo]:
118118
return model.__fields__ # type: ignore
119119

120120

121-
def model_copy(model: _ModelT) -> _ModelT:
121+
def model_copy(model: _ModelT, *, deep: bool = False) -> _ModelT:
122122
if PYDANTIC_V2:
123-
return model.model_copy()
124-
return model.copy() # type: ignore
123+
return model.model_copy(deep=deep)
124+
return model.copy(deep=deep) # type: ignore
125125

126126

127127
def model_json(model: pydantic.BaseModel, *, indent: int | None = None) -> str:

0 commit comments

Comments
 (0)