Reformat

Author: bagel897

src/codegen/extensions/attribution/git_history.py

@@ -3,13 +3,14 @@
 from datetime import datetime
 
 import pygit2
-from graph_sitter.core.codebase import Codebase
-from graph_sitter.core.file import SourceFile
-from graph_sitter.core.symbol import Symbol
 from intervaltree import IntervalTree
 from pygit2 import Commit, Patch
 from pygit2.enums import CheckoutStrategy, DeltaStatus, SortMode
 
+from graph_sitter.core.codebase import Codebase
+from graph_sitter.core.file import SourceFile
+from graph_sitter.core.symbol import Symbol
+
 
 class GitAttributionTracker:
     """Tracks attribution information for code symbols based on git history."""

src/codegen/extensions/attribution/main.py

@@ -1,6 +1,5 @@
-from graph_sitter.core.codebase import Codebase
-
 from codegen.extensions.attribution.git_history import GitAttributionTracker
+from graph_sitter.core.codebase import Codebase
 
 
 def analyze_ai_impact(codebase: Codebase, ai_authors: list[str] | None = None, max_commits: int | None = None) -> dict:

src/codegen/extensions/graph/create_graph.py

@@ -1,11 +1,10 @@
+from codegen.extensions.graph.utils import Node, NodeLabel, Relation, RelationLabel, SimpleGraph
 from graph_sitter.code_generation.doc_utils.utils import safe_get_class
 from graph_sitter.core.class_definition import Class
 from graph_sitter.core.external_module import ExternalModule
 from graph_sitter.core.function import Function
 from graph_sitter.python.class_definition import PyClass
 
-from codegen.extensions.graph.utils import Node, NodeLabel, Relation, RelationLabel, SimpleGraph
-
 
 def create_codebase_graph(codebase):
     """Create a SimpleGraph representing the codebase structure."""

src/codegen/extensions/graph/main.py

@@ -1,7 +1,6 @@
-from graph_sitter.core.codebase import Codebase
-
 from codegen.extensions.graph.create_graph import create_codebase_graph
 from codegen.extensions.graph.neo4j_exporter import Neo4jExporter
+from graph_sitter.core.codebase import Codebase
 
 
 def visualize_codebase(codebase, neo4j_uri: str, username: str, password: str):

src/codegen/extensions/index/code_index.py

@@ -5,6 +5,7 @@
 from typing import TypeVar
 
 import numpy as np
+
 from graph_sitter.core.codebase import Codebase
 
 T = TypeVar("T")  # Type of the items being indexed (e.g., File, Symbol)

src/codegen/extensions/lsp/codemods/base.py

@@ -1,9 +1,10 @@
 from abc import ABC, abstractmethod
 from typing import TYPE_CHECKING, ClassVar
 
-from graph_sitter.core.interfaces.editable import Editable
 from lsprotocol import types
 
+from graph_sitter.core.interfaces.editable import Editable
+
 if TYPE_CHECKING:
     from codegen.extensions.lsp.server import CodegenLanguageServer
 

src/codegen/extensions/lsp/codemods/move_symbol_to_file.py

@@ -1,8 +1,7 @@
 from typing import TYPE_CHECKING
 
-from graph_sitter.core.interfaces.editable import Editable
-
 from codegen.extensions.lsp.codemods.base import CodeAction
+from graph_sitter.core.interfaces.editable import Editable
 
 if TYPE_CHECKING:
     from codegen.extensions.lsp.server import CodegenLanguageServer

src/codegen/extensions/lsp/codemods/split_tests.py

@@ -1,10 +1,9 @@
 from typing import TYPE_CHECKING
 
+from codegen.extensions.lsp.codemods.base import CodeAction
 from graph_sitter.core.function import Function
 from graph_sitter.core.interfaces.editable import Editable
 
-from codegen.extensions.lsp.codemods.base import CodeAction
-
 if TYPE_CHECKING:
     from codegen.extensions.lsp.server import CodegenLanguageServer
 

src/codegen/extensions/lsp/document_symbol.py

@@ -1,10 +1,10 @@
-from graph_sitter.core.class_definition import Class
-from graph_sitter.core.interfaces.editable import Editable
-from graph_sitter.extensions.sort import sort_editables
 from lsprotocol.types import DocumentSymbol
 
 from codegen.extensions.lsp.kind import get_kind
 from codegen.extensions.lsp.range import get_range
+from graph_sitter.core.class_definition import Class
+from graph_sitter.core.interfaces.editable import Editable
+from graph_sitter.extensions.sort import sort_editables
 
 
 def get_document_symbol(node: Editable) -> DocumentSymbol:

src/codegen/extensions/lsp/kind.py

@@ -1,3 +1,5 @@
+from lsprotocol.types import SymbolKind
+
 from graph_sitter.core.assignment import Assignment
 from graph_sitter.core.class_definition import Class
 from graph_sitter.core.file import File
@@ -6,7 +8,6 @@
 from graph_sitter.core.interfaces.editable import Editable
 from graph_sitter.core.statements.attribute import Attribute
 from graph_sitter.typescript.namespace import TSNamespace
-from lsprotocol.types import SymbolKind
 
 kinds = {
     File: SymbolKind.File,