From b794e1f9b82d11fe5fec6d7f1b37d4eff3c3527c Mon Sep 17 00:00:00 2001 From: Hugo Herter Date: Fri, 10 Mar 2023 13:25:31 +0100 Subject: [PATCH] Feature: Uses could not use custom timeout --- src/aleph/sdk/client.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/aleph/sdk/client.py b/src/aleph/sdk/client.py index 6f42f8b2..bf992200 100644 --- a/src/aleph/sdk/client.py +++ b/src/aleph/sdk/client.py @@ -443,9 +443,11 @@ class AlephClient: api_server: str http_session: aiohttp.ClientSession - def __init__(self, api_server: str): + def __init__( + self, api_server: str, timeout: Optional[aiohttp.ClientTimeout] = None + ): self.api_server = api_server - self.http_session = aiohttp.ClientSession(base_url=api_server) + self.http_session = aiohttp.ClientSession(base_url=api_server, timeout=timeout) def __enter__(self) -> UserSessionSync: return UserSessionSync(async_session=self)