From d2267a6af6b39d986a0e3e6b56dbef930bf12d73 Mon Sep 17 00:00:00 2001 From: spirpinias Date: Fri, 28 Jun 2024 15:10:27 -0400 Subject: [PATCH 1/2] fix: add TCPKeepAlive to client requests --- src/codeocean/client.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/codeocean/client.py b/src/codeocean/client.py index d9f157e..8a5776c 100644 --- a/src/codeocean/client.py +++ b/src/codeocean/client.py @@ -1,4 +1,5 @@ from requests_toolbelt.sessions import BaseUrlSession +from requests_toolbelt.adapters.socket_options import TCPKeepAliveAdapter from dataclasses import dataclass from codeocean.capsule import Capsules @@ -15,6 +16,7 @@ class CodeOcean: def __post_init__(self): self.session = BaseUrlSession(base_url=f"{self.domain}/api/v1/") self.session.auth = (self.token, "") + self.session.mount("http://", TCPKeepAliveAdapter()) self.session.headers.update({"Content-Type": "application/json"}) self.session.hooks["response"] = [ lambda response, *args, **kwargs: response.raise_for_status() From 25b3e8fb5f0a3003b3ec6b3652437750b100b50f Mon Sep 17 00:00:00 2001 From: Stephen Pirpinias Date: Fri, 28 Jun 2024 15:26:16 -0400 Subject: [PATCH 2/2] Update src/codeocean/client.py Co-authored-by: Zvika Gart --- src/codeocean/client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/codeocean/client.py b/src/codeocean/client.py index 8a5776c..7d291b5 100644 --- a/src/codeocean/client.py +++ b/src/codeocean/client.py @@ -16,7 +16,7 @@ class CodeOcean: def __post_init__(self): self.session = BaseUrlSession(base_url=f"{self.domain}/api/v1/") self.session.auth = (self.token, "") - self.session.mount("http://", TCPKeepAliveAdapter()) + self.session.mount(self.domain, TCPKeepAliveAdapter()) self.session.headers.update({"Content-Type": "application/json"}) self.session.hooks["response"] = [ lambda response, *args, **kwargs: response.raise_for_status()