Skip to content

Commit 7f515bc

Browse files
committed
feat(api): implement user agent parameters in Gradient and AsyncGradient classes to fix failing tests
1 parent f8cd02b commit 7f515bc

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/gradient/_client.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -299,6 +299,8 @@ def copy(
299299
set_default_headers: Mapping[str, str] | None = None,
300300
default_query: Mapping[str, object] | None = None,
301301
set_default_query: Mapping[str, object] | None = None,
302+
user_agent: str | None = None,
303+
user_agent_version: str | None = None,
302304
_extra_kwargs: Mapping[str, Any] = {},
303305
) -> Self:
304306
"""
@@ -335,6 +337,8 @@ def copy(
335337
max_retries=max_retries if is_given(max_retries) else self.max_retries,
336338
default_headers=headers,
337339
default_query=params,
340+
user_agent=user_agent or self._user_agent_name,
341+
user_agent_version=user_agent_version or self._user_agent_version,
338342
**_extra_kwargs,
339343
)
340344
client._base_url_overridden = self._base_url_overridden or base_url is not None
@@ -608,6 +612,8 @@ def copy(
608612
set_default_headers: Mapping[str, str] | None = None,
609613
default_query: Mapping[str, object] | None = None,
610614
set_default_query: Mapping[str, object] | None = None,
615+
user_agent: str | None = None,
616+
user_agent_version: str | None = None,
611617
_extra_kwargs: Mapping[str, Any] = {},
612618
) -> Self:
613619
"""
@@ -644,6 +650,8 @@ def copy(
644650
max_retries=max_retries if is_given(max_retries) else self.max_retries,
645651
default_headers=headers,
646652
default_query=params,
653+
user_agent=user_agent or self._user_agent_name,
654+
user_agent_version=user_agent_version or self._user_agent_version,
647655
**_extra_kwargs,
648656
)
649657
client._base_url_overridden = self._base_url_overridden or base_url is not None

0 commit comments

Comments
 (0)