lint: update pre-commit and refomat (#11)

# Motivation
-  Testing github CI

Author: eacodegen

.pre-commit-config.yaml

@@ -7,7 +7,7 @@ repos:
     hooks:
       - id: taplo-format
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.9.6
+    rev: v0.11.7
     hooks:
       # Run the linter.
       - id: ruff
@@ -19,7 +19,7 @@ repos:
         types_or: [python, jupyter, pyi]
 
   - repo: https://github.com/biomejs/pre-commit
-    rev: "v0.6.1"
+    rev: "v2.0.0-beta.1"
     hooks:
       - id: biome-check
         language: node
@@ -48,7 +48,7 @@ repos:
       - id: check-merge-conflict
 
   - repo: https://github.com/python-jsonschema/check-jsonschema
-    rev: 0.31.1
+    rev: 0.33.0
     hooks:
       - id: check-github-workflows
       - id: check-github-actions
@@ -77,12 +77,12 @@ repos:
         entry: bash -c "uv run --frozen --all-extras --dev deptry src --ignore DEP001 --extend-exclude 'codegen-examples/.*'"
 
   - repo: https://github.com/renovatebot/pre-commit-hooks
-    rev: 39.169.3
+    rev: 39.263.1
     hooks:
       - id: renovate-config-validator
 
   - repo: https://github.com/astral-sh/uv-pre-commit
-    rev: "0.5.31"
+    rev: "0.6.17"
     hooks:
       - id: uv-sync
         args: ["--frozen", "--all-packages", "--all-extras"]

src/codegen/agents/agent.py

@@ -1,5 +1,5 @@
 import os
-from typing import Any, Optional
+from typing import Any
 
 from codegen_api_client.api.agents_api import AgentsApi
 from codegen_api_client.api_client import ApiClient
@@ -45,7 +45,7 @@ def refresh(self) -> None:
 class Agent:
     """API client for interacting with Codegen AI agents."""
 
-    def __init__(self, token: str, org_id: Optional[int] = None, base_url: Optional[str] = CODEGEN_BASE_API_URL):
+    def __init__(self, token: str, org_id: int | None = None, base_url: str | None = CODEGEN_BASE_API_URL):
         """Initialize a new Agent client.
 
         Args:
@@ -82,7 +82,7 @@ def run(self, prompt: str) -> AgentTask:
         self.current_job = job
         return job
 
-    def get_status(self) -> Optional[dict[str, Any]]:
+    def get_status(self) -> dict[str, Any] | None:
         """Get the status of the current job.
 
         Returns:

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

@@ -4,7 +4,6 @@
 import subprocess
 import sys
 from pathlib import Path
-from typing import Optional
 
 import rich
 import rich_click as click
@@ -89,7 +88,7 @@ def create_app_module(file_path: Path) -> str:
     return f"{module_name}:app"
 
 
-def start_ngrok(port: int) -> Optional[str]:
+def start_ngrok(port: int) -> str | None:
     """Start ngrok and return the public URL"""
     try:
         import requests

src/codegen/extensions/attribution/git_history.py

@@ -1,7 +1,6 @@
 import time
 from collections import defaultdict, deque
 from datetime import datetime
-from typing import Optional
 
 import pygit2
 from intervaltree import IntervalTree
@@ -16,7 +15,7 @@
 class GitAttributionTracker:
     """Tracks attribution information for code symbols based on git history."""
 
-    def __init__(self, codebase: Codebase, ai_authors: Optional[list[str]] = None):
+    def __init__(self, codebase: Codebase, ai_authors: list[str] | None = None):
         """Initialize the attribution tracker.
 
         Args:
@@ -43,7 +42,7 @@ def __init__(self, codebase: Codebase, ai_authors: Optional[list[str]] = None):
 
         self._commits: deque[Commit]
 
-    def build_history(self, max_commits: Optional[int] = None) -> None:
+    def build_history(self, max_commits: int | None = None) -> None:
         """Build the git history for the codebase.
 
         Args:
@@ -206,7 +205,7 @@ def map_symbols_to_history(self, force=False) -> None:
         start_time = time.time()
 
         print("Stashing any working directory changes...")
-        stash_msg = f"Codegen Attribution Stash @ {datetime.now().timestamp()}"
+        stash_msg = f"Codegen Attribution Stash @ {datetime.now().timestamp()}"  # noqa: DTZ005
         stash_id = None
         try:
             stash_id = self.repo.stash(self.repo.default_signature, stash_msg, include_untracked=True)
@@ -325,7 +324,7 @@ def get_symbol_history(self, symbol: Symbol) -> list[dict]:
         symbol_id = f"{symbol.filepath}:{symbol.name}"
         return self._symbol_history.get(symbol_id, [])
 
-    def get_symbol_last_editor(self, symbol: Symbol) -> Optional[str]:
+    def get_symbol_last_editor(self, symbol: Symbol) -> str | None:
         """Get the last person who edited a symbol.
 
         Args:
@@ -423,12 +422,12 @@ def get_ai_contribution_timeline(self) -> list[tuple[datetime, int]]:
             if any(name in author for name in self.ai_authors):
                 for commit in commits:
                     # Convert timestamp to year-month
-                    dt = datetime.fromtimestamp(commit["timestamp"])
+                    dt = datetime.fromtimestamp(commit["timestamp"])  # noqa: DTZ006
                     month_key = f"{dt.year}-{dt.month:02d}"
                     monthly_counts[month_key] += 1
 
         # Sort by date
         timeline = sorted(monthly_counts.items())
 
         # Convert to datetime objects
-        return [(datetime.strptime(month, "%Y-%m"), count) for month, count in timeline]
+        return [(datetime.strptime(month, "%Y-%m"), count) for month, count in timeline]  # noqa: DTZ007

src/codegen/extensions/attribution/main.py

@@ -1,10 +1,8 @@
-from typing import Optional
-
 from codegen.extensions.attribution.git_history import GitAttributionTracker
 from codegen.sdk.core.codebase import Codebase
 
 
-def analyze_ai_impact(codebase: Codebase, ai_authors: Optional[list[str]] = None, max_commits: Optional[int] = None) -> dict:
+def analyze_ai_impact(codebase: Codebase, ai_authors: list[str] | None = None, max_commits: int | None = None) -> dict:
     """Analyze the impact of AI on a codebase.
 
     Args:
@@ -57,7 +55,7 @@ def analyze_ai_impact(codebase: Codebase, ai_authors: Optional[list[str]] = None
     }
 
 
-def add_attribution_to_symbols(codebase: Codebase, ai_authors: Optional[list[str]] = None) -> None:
+def add_attribution_to_symbols(codebase: Codebase, ai_authors: list[str] | None = None) -> None:
     """Add attribution information to symbols in the codebase.
 
     This adds the following attributes to each symbol:

src/codegen/extensions/events/codegen_app.py

@@ -1,5 +1,5 @@
 import os
-from typing import Any, Optional
+from typing import Any
 
 from fastapi import FastAPI, Request
 from fastapi.responses import HTMLResponse
@@ -23,7 +23,7 @@ class CodegenApp:
     linear: Linear
     slack: Slack
 
-    def __init__(self, name: str, repo: Optional[str] = None, tmp_dir: str = "/tmp/codegen", commit: str | None = "latest"):
+    def __init__(self, name: str, repo: str | None = None, tmp_dir: str = "/tmp/codegen", commit: str | None = "latest"):
         self.name = name
         self.tmp_dir = tmp_dir
 

src/codegen/extensions/events/github.py

@@ -1,6 +1,7 @@
 import logging
 import os
-from typing import Any, Callable, TypeVar
+from collections.abc import Callable
+from typing import Any, TypeVar
 
 from fastapi import Request
 from github import Github

src/codegen/extensions/events/github_types.py

@@ -1,5 +1,4 @@
 from datetime import datetime
-from typing import Optional
 
 
 class GitHubRepository:
@@ -37,11 +36,11 @@ class GitHubInstallation:
     events: list[str]
     created_at: datetime
     updated_at: datetime
-    single_file_name: Optional[str]
+    single_file_name: str | None
     has_multiple_single_files: bool
     single_file_paths: list[str]
-    suspended_by: Optional[str]
-    suspended_at: Optional[datetime]
+    suspended_by: str | None
+    suspended_at: datetime | None
 
 
 class GitHubUser:
@@ -58,5 +57,5 @@ class GitHubInstallationEvent:
     action: str
     installation: GitHubInstallation
     repositories: list[GitHubRepository]
-    requester: Optional[dict]
+    requester: dict | None
     sender: GitHubUser

src/codegen/extensions/events/linear.py

@@ -1,5 +1,6 @@
 import logging
-from typing import Any, Callable, TypeVar
+from collections.abc import Callable
+from typing import Any, TypeVar
 
 from pydantic import BaseModel
 

src/codegen/extensions/github/types/pull_request.py

@@ -1,4 +1,4 @@
-from typing import Literal, Optional
+from typing import Literal
 
 from pydantic import BaseModel
 
@@ -47,30 +47,30 @@ class PullRequest(BaseModel):
     locked: bool
     title: str
     user: GitHubUser
-    body: Optional[str]
+    body: str | None
     created_at: str
     updated_at: str
-    closed_at: Optional[str]
-    merged_at: Optional[str]
-    merge_commit_sha: Optional[str]
-    assignee: Optional[GitHubUser]
+    closed_at: str | None
+    merged_at: str | None
+    merge_commit_sha: str | None
+    assignee: GitHubUser | None
     assignees: list[GitHubUser]
     requested_reviewers: list[GitHubUser]
     requested_teams: list[dict]
     labels: list[Label]
-    milestone: Optional[dict]
+    milestone: dict | None
     draft: bool
     head: PullRequestRef
     base: PullRequestRef
     _links: PullRequestLinks
     author_association: str
-    auto_merge: Optional[dict]
-    active_lock_reason: Optional[str]
+    auto_merge: dict | None
+    active_lock_reason: str | None
     merged: bool
-    mergeable: Optional[bool]
-    rebaseable: Optional[bool]
+    mergeable: bool | None
+    rebaseable: bool | None
     mergeable_state: str
-    merged_by: Optional[GitHubUser]
+    merged_by: GitHubUser | None
     comments: int
     review_comments: int
     maintainer_can_modify: bool