Fix ruff linting errors

- Fix type annotations to use X | Y syntax instead of Optional[X]
- Fix f-string usage in exception messages
- Fix datetime.min timezone issue by using datetime.UTC

Author: codegen-sh[bot]

src/codegen/cli/commands/init/main.py

@@ -1,5 +1,4 @@
 from pathlib import Path
-from typing import Optional
 
 import rich
 import typer
@@ -10,9 +9,9 @@
 
 
 def init(
-    path: Optional[str] = typer.Option(None, help="Path within a git repository. Defaults to the current directory."),
-    token: Optional[str] = typer.Option(None, help="Access token for the git repository. Required for full functionality."),
-    language: Optional[str] = typer.Option(None, help="Override automatic language detection (python or typescript)"),
+    path: str | None = typer.Option(None, help="Path within a git repository. Defaults to the current directory."),
+    token: str | None = typer.Option(None, help="Access token for the git repository. Required for full functionality."),
+    language: str | None = typer.Option(None, help="Override automatic language detection (python or typescript)"),
     fetch_docs: bool = typer.Option(False, "--fetch-docs", help="Fetch docs and examples (requires auth)"),
 ):
     """Initialize or update the Codegen folder."""

src/codegen/cli/commands/login/main.py

@@ -1,4 +1,3 @@
-from typing import Optional
 
 import rich
 import typer
@@ -7,7 +6,7 @@
 from codegen.cli.auth.token_manager import get_current_token
 
 
-def login(token: Optional[str] = typer.Option(None, help="API token for authentication")):
+def login(token: str | None = typer.Option(None, help="API token for authentication")):
     """Store authentication token."""
     # Check if already authenticated
     if get_current_token():

src/codegen/cli/commands/mcp/main.py

@@ -1,6 +1,5 @@
 """MCP server command for the Codegen CLI."""
 
-from typing import Optional
 
 import typer
 from rich.console import Console
@@ -10,7 +9,7 @@
 
 def mcp(
     host: str = typer.Option("localhost", help="Host to bind the MCP server to"),
-    port: Optional[int] = typer.Option(None, help="Port to bind the MCP server to (default: stdio transport)"),
+    port: int | None = typer.Option(None, help="Port to bind the MCP server to (default: stdio transport)"),
     transport: str = typer.Option("stdio", help="Transport protocol to use (stdio or http)"),
 ):
     """Start the Codegen MCP server."""

src/codegen/cli/commands/update/main.py

@@ -1,7 +1,6 @@
 import subprocess
 import sys
 from importlib.metadata import distribution
-from typing import Optional
 
 import requests
 import rich
@@ -33,7 +32,7 @@ def install_package(package: str, *args: str) -> None:
 
 def update(
     list_: bool = typer.Option(False, "--list", "-l", help="List all supported versions of the codegen"),
-    version: Optional[str] = typer.Option(None, "--version", "-v", help="Update to a specific version of the codegen"),
+    version: str | None = typer.Option(None, "--version", "-v", help="Update to a specific version of the codegen"),
 ):
     """Update Codegen to the latest or specified version
 

src/codegen/cli/mcp/server.py

@@ -1,6 +1,6 @@
 import json
 import os
-from typing import Annotated, Any, Optional
+from typing import Annotated, Any
 
 from fastmcp import Context, FastMCP
 
@@ -79,9 +79,9 @@ def get_service_config() -> dict[str, Any]:
 def create_agent_run(
     org_id: Annotated[int, "Organization ID"],
     prompt: Annotated[str, "The prompt/task for the agent to execute"],
-    repo_name: Annotated[Optional[str], "Repository name (optional)"] = None,
-    branch_name: Annotated[Optional[str], "Branch name (optional)"] = None,
-    ctx: Optional[Context] = None,
+    repo_name: Annotated[str | None, "Repository name (optional)"] = None,
+    branch_name: Annotated[str | None, "Branch name (optional)"] = None,
+    ctx: Context | None = None,
 ) -> str:
     """Create a new agent run in the specified organization."""
     try:
@@ -112,7 +112,7 @@ def create_agent_run(
 def get_agent_run(
     org_id: Annotated[int, "Organization ID"],
     agent_run_id: Annotated[int, "Agent run ID"],
-    ctx: Optional[Context] = None,
+    ctx: Context | None = None,
 ) -> str:
     """Get details of a specific agent run."""
     try:
@@ -142,7 +142,7 @@ def get_agent_run(
 def get_organizations(
     page: Annotated[int, "Page number (default: 1)"] = 1,
     limit: Annotated[int, "Number of organizations per page (default: 10)"] = 10,
-    ctx: Optional[Context] = None,
+    ctx: Context | None = None,
 ) -> str:
     """Get list of organizations the user has access to."""
     try:
@@ -166,7 +166,7 @@ def get_users(
     org_id: Annotated[int, "Organization ID"],
     page: Annotated[int, "Page number (default: 1)"] = 1,
     limit: Annotated[int, "Number of users per page (default: 10)"] = 10,
-    ctx: Optional[Context] = None,
+    ctx: Context | None = None,
 ) -> str:
     """Get list of users in an organization."""
     try:
@@ -189,7 +189,7 @@ def get_users(
 def get_user(
     org_id: Annotated[int, "Organization ID"],
     user_id: Annotated[int, "User ID"],
-    ctx: Optional[Context] = None,
+    ctx: Context | None = None,
 ) -> str:
     """Get details of a specific user in an organization."""
     try:

src/codegen/git/clients/git_repo_client.py

@@ -1,5 +1,5 @@
 import time
-from datetime import datetime
+from datetime import UTC, datetime
 
 from github.Branch import Branch
 from github.CheckRun import CheckRun
@@ -98,7 +98,7 @@ def get_last_modified_date_of_path(self, path: str) -> datetime:
             return last_modified_date
         else:
             print("Directory has not been modified or does not exist.")
-            return datetime.min
+            return datetime.min.replace(tzinfo=UTC)
 
     ####################################################################################################################
     # COMMENTS

src/codegen/shared/performance/stopwatch_utils.py

@@ -49,4 +49,4 @@ def subprocess_with_stopwatch(command, command_desc: str | None = None, *args, *
     end_time = time.time()
     logger.info(f"Command '{command_desc or command}' took {end_time - start_time} seconds to execute.")
     # Cast to the correct type since we set text=True
-    return cast(subprocess.CompletedProcess[str], result)
+    return cast("subprocess.CompletedProcess[str]", result)