UncountablePythonSDK 0.0.37__tar.gz → 0.0.38__tar.gz

{uncountablepythonsdk-0.0.37/UncountablePythonSDK.egg-info → uncountablepythonsdk-0.0.38}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.37
+Version: 0.0.38
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{uncountablepythonsdk-0.0.37 → uncountablepythonsdk-0.0.38/UncountablePythonSDK.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.37
+Version: 0.0.38
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{uncountablepythonsdk-0.0.37 → uncountablepythonsdk-0.0.38}/uncountable/core/client.py

@@ -60,6 +60,8 @@ HTTPRequest = HTTPPostRequest | HTTPGetRequest
 @dataclass(kw_only=True)
 class ClientConfig:
     allow_insecure_tls: bool = False
+    extra_headers: dict[str, str] | None = None
+    transform_request: typing.Callable[[requests.Request], requests.Request] | None = None
 
 
 OAUTH_REFRESH_WINDOW_SECONDS = 60 * 5
@@ -144,6 +146,7 @@ class Client(ClientMethods):
     _file_uploader: FileUploader
     _cfg: ClientConfig
     _oauth_bearer_token_cache: OAuthBearerTokenCache | None = None
+    _session: requests.Session
 
     def __init__(
         self,
@@ -156,6 +159,8 @@ class Client(ClientMethods):
         self._base_url = base_url
         self._file_uploader = FileUploader(self._base_url, self._auth_details)
         self._cfg = config or ClientConfig()
+        self._session = requests.Session()
+        self._session.verify = not self._cfg.allow_insecure_tls
 
     def _get_response_json(
         self, response: requests.Response, request_id: str
@@ -176,6 +181,15 @@ class Client(ClientMethods):
         except JSONDecodeError as e:
             raise SDKError("unable to process response", request_id=request_id) from e
 
+    def _send_request(self, request: requests.Request) -> requests.Response:
+        if self._cfg.extra_headers is not None:
+            request.headers = {**request.headers, **self._cfg.extra_headers}
+        if self._cfg.transform_request is not None:
+            request = self._cfg.transform_request(request)
+        prepared_request = request.prepare()
+        response = self._session.send(prepared_request)
+        return response
+
     def do_request(self, *, api_request: APIRequest, return_type: type[DT]) -> DT:
         request_id = str(uuid4())
         http_request = self._build_http_request(
@@ -183,21 +197,15 @@ class Client(ClientMethods):
         )
         match http_request:
             case HTTPGetRequest():
-                response = requests.get(
-                    http_request.url,
-                    headers=http_request.headers,
-                    params=http_request.query_params,
-                    verify=not self._cfg.allow_insecure_tls,
-                )
+                request = requests.Request("GET", http_request.url)
+                request.params = http_request.query_params
             case HTTPPostRequest():
-                response = requests.post(
-                    http_request.url,
-                    headers=http_request.headers,
-                    data=http_request.body,
-                    verify=not self._cfg.allow_insecure_tls,
-                )
+                request = requests.Request("POST", http_request.url)
+                request.data = http_request.body
             case _:
                 typing.assert_never(http_request)
+        request.headers = http_request.headers
+        response = self._send_request(request)
         response_data = self._get_response_json(response, request_id=request_id)
         cached_parser = self._get_cached_parser(return_type)
         try:
@@ -220,15 +228,13 @@ class Client(ClientMethods):
             < OAUTH_REFRESH_WINDOW_SECONDS
         ):
             refresh_url = urljoin(self._base_url, "/token/get_bearer_token")
-            response = requests.post(
-                refresh_url,
-                data={
-                    "client_secret": oauth_details.refresh_token,
-                    "scope": oauth_details.scope,
-                    "grant_type": "client_credentials",
-                },
-                verify=not self._cfg.allow_insecure_tls,
-            )
+            request = requests.Request("POST", refresh_url)
+            request.data = {
+                "client_secret": oauth_details.refresh_token,
+                "scope": oauth_details.scope,
+                "grant_type": "client_credentials",
+            }
+            response = self._send_request(request)
             data = self._get_response_json(response, request_id=str(uuid4()))
             token_data = oauth_bearer_token_data_parser.parse_storage(data)
             self._oauth_bearer_token_cache = OAuthBearerTokenCache(