From 8cad3aae18f10e15679ba0c87a589139e442310c Mon Sep 17 00:00:00 2001
From: Asher Foa <asher@asherfoa.com>
Date: Fri, 27 Jun 2025 21:22:34 -0400
Subject: [PATCH] Enable pyupgrade on skyvern/cli (#2806)

---
 .pre-commit-config.yaml | 9 ++++-----
 skyvern/cli/tasks.py    | 5 ++---
 skyvern/cli/workflow.py | 9 ++++-----
 3 files changed, 10 insertions(+), 13 deletions(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index cb6fc8e4..47b50cb3 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -23,7 +23,7 @@ repos:
         always_run: true
   - repo: https://github.com/astral-sh/ruff-pre-commit
     # Ruff version.
-    rev: v0.11.13
+    rev: v0.12.1
     hooks:
       # Run the linter.
       - id: ruff
@@ -57,11 +57,10 @@ repos:
       - id: pyupgrade
         exclude: |
           (?x)(
-            ^skyvern/client/.*|
-            ^skyvern/cli/.*
+          ^skyvern/client/.*
           )
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.16.0
+    rev: v1.16.1
     hooks:
       - id: mypy
         args: [--show-error-codes, --warn-unused-configs, --disallow-untyped-calls, --disallow-untyped-defs, --disallow-incomplete-defs, --check-untyped-defs]
@@ -116,6 +115,6 @@ repos:
     hooks:
       - id: shellcheck
   - repo: https://github.com/google/yamlfmt
-    rev: v0.17.0
+    rev: v0.17.2
     hooks:
       - id: yamlfmt
diff --git a/skyvern/cli/tasks.py b/skyvern/cli/tasks.py
index 214cb134..823eae86 100644
--- a/skyvern/cli/tasks.py
+++ b/skyvern/cli/tasks.py
@@ -4,7 +4,6 @@ from __future__ import annotations
 
 import json
 import os
-from typing import Optional
 
 import typer
 from dotenv import load_dotenv
@@ -21,7 +20,7 @@ tasks_app = typer.Typer(help="Manage Skyvern tasks and operations.")
 @tasks_app.callback()
 def tasks_callback(
     ctx: typer.Context,
-    api_key: Optional[str] = typer.Option(
+    api_key: str | None = typer.Option(
         None,
         "--api-key",
         help="Skyvern API key",
@@ -32,7 +31,7 @@ def tasks_callback(
     ctx.obj = {"api_key": api_key}
 
 
-def _get_client(api_key: Optional[str] = None) -> Skyvern:
+def _get_client(api_key: str | None = None) -> Skyvern:
     """Instantiate a Skyvern SDK client using environment variables."""
     load_dotenv()
     load_dotenv(".env")
diff --git a/skyvern/cli/workflow.py b/skyvern/cli/workflow.py
index 75142622..7e307a1a 100644
--- a/skyvern/cli/workflow.py
+++ b/skyvern/cli/workflow.py
@@ -4,7 +4,6 @@ from __future__ import annotations
 
 import json
 import os
-from typing import Optional
 
 import typer
 from dotenv import load_dotenv
@@ -22,7 +21,7 @@ workflow_app = typer.Typer(help="Manage Skyvern workflows.")
 @workflow_app.callback()
 def workflow_callback(
     ctx: typer.Context,
-    api_key: Optional[str] = typer.Option(
+    api_key: str | None = typer.Option(
         None,
         "--api-key",
         help="Skyvern API key",
@@ -33,7 +32,7 @@ def workflow_callback(
     ctx.obj = {"api_key": api_key}
 
 
-def _get_client(api_key: Optional[str] = None) -> Skyvern:
+def _get_client(api_key: str | None = None) -> Skyvern:
     """Instantiate a Skyvern SDK client using environment variables."""
     load_dotenv()
     load_dotenv(".env")
@@ -46,8 +45,8 @@ def run_workflow(
     ctx: typer.Context,
     workflow_id: str = typer.Argument(..., help="Workflow permanent ID"),
     parameters: str = typer.Option("{}", "--parameters", "-p", help="JSON parameters for the workflow"),
-    title: Optional[str] = typer.Option(None, "--title", help="Title for the workflow run"),
-    max_steps: Optional[int] = typer.Option(None, "--max-steps", help="Override the workflow max steps"),
+    title: str | None = typer.Option(None, "--title", help="Title for the workflow run"),
+    max_steps: int | None = typer.Option(None, "--max-steps", help="Override the workflow max steps"),
 ) -> None:
     """Run a workflow."""
     try: