black format

V2arK · V2arK · commit 8b370228ee1f · 2025-09-02T09:55:39.000-04:00
diff --git a/centml/cli/login.py b/centml/cli/login.py
@@ -26,7 +26,11 @@
 
 def generate_pkce_pair():
     verifier = secrets.token_urlsafe(64)
-    challenge = base64.urlsafe_b64encode(hashlib.sha256(verifier.encode()).digest()).decode().rstrip("=")
+    challenge = (
+        base64.urlsafe_b64encode(hashlib.sha256(verifier.encode()).digest())
+        .decode()
+        .rstrip("=")
+    )
     return verifier, challenge
 
 
@@ -102,14 +106,18 @@ def login(token_file):
     else:
         click.echo("Logging into CentML...")
 
-        choice = click.confirm("Do you want to log in with your browser now?", default=True)
+        choice = click.confirm(
+            "Do you want to log in with your browser now?", default=True
+        )
         if choice:
             try:
                 # PKCE Flow
                 code_verifier, code_challenge = generate_pkce_pair()
                 auth_url = build_auth_url(CLIENT_ID, REDIRECT_URI, code_challenge)
                 click.echo("A browser window will open for you to authenticate.")
-                click.echo("If it doesn't open automatically, you can copy and paste this URL:")
+                click.echo(
+                    "If it doesn't open automatically, you can copy and paste this URL:"
+                )
                 click.echo(f"   {auth_url}\n")
                 webbrowser.open(auth_url)
                 click.echo("Waiting for authentication...")
@@ -121,9 +129,13 @@ def login(token_file):
                     click.echo("Login failed. Please try again.")
                 else:
                     cred = {
-                        key: response_dict[key] for key in ("access_token", "refresh_token") if key in response_dict
+                        key: response_dict[key]
+                        for key in ("access_token", "refresh_token")
+                        if key in response_dict
                     }
-                    os.makedirs(os.path.dirname(settings.CENTML_CRED_FILE_PATH), exist_ok=True)
+                    os.makedirs(
+                        os.path.dirname(settings.CENTML_CRED_FILE_PATH), exist_ok=True
+                    )
                     with open(settings.CENTML_CRED_FILE_PATH, "w") as f:
                         json.dump(cred, f)
                     click.echo("✅ Login successful")
diff --git a/centml/sdk/auth.py b/centml/sdk/auth.py
@@ -29,7 +29,11 @@ def refresh_centml_token(refresh_token):
             os.remove(settings.CENTML_CRED_FILE_PATH)
         cred = None
     else:
-        cred = {key: response_dict[key] for key in ("access_token", "refresh_token") if key in response_dict}
+        cred = {
+            key: response_dict[key]
+            for key in ("access_token", "refresh_token")
+            if key in response_dict
+        }
         with open(settings.CENTML_CRED_FILE_PATH, "w") as f:
             json.dump(cred, f)
 
@@ -61,7 +65,9 @@ def get_centml_token():
     cred = load_centml_cred()
     if not cred:
         sys.exit("CentML credentials not found. Please login...")
-    exp_time = int(jwt.decode(cred["access_token"], options={"verify_signature": False})["exp"])
+    exp_time = int(
+        jwt.decode(cred["access_token"], options={"verify_signature": False})["exp"]
+    )
 
     if time.time() >= exp_time - 100:
         cred = refresh_centml_token(cred["refresh_token"])
diff --git a/centml/sdk/config.py b/centml/sdk/config.py
@@ -8,13 +8,19 @@ class Config(BaseSettings):
     model_config = SettingsConfigDict(env_file=Path(".env"))
 
     CENTML_WEB_URL: str = os.getenv("CENTML_WEB_URL", default="https://app.centml.com/")
-    CENTML_CONFIG_PATH: str = os.getenv("CENTML_CONFIG_PATH", default=os.path.expanduser("~/.centml"))
+    CENTML_CONFIG_PATH: str = os.getenv(
+        "CENTML_CONFIG_PATH", default=os.path.expanduser("~/.centml")
+    )
     CENTML_CRED_FILE: str = os.getenv("CENTML_CRED_FILE", default="credentials.json")
     CENTML_CRED_FILE_PATH: str = os.path.join(CENTML_CONFIG_PATH, CENTML_CRED_FILE)
 
-    CENTML_PLATFORM_API_URL: str = os.getenv("CENTML_PLATFORM_API_URL", default="https://api.centml.com")
+    CENTML_PLATFORM_API_URL: str = os.getenv(
+        "CENTML_PLATFORM_API_URL", default="https://api.centml.com"
+    )
 
-    CENTML_WORKOS_CLIENT_ID: str = os.getenv("CENTML_WORKOS_CLIENT_ID", default="client_01JP5TWW2997MF8AYQXHJEGYR0")
+    CENTML_WORKOS_CLIENT_ID: str = os.getenv(
+        "CENTML_WORKOS_CLIENT_ID", default="client_01JP5TWW2997MF8AYQXHJEGYR0"
+    )
 
 
 settings = Config()
diff --git a/centml/sdk/utils/client_certs.py b/centml/sdk/utils/client_certs.py
@@ -23,7 +23,9 @@ def generate_ca_client_triplet(service_name: str) -> CAClientCertTriplet:
 
     # Details about who we are. For a self-signed certificate, the subject
     # and issuer are always the same.
-    ca_subject = x509.Name([x509.NameAttribute(NameOID.COMMON_NAME, f"ca.{service_name}.user.centml.ai")])
+    ca_subject = x509.Name(
+        [x509.NameAttribute(NameOID.COMMON_NAME, f"ca.{service_name}.user.centml.ai")]
+    )
 
     ca_certificate = (
         x509.CertificateBuilder()
@@ -44,7 +46,13 @@ def generate_ca_client_triplet(service_name: str) -> CAClientCertTriplet:
     client_private_key = ec.generate_private_key(ec.SECP384R1())
 
     # Information about the client
-    client_subject = x509.Name([x509.NameAttribute(NameOID.COMMON_NAME, f"client.{service_name}.user.centml.ai")])
+    client_subject = x509.Name(
+        [
+            x509.NameAttribute(
+                NameOID.COMMON_NAME, f"client.{service_name}.user.centml.ai"
+            )
+        ]
+    )
 
     client_certificate = (
         x509.CertificateBuilder()
@@ -63,8 +71,12 @@ def generate_ca_client_triplet(service_name: str) -> CAClientCertTriplet:
     )
 
     return CAClientCertTriplet(
-        certificate_authority=ca_certificate.public_bytes(serialization.Encoding.PEM).decode("ascii"),
-        client_certificate=client_certificate.public_bytes(serialization.Encoding.PEM).decode("ascii"),
+        certificate_authority=ca_certificate.public_bytes(
+            serialization.Encoding.PEM
+        ).decode("ascii"),
+        client_certificate=client_certificate.public_bytes(
+            serialization.Encoding.PEM
+        ).decode("ascii"),
         client_private_key=client_private_key.private_bytes(
             encoding=serialization.Encoding.PEM,
             format=serialization.PrivateFormat.PKCS8,
@@ -85,9 +97,11 @@ def save_pem_file(service_name, client_private_key, client_certificate):
 
     try:
         # Save the combined PEM file
-        with open(ca_file_path, 'w') as combined_pem_file:
+        with open(ca_file_path, "w") as combined_pem_file:
             combined_pem_file.write(client_private_key + client_certificate)
-        click.echo(f"Combined PEM file for accessing the private endpoint has been saved to {ca_file_path}")
+        click.echo(
+            f"Combined PEM file for accessing the private endpoint has been saved to {ca_file_path}"
+        )
 
     except Exception as e:
         click.echo(f"Error saving PEM files: {e}")
