update pre-commit and refomat

.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 @@
 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:
@@ -79,10 +79,10 @@
         # Convert API response to dict for Job initialization
 
         job = AgentTask(agent_run_response, self.api_client, self.org_id)
         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
@@ -29,7 +30,7 @@
             msg = "GITHUB_TOKEN is not set"
             logger.exception(msg)
             raise ValueError(msg)
         if not self._client:
             self._client = Github(os.getenv("GITHUB_TOKEN"))
         return self._client
 
@@ -68,7 +69,7 @@
                         raise
                 else:
                     # Pass through raw dict if no type validation needed
                     return func(raw_event)
 
             self.registered_handlers[event_name] = new_func
             return new_func

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 @@
     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

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

@@ -1,5 +1,3 @@
-from typing import Optional
-
 from pydantic import BaseModel
 
 from .base import GitHubRepository, GitHubUser
@@ -23,7 +21,7 @@
     created: bool
     deleted: bool
     forced: bool
-    base_ref: Optional[str]
+    base_ref: str | None
     compare: str
     commits: list[GitHubCommit]
     head_commit: GitHubCommit

src/codegen/extensions/graph/create_graph.py

@@ -1,5 +1,3 @@
-from typing import Optional
-
 from codegen.extensions.graph.utils import Node, NodeLabel, Relation, RelationLabel, SimpleGraph
 from codegen.sdk.code_generation.doc_utils.utils import safe_get_class
 from codegen.sdk.core.class_definition import Class
@@ -16,7 +14,7 @@
     # Track existing nodes by name to prevent duplicates
     node_registry = {}  # name -> node_id mapping
 
-    def get_or_create_node(name: str, label: NodeLabel, parent_name: Optional[str] = None, properties: dict | None = None):
+    def get_or_create_node(name: str, label: NodeLabel, parent_name: str | None = None, properties: dict | None = None):
         """Get existing node or create new one if it doesn't exist."""
         full_name = f"{parent_name}.{name}" if parent_name and parent_name != "Class" else name
         if full_name in node_registry:

src/codegen/extensions/graph/neo4j_exporter.py

@@ -26,7 +26,7 @@
         with self.driver.session() as session:
             # Create nodes
             for node in graph.nodes.values():
-                properties = {"name": node.name, "full_name": node.full_name, **{k: str(v) if isinstance(v, (dict, list)) else v for k, v in node.properties.items()}}
+                properties = {"name": node.name, "full_name": node.full_name, **{k: str(v) if isinstance(v, dict | list) else v for k, v in node.properties.items()}}
 
                 query = f"CREATE (n:{node.label} {{{', '.join(f'{k}: ${k}' for k in properties.keys())}}})"
                 session.run(query, properties)
@@ -36,7 +36,7 @@
                 source_node = graph.nodes[relation.source_id]
                 target_node = graph.nodes[relation.target_id]
 
-                properties = {**{k: str(v) if isinstance(v, (dict, list)) else v for k, v in relation.properties.items()}}
+                properties = {**{k: str(v) if isinstance(v, dict | list) else v for k, v in relation.properties.items()}}
 
                 query = (
                     f"MATCH (source:{source_node.label} {{full_name: $source_name}}), "

src/codegen/extensions/index/code_index.py

@@ -2,7 +2,7 @@
 
 from abc import ABC, abstractmethod
 from pathlib import Path
-from typing import Optional, TypeVar
+from typing import TypeVar
 
 import numpy as np
 
@@ -34,9 +34,9 @@
             codebase: The codebase to index
         """
         self.codebase = codebase
-        self.E: Optional[np.ndarray] = None
-        self.items: Optional[np.ndarray] = None
-        self.commit_hash: Optional[str] = None
+        self.E: np.ndarray | None = None
+        self.items: np.ndarray | None = None
+        self.commit_hash: str | None = None
 
     @property
     @abstractmethod
@@ -151,7 +151,7 @@
         # Update commit hash
         self.commit_hash = self._get_current_commit()
 
-    def save(self, save_path: Optional[str] = None) -> None:
+    def save(self, save_path: str | None = None) -> None:
         """Save the index to disk."""
         if self.E is None or self.items is None:
             msg = "No embeddings to save. Call create() first."
@@ -162,7 +162,7 @@
 
         self._save_index(save_path)
 
-    def load(self, load_path: Optional[str] = None) -> None:
+    def load(self, load_path: str | None = None) -> None:
         """Load the index from disk."""
         load_path = Path(load_path) if load_path else self._get_default_save_path()
 

src/codegen/extensions/index/file_index.py

@@ -2,7 +2,6 @@
 
 import pickle
 from pathlib import Path
-from typing import Optional
 
 import modal
 import numpy as np
@@ -87,7 +86,7 @@ def delete_modal_dict(self) -> bool:
             logger.exception(f"Failed to delete Modal Dict: {e}")
             return False
 
-    def modal_dict_exists(self, commit_hash: Optional[str] = None) -> bool:
+    def modal_dict_exists(self, commit_hash: str | None = None) -> bool:
         """Check if a Modal Dict exists for a specific commit.
 
         Args:

src/codegen/extensions/linear/linear_client.py

@@ -1,5 +1,4 @@
 import os
-from typing import Optional
 
 import requests
 from requests.adapters import HTTPAdapter
@@ -15,7 +14,7 @@ class LinearClient:
     api_headers: dict
     api_endpoint = "https://api.linear.app/graphql"
 
-    def __init__(self, access_token: Optional[str] = None, team_id: Optional[str] = None, max_retries: int = 3, backoff_factor: float = 0.5):
+    def __init__(self, access_token: str | None = None, team_id: str | None = None, max_retries: int = 3, backoff_factor: float = 0.5):
         if not access_token:
             access_token = os.getenv("LINEAR_ACCESS_TOKEN")
             if not access_token: