label-studio-sdk 1.0.14__py3-none-any.whl → 1.0.15__py3-none-any.whl

label_studio_sdk/tokens/client_ext.py

@@ -1,7 +1,7 @@
 import threading
 import typing
 from datetime import datetime, timezone
-import inspect
+import ssl
 
 import httpx
 import jwt
@@ -78,34 +78,63 @@ class TokensClientExt:
         
         return self._access_token
 
+    def _get_client_params(self, existing_client: httpx.AsyncClient) -> dict:
+        """Extract parameters from an existing client to create a new one.
+
+        Args:
+            existing_client: The existing client to extract parameters from.
+
+        Returns:
+            dict: Parameters for creating a new client.
+        """
+        return {
+            'auth': existing_client.auth,
+            'params': existing_client.params,
+            'headers': existing_client.headers,
+            'cookies': existing_client.cookies,
+            'timeout': existing_client.timeout,
+            'follow_redirects': existing_client.follow_redirects,
+            'max_redirects': existing_client.max_redirects,
+            'event_hooks': existing_client.event_hooks,
+            'base_url': existing_client.base_url,
+            'trust_env': existing_client.trust_env,
+            'default_encoding': existing_client._default_encoding,
+            'verify': existing_client._transport._pool._ssl_context.verify_mode != ssl.CERT_NONE,
+            'http1': existing_client._transport._pool._http1,
+            'http2': existing_client._transport._pool._http2,
+            'limits': httpx.Limits(
+                max_connections=existing_client._transport._pool._max_connections,
+                max_keepalive_connections=existing_client._transport._pool._max_keepalive_connections,
+                keepalive_expiry=existing_client._transport._pool._keepalive_expiry
+            )
+        }
+
     def refresh(self) -> AccessTokenResponse:
         """Refresh the access token and return the token response."""
-        # We don't do this often, just use a separate sync httpx client for simplicity here
-        # (avoids complicated state management and sync vs async handling)
-        # Create a new client with the same parameters as the existing one
         existing_client = self._client_wrapper.httpx_client.httpx_client
 
-        # Get all parameters from httpx.Client.__init__
-        client_params = {}
-        sig = inspect.signature(httpx.Client.__init__)
-        for param_name in sig.parameters:
-            if param_name != 'self':  # Skip 'self' parameter
-                try:
-                    value = getattr(existing_client, param_name, None)
-                    if value is not None:
-                        client_params[param_name] = value
-                except AttributeError:
-                    continue
-
-        with httpx.Client(**client_params) as sync_client:
-            response = sync_client.request(
+        # For sync client, use it directly
+        if isinstance(existing_client, httpx.Client):
+            print(f"\nverify:{existing_client._transport._pool._ssl_context.verify_mode != ssl.CERT_NONE}")
+            response = existing_client.request(
                 method="POST",
                 url=f"{self._base_url}/api/token/refresh/",
                 json={"refresh": self._api_key},
                 headers={"Content-Type": "application/json"},
             )
+        else:
+            # If an async client was used, get all parameters from the client to init a new sync client
+            client_params = self._get_client_params(existing_client)
+
+            with httpx.Client(**client_params) as sync_client:
+                response = sync_client.request(
+                    method="POST",
+                    url=f"{self._base_url}/api/token/refresh/",
+                    json={"refresh": self._api_key},
+                    headers={"Content-Type": "application/json"},
+                )
             
-            if response.status_code == 200:
-                return AccessTokenResponse.parse_obj(response.json())
-            else:
-                raise ApiError(status_code=response.status_code, body=response.json())
+        if response.status_code == 200:
+            return AccessTokenResponse.parse_obj(response.json())
+        else:
+            raise ApiError(status_code=response.status_code, body=response.json())

{label_studio_sdk-1.0.14.dist-info → label_studio_sdk-1.0.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: label-studio-sdk
-Version: 1.0.14
+Version: 1.0.15
 Summary: 
 Requires-Python: >=3.9,<4
 Classifier: Intended Audience :: Developers

{label_studio_sdk-1.0.14.dist-info → label_studio_sdk-1.0.15.dist-info}/RECORD

@@ -215,7 +215,7 @@ label_studio_sdk/tasks/types/tasks_list_request_fields.py,sha256=5YXxQgyzoaL0QjS
 label_studio_sdk/tasks/types/tasks_list_response.py,sha256=j1pNluAWQOQ8-d9YXQyRQAefnrl8uLQEB7_L55Z8DME,1136
 label_studio_sdk/tokens/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 label_studio_sdk/tokens/client.py,sha256=SvBcKXIsrTihMJC72Ifxv0U1N3gtLGz3JxqdXYA_hD4,19101
-label_studio_sdk/tokens/client_ext.py,sha256=LHy29mBizrxVs_xOVMiIJwNsQKcOjS8V8LWfWMdAdYU,4730
+label_studio_sdk/tokens/client_ext.py,sha256=Rma4ROcJPQOcBBGsA8Iteqk7WrbW7DTU381Ay5pJMiY,6134
 label_studio_sdk/types/__init__.py,sha256=fQykjzHpX04ftslk5I_hWSJQ_H9Kd8XJAmSv18EVOIc,8905
 label_studio_sdk/types/access_token_response.py,sha256=RV9FqkIiFR_9kmKueB-KiqjVyneiqUkMVueAlk5fUyc,624
 label_studio_sdk/types/annotation.py,sha256=AnHm2VjMasWZsaNXVSUzLYbpYrmM4NPZgWQh7WGa6ZQ,3157
@@ -364,7 +364,7 @@ label_studio_sdk/workspaces/members/client.py,sha256=IVM52Yq_9zMQ3TUHT0AkZ5BTQ9a
 label_studio_sdk/workspaces/members/types/__init__.py,sha256=ZIa_rd7d6K9ZITjTU6fptyGgvjNDySksJ7Rbn4wyhD4,252
 label_studio_sdk/workspaces/members/types/members_create_response.py,sha256=7Hp5FSWm4xR5ZOEmEIglq5HYtM9KWZZBDp87jw7jYFg,668
 label_studio_sdk/workspaces/members/types/members_list_response_item.py,sha256=DIc5DJoVahI9olBis_iFgOJrAf05m2fCE8g4R5ZeDko,712
-label_studio_sdk-1.0.14.dist-info/LICENSE,sha256=ymVrFcHiJGjHeY30NWZgdV-xzNEtfuC63oK9ZeMDjhs,11341
-label_studio_sdk-1.0.14.dist-info/METADATA,sha256=v1D-HsVQxvhu1dESF077Nr-mtaIylYb1KQx6pOxoqww,6033
-label_studio_sdk-1.0.14.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-label_studio_sdk-1.0.14.dist-info/RECORD,,
+label_studio_sdk-1.0.15.dist-info/LICENSE,sha256=ymVrFcHiJGjHeY30NWZgdV-xzNEtfuC63oK9ZeMDjhs,11341
+label_studio_sdk-1.0.15.dist-info/METADATA,sha256=Ct3zUaMfhUAHG0IuTDHQ3jCjtD0HcPVxN0Ocj-JPajI,6033
+label_studio_sdk-1.0.15.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+label_studio_sdk-1.0.15.dist-info/RECORD,,