opik 1.8.39__py3-none-any.whl → 1.9.71__py3-none-any.whl

opik/rest_api/types/webhook_examples.py

@@ -0,0 +1,19 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class WebhookExamples(UniversalBaseModel):
+    response_examples: typing.Optional[typing.Dict[str, typing.Dict[str, typing.Optional[typing.Any]]]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/webhook_public.py

@@ -0,0 +1,28 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class WebhookPublic(UniversalBaseModel):
+    id: typing.Optional[str] = None
+    name: typing.Optional[str] = None
+    url: str
+    secret_token: typing.Optional[str] = None
+    headers: typing.Optional[typing.Dict[str, str]] = None
+    created_at: typing.Optional[dt.datetime] = None
+    created_by: typing.Optional[str] = None
+    last_updated_at: typing.Optional[dt.datetime] = None
+    last_updated_by: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/webhook_test_result.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .webhook_test_result_status import WebhookTestResultStatus
+
+
+class WebhookTestResult(UniversalBaseModel):
+    status: typing.Optional[WebhookTestResultStatus] = None
+    status_code: typing.Optional[int] = None
+    request_body: typing.Optional[str] = None
+    error_message: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/webhook_test_result_status.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+WebhookTestResultStatus = typing.Union[typing.Literal["success", "failure"], typing.Any]

opik/rest_api/types/webhook_write.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class WebhookWrite(UniversalBaseModel):
+    id: typing.Optional[str] = None
+    name: typing.Optional[str] = None
+    url: str
+    secret_token: typing.Optional[str] = None
+    headers: typing.Optional[typing.Dict[str, str]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/welcome_wizard_tracking.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class WelcomeWizardTracking(UniversalBaseModel):
+    completed: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    Whether the welcome wizard has been completed
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/workspace_configuration.py

@@ -12,6 +12,11 @@ class WorkspaceConfiguration(UniversalBaseModel):
     Duration in ISO-8601 format (e.g., PT30M for 30 minutes, PT2H for 2 hours, P1D for 1 day). Minimum precision supported is seconds, please use a duration with seconds precision or higher. Also, the max duration allowed is 7 days.
     """
 
+    truncation_on_tables: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    Enable or disable data truncation in table views. When disabled, the frontend will limit pagination to prevent performance issues. Default: true (truncation enabled).
+    """
+
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:

opik/rest_api/welcome_wizard/__init__.py

@@ -0,0 +1,4 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+

opik/rest_api/welcome_wizard/client.py

@@ -0,0 +1,195 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
+from ..types.welcome_wizard_tracking import WelcomeWizardTracking
+from .raw_client import AsyncRawWelcomeWizardClient, RawWelcomeWizardClient
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class WelcomeWizardClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawWelcomeWizardClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawWelcomeWizardClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawWelcomeWizardClient
+        """
+        return self._raw_client
+
+    def get_welcome_wizard_status(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> WelcomeWizardTracking:
+        """
+        Get welcome wizard tracking status for the current workspace
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        WelcomeWizardTracking
+            Welcome wizard tracking status
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.welcome_wizard.get_welcome_wizard_status()
+        """
+        _response = self._raw_client.get_welcome_wizard_status(request_options=request_options)
+        return _response.data
+
+    def submit_welcome_wizard(
+        self,
+        *,
+        role: typing.Optional[str] = OMIT,
+        integrations: typing.Optional[typing.Sequence[str]] = OMIT,
+        email: typing.Optional[str] = OMIT,
+        join_beta_program: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Submit welcome wizard with user information
+
+        Parameters
+        ----------
+        role : typing.Optional[str]
+            Optional user role
+
+        integrations : typing.Optional[typing.Sequence[str]]
+            List of integrations the user selected
+
+        email : typing.Optional[str]
+            Optional user email
+
+        join_beta_program : typing.Optional[bool]
+            Whether user wants to join beta programs
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.welcome_wizard.submit_welcome_wizard()
+        """
+        _response = self._raw_client.submit_welcome_wizard(
+            role=role,
+            integrations=integrations,
+            email=email,
+            join_beta_program=join_beta_program,
+            request_options=request_options,
+        )
+        return _response.data
+
+
+class AsyncWelcomeWizardClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawWelcomeWizardClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawWelcomeWizardClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawWelcomeWizardClient
+        """
+        return self._raw_client
+
+    async def get_welcome_wizard_status(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> WelcomeWizardTracking:
+        """
+        Get welcome wizard tracking status for the current workspace
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        WelcomeWizardTracking
+            Welcome wizard tracking status
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.welcome_wizard.get_welcome_wizard_status()
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_welcome_wizard_status(request_options=request_options)
+        return _response.data
+
+    async def submit_welcome_wizard(
+        self,
+        *,
+        role: typing.Optional[str] = OMIT,
+        integrations: typing.Optional[typing.Sequence[str]] = OMIT,
+        email: typing.Optional[str] = OMIT,
+        join_beta_program: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Submit welcome wizard with user information
+
+        Parameters
+        ----------
+        role : typing.Optional[str]
+            Optional user role
+
+        integrations : typing.Optional[typing.Sequence[str]]
+            List of integrations the user selected
+
+        email : typing.Optional[str]
+            Optional user email
+
+        join_beta_program : typing.Optional[bool]
+            Whether user wants to join beta programs
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.welcome_wizard.submit_welcome_wizard()
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.submit_welcome_wizard(
+            role=role,
+            integrations=integrations,
+            email=email,
+            join_beta_program=join_beta_program,
+            request_options=request_options,
+        )
+        return _response.data

opik/rest_api/welcome_wizard/raw_client.py

@@ -0,0 +1,208 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.http_response import AsyncHttpResponse, HttpResponse
+from ..core.pydantic_utilities import parse_obj_as
+from ..core.request_options import RequestOptions
+from ..types.welcome_wizard_tracking import WelcomeWizardTracking
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class RawWelcomeWizardClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def get_welcome_wizard_status(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[WelcomeWizardTracking]:
+        """
+        Get welcome wizard tracking status for the current workspace
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[WelcomeWizardTracking]
+            Welcome wizard tracking status
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/private/welcome-wizard",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    WelcomeWizardTracking,
+                    parse_obj_as(
+                        type_=WelcomeWizardTracking,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def submit_welcome_wizard(
+        self,
+        *,
+        role: typing.Optional[str] = OMIT,
+        integrations: typing.Optional[typing.Sequence[str]] = OMIT,
+        email: typing.Optional[str] = OMIT,
+        join_beta_program: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[None]:
+        """
+        Submit welcome wizard with user information
+
+        Parameters
+        ----------
+        role : typing.Optional[str]
+            Optional user role
+
+        integrations : typing.Optional[typing.Sequence[str]]
+            List of integrations the user selected
+
+        email : typing.Optional[str]
+            Optional user email
+
+        join_beta_program : typing.Optional[bool]
+            Whether user wants to join beta programs
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[None]
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/private/welcome-wizard",
+            method="POST",
+            json={
+                "role": role,
+                "integrations": integrations,
+                "email": email,
+                "join_beta_program": join_beta_program,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return HttpResponse(response=_response, data=None)
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+
+class AsyncRawWelcomeWizardClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def get_welcome_wizard_status(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[WelcomeWizardTracking]:
+        """
+        Get welcome wizard tracking status for the current workspace
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[WelcomeWizardTracking]
+            Welcome wizard tracking status
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/private/welcome-wizard",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    WelcomeWizardTracking,
+                    parse_obj_as(
+                        type_=WelcomeWizardTracking,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def submit_welcome_wizard(
+        self,
+        *,
+        role: typing.Optional[str] = OMIT,
+        integrations: typing.Optional[typing.Sequence[str]] = OMIT,
+        email: typing.Optional[str] = OMIT,
+        join_beta_program: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[None]:
+        """
+        Submit welcome wizard with user information
+
+        Parameters
+        ----------
+        role : typing.Optional[str]
+            Optional user role
+
+        integrations : typing.Optional[typing.Sequence[str]]
+            List of integrations the user selected
+
+        email : typing.Optional[str]
+            Optional user email
+
+        join_beta_program : typing.Optional[bool]
+            Whether user wants to join beta programs
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[None]
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/private/welcome-wizard",
+            method="POST",
+            json={
+                "role": role,
+                "integrations": integrations,
+                "email": email,
+                "join_beta_program": join_beta_program,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return AsyncHttpResponse(response=_response, data=None)
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

opik/rest_api/workspaces/client.py

@@ -105,6 +105,7 @@ class WorkspacesClient:
         self,
         *,
         timeout_to_mark_thread_as_inactive: typing.Optional[str] = OMIT,
+        truncation_on_tables: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> WorkspaceConfiguration:
         """
@@ -115,6 +116,9 @@ class WorkspacesClient:
         timeout_to_mark_thread_as_inactive : typing.Optional[str]
             Duration in ISO-8601 format (e.g., PT30M for 30 minutes, PT2H for 2 hours, P1D for 1 day). Minimum precision supported is seconds, please use a duration with seconds precision or higher. Also, the max duration allowed is 7 days.
 
+        truncation_on_tables : typing.Optional[bool]
+            Enable or disable data truncation in table views. When disabled, the frontend will limit pagination to prevent performance issues. Default: true (truncation enabled).
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -130,7 +134,9 @@ class WorkspacesClient:
         client.workspaces.upsert_workspace_configuration()
         """
         _response = self._raw_client.upsert_workspace_configuration(
-            timeout_to_mark_thread_as_inactive=timeout_to_mark_thread_as_inactive, request_options=request_options
+            timeout_to_mark_thread_as_inactive=timeout_to_mark_thread_as_inactive,
+            truncation_on_tables=truncation_on_tables,
+            request_options=request_options,
         )
         return _response.data
 
@@ -391,6 +397,7 @@ class AsyncWorkspacesClient:
         self,
         *,
         timeout_to_mark_thread_as_inactive: typing.Optional[str] = OMIT,
+        truncation_on_tables: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> WorkspaceConfiguration:
         """
@@ -401,6 +408,9 @@ class AsyncWorkspacesClient:
         timeout_to_mark_thread_as_inactive : typing.Optional[str]
             Duration in ISO-8601 format (e.g., PT30M for 30 minutes, PT2H for 2 hours, P1D for 1 day). Minimum precision supported is seconds, please use a duration with seconds precision or higher. Also, the max duration allowed is 7 days.
 
+        truncation_on_tables : typing.Optional[bool]
+            Enable or disable data truncation in table views. When disabled, the frontend will limit pagination to prevent performance issues. Default: true (truncation enabled).
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -419,7 +429,9 @@ class AsyncWorkspacesClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.upsert_workspace_configuration(
-            timeout_to_mark_thread_as_inactive=timeout_to_mark_thread_as_inactive, request_options=request_options
+            timeout_to_mark_thread_as_inactive=timeout_to_mark_thread_as_inactive,
+            truncation_on_tables=truncation_on_tables,
+            request_options=request_options,
         )
         return _response.data
 

opik/rest_api/workspaces/raw_client.py

@@ -147,6 +147,7 @@ class RawWorkspacesClient:
         self,
         *,
         timeout_to_mark_thread_as_inactive: typing.Optional[str] = OMIT,
+        truncation_on_tables: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[WorkspaceConfiguration]:
         """
@@ -157,6 +158,9 @@ class RawWorkspacesClient:
         timeout_to_mark_thread_as_inactive : typing.Optional[str]
             Duration in ISO-8601 format (e.g., PT30M for 30 minutes, PT2H for 2 hours, P1D for 1 day). Minimum precision supported is seconds, please use a duration with seconds precision or higher. Also, the max duration allowed is 7 days.
 
+        truncation_on_tables : typing.Optional[bool]
+            Enable or disable data truncation in table views. When disabled, the frontend will limit pagination to prevent performance issues. Default: true (truncation enabled).
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -170,6 +174,7 @@ class RawWorkspacesClient:
             method="PUT",
             json={
                 "timeout_to_mark_thread_as_inactive": timeout_to_mark_thread_as_inactive,
+                "truncation_on_tables": truncation_on_tables,
             },
             headers={
                 "content-type": "application/json",
@@ -593,6 +598,7 @@ class AsyncRawWorkspacesClient:
         self,
         *,
         timeout_to_mark_thread_as_inactive: typing.Optional[str] = OMIT,
+        truncation_on_tables: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[WorkspaceConfiguration]:
         """
@@ -603,6 +609,9 @@ class AsyncRawWorkspacesClient:
         timeout_to_mark_thread_as_inactive : typing.Optional[str]
             Duration in ISO-8601 format (e.g., PT30M for 30 minutes, PT2H for 2 hours, P1D for 1 day). Minimum precision supported is seconds, please use a duration with seconds precision or higher. Also, the max duration allowed is 7 days.
 
+        truncation_on_tables : typing.Optional[bool]
+            Enable or disable data truncation in table views. When disabled, the frontend will limit pagination to prevent performance issues. Default: true (truncation enabled).
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -616,6 +625,7 @@ class AsyncRawWorkspacesClient:
             method="PUT",
             json={
                 "timeout_to_mark_thread_as_inactive": timeout_to_mark_thread_as_inactive,
+                "truncation_on_tables": truncation_on_tables,
             },
             headers={
                 "content-type": "application/json",

opik/s3_httpx_client.py

@@ -3,6 +3,8 @@ import functools
 import httpx
 import tenacity
 
+from opik import hooks
+
 KEEPALIVE_EXPIRY_SECONDS = 10
 CONNECT_TIMEOUT_SECONDS = 20
 READ_TIMEOUT_SECONDS = 100
@@ -27,7 +29,18 @@ def get() -> httpx.Client:
         pool=POOL_TIMEOUT_SECONDS,
     )
 
-    return httpx.Client(limits=limits, timeout=timeout)
+    # build HTTPX client arguments
+    kwargs = {
+        "limits": limits,
+        "timeout": timeout,
+    }
+    kwargs = hooks.httpx_client_hook.build_init_arguments(kwargs)
+
+    client = httpx.Client(**kwargs)
+
+    hooks.httpx_client_hook.apply_httpx_client_hooks(client)
+
+    return client
 
 
 def _allowed_to_retry(exception: Exception) -> bool: