Replace black/flake8/isort with ruff and bump pre-commit hooks

- Replace black, flake8, isort, pycln with ruff for linting and formatting
- Bump pre-commit hooks: ruff-pre-commit v0.15.1, uv-pre-commit 0.10.2,
  docformatter v1.7.7
- Fix deprecated default_stages: 'commit' -> 'pre-commit'
- Bump deps: ruff>=0.15.0, pre-commit>=4.0.0 (dropping autopep8, black,
  flake8, flake8-pyproject, isort)
- Update CI: astral-sh/setup-uv@v7, remove separate setup-python step
- Apply ruff formatting to source and tests

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: muhammad-ali-e

.github/workflows/main.yml

@@ -17,14 +17,11 @@ jobs:
         with:
           ref: ${{ github.event.release.tag_name }}
 
-      - uses: actions/setup-python@v5
-        with:
-          python-version: "3.12"
-
       - name: Install uv
-        uses: astral-sh/setup-uv@v5
+        uses: astral-sh/setup-uv@v7
         with:
-          version: "0.6.14"
+          python-version: "3.12"
+          enable-cache: true
 
       - name: Build package
         run: uv build

.pre-commit-config.yaml

@@ -6,7 +6,7 @@
 default_language_version:
   python: python3
 default_stages:
-  - commit
+  - pre-commit
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v4.5.0
@@ -48,46 +48,14 @@ repos:
     hooks:
       - id: actionlint-docker
         args: [-ignore, 'label ".+" is unknown']
-  - repo: https://github.com/psf/black
-    rev: 24.2.0
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.15.1
     hooks:
-      - id: black
-        args: [--config=pyproject.toml, -l 88]
-        language: system
-        exclude: |
-          (?x)^(
-              pkgs/unstract-flags/src/unstract/flags/evaluation_.*\.py|
-          )$
-  - repo: https://github.com/pycqa/flake8
-    rev: 7.0.0
-    hooks:
-      - id: flake8
-        args: [--max-line-length=88]
-        exclude: |
-          (?x)^(
-              .*migrations/.*\.py|
-              unstract-core/tests/.*|
-              pkgs/unstract-flags/src/unstract/flags/evaluation_.*\.py|
-          )$
-  - repo: https://github.com/pycqa/isort
-    rev: 5.13.2
-    hooks:
-      - id: isort
-        files: "\\.(py)$"
-        args:
-          [
-            "--profile",
-            "black",
-            "--filter-files",
-            --settings-path=pyproject.toml,
-          ]
-  - repo: https://github.com/hadialqattan/pycln
-    rev: v2.4.0
-    hooks:
-      - id: pycln
-        args: [--config=pyproject.toml]
+      - id: ruff
+        args: [--fix]
+      - id: ruff-format
   - repo: https://github.com/pycqa/docformatter
-    rev: v1.7.5
+    rev: v1.7.7
     hooks:
       - id: docformatter
   # - repo: https://github.com/MarcoGorelli/absolufy-imports
@@ -155,6 +123,6 @@ repos:
       - id: markdownlint-fix
         args: [--disable, MD013]
   - repo: https://github.com/astral-sh/uv-pre-commit
-    rev: 0.6.14
+    rev: 0.10.2
     hooks:
       - id: uv-lock

pyproject.toml

@@ -50,28 +50,18 @@ test = [
     "python-dotenv>=1.0.0",
 ]
 lint = [
-    "autopep8~=2.0.2",
-    "black~=23.3.0",
+    "ruff>=0.15.0",
     "docutils~=0.20.1",
-    "flake8~=6.0.0",
-    "flake8-pyproject~=1.2.2",
-    "isort~=5.12.0",
-    "pre-commit~=3.3.1",
+    "pre-commit>=4.0.0",
     "yamllint>=1.35.1",
     "mypy~=1.10.0",
 ]
 
-[tool.isort]
-line_length = 88
-multi_line_output = 3
-include_trailing_comma = true
-force_grid_wrap = 0
-use_parentheses = true
-ensure_newline_before_comments = true
-profile = "black"
+[tool.ruff]
+line-length = 88
 
-[tool.flake8]
-max-line-length = 88
+[tool.ruff.lint]
+select = ["E", "F", "W", "I"]
 
 [tool.pytest.ini_options]
 env_files = ["tests/.env"]

src/unstract/api_deployments/__init__.py

@@ -1,6 +1,6 @@
 __version__ = "1.1.0"
 
-from .client import APIDeploymentsClient
+from .client import APIDeploymentsClient as APIDeploymentsClient
 
 
 def get_sdk_version():

src/unstract/api_deployments/client.py

@@ -66,10 +66,7 @@ def __call__(self, retry_state: RetryCallState) -> float:
         outcome = retry_state.outcome
         if outcome and not outcome.failed:
             response = outcome.result()
-            if (
-                response is not None
-                and getattr(response, "status_code", None) == 429
-            ):
+            if response is not None and getattr(response, "status_code", None) == 429:
                 retry_after = response.headers.get("Retry-After")
                 if retry_after is not None:
                     try:
@@ -209,8 +206,7 @@ def _before_sleep(retry_state: RetryCallState):
             if outcome.failed:
                 exc = outcome.exception()
                 self.logger.warning(
-                    "%s during request to %s. Retrying in %.1fs "
-                    "(attempt %d/%d).",
+                    "%s during request to %s. Retrying in %.1fs (attempt %d/%d).",
                     type(exc).__name__,
                     url,
                     delay,
@@ -220,8 +216,7 @@ def _before_sleep(retry_state: RetryCallState):
             else:
                 response = outcome.result()
                 self.logger.warning(
-                    "Request to %s returned %d. Retrying in %.1fs "
-                    "(attempt %d/%d).",
+                    "Request to %s returned %d. Retrying in %.1fs (attempt %d/%d).",
                     url,
                     response.status_code,
                     delay,
@@ -263,9 +258,7 @@ def _retry_error_callback(retry_state: RetryCallState):
                 jitter=self.jitter,
             ),
             retry=(
-                retry_if_result(
-                    lambda r: self._is_retryable_status(r.status_code)
-                )
+                retry_if_result(lambda r: self._is_retryable_status(r.status_code))
                 | retry_if_exception_type((ConnectionError, Timeout))
             ),
             before_sleep=_before_sleep,

src/unstract/api_deployments/utils.py

@@ -5,7 +5,7 @@ def redact_key(api_key: str, reveal_length=4) -> str:
 
         Args:
             api_key (str): API key to redact
-            reveal_length (int): Number of characters to reveal from the start of the key
+            reveal_length (int): Number of characters to reveal from the start
 
         Returns:
             str: Redacted API key

tests/test_retry.py

@@ -108,16 +108,16 @@ def test_default_retries_504(self, client):
     def test_default_retries_any_5xx(self, client):
         """Any 5xx status code should be retried by default."""
         for code in (500, 501, 502, 503, 504, 507, 511, 599):
-            assert (
-                client._is_retryable_status(code) is True
-            ), f"Expected {code} retryable"
+            assert client._is_retryable_status(code) is True, (
+                f"Expected {code} retryable"
+            )
 
     def test_default_no_retry_4xx_except_429(self, client):
         """4xx codes (except 429) should not be retried by default."""
         for code in (400, 401, 403, 404, 405, 408, 422):
-            assert (
-                client._is_retryable_status(code) is False
-            ), f"Expected {code} not retryable"
+            assert client._is_retryable_status(code) is False, (
+                f"Expected {code} not retryable"
+            )
 
     def test_default_no_retry_2xx(self, client):
         assert client._is_retryable_status(200) is False
@@ -203,9 +203,7 @@ def test_exception_outcome_uses_exponential(self):
         wait = _WaitRetryAfterOrExponentialJitter(
             initial=2.0, max=60.0, exp_base=2.0, jitter=1.0
         )
-        rs = self._make_retry_state(
-            attempt_number=1, exception=ConnectionError("fail")
-        )
+        rs = self._make_retry_state(attempt_number=1, exception=ConnectionError("fail"))
         delay = wait(rs)
         assert 2.0 <= delay <= 3.0