Fix type checking issues in language.py and pr_review.py

- Add type ignore comments for SDK imports that may not be available
- Fix invalid type annotation for Repository class parameter
- Fix None assignment to typed dict attribute
- Remove unused import after ruff formatting

Co-authored-by: Jay Hack <[email protected]>

Author: codegen-sh[bot]

src/codegen/git/utils/language.py

@@ -47,8 +47,8 @@ def _determine_language_by_file_count(folder_path: str) -> ProgrammingLanguage:
         or if less than MIN_LANGUAGE_RATIO of files match the dominant language
     """
     try:
-        from codegen.sdk.python import PyFile
-        from codegen.sdk.typescript.file import TSFile
+        from codegen.sdk.python import PyFile  # type: ignore
+        from codegen.sdk.typescript.file import TSFile  # type: ignore
 
         EXTENSIONS = {
             ProgrammingLanguage.PYTHON: PyFile.get_extensions(),
@@ -118,9 +118,9 @@ def _determine_language_by_git_file_count(folder_path: str) -> ProgrammingLangua
     from codegen.git.schemas.repo_config import RepoConfig
 
     try:
-        from codegen.sdk.codebase.codebase_context import GLOBAL_FILE_IGNORE_LIST
-        from codegen.sdk.python import PyFile
-        from codegen.sdk.typescript.file import TSFile
+        from codegen.sdk.codebase.codebase_context import GLOBAL_FILE_IGNORE_LIST  # type: ignore
+        from codegen.sdk.python import PyFile  # type: ignore
+        from codegen.sdk.typescript.file import TSFile  # type: ignore
 
         EXTENSIONS = {
             ProgrammingLanguage.PYTHON: PyFile.get_extensions(),

src/codegen/git/utils/pr_review.py

@@ -1,17 +1,16 @@
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
-from github import Repository
 from github.PullRequest import PullRequest
 from unidiff import PatchSet
 
 from codegen.git.models.pull_request_context import PullRequestContext
 from codegen.git.repo_operator.repo_operator import RepoOperator
 
 if TYPE_CHECKING:
-    from codegen.sdk.core.codebase import Codebase, Editable, File
+    from codegen.sdk.core.codebase import Codebase, Editable, File  # type: ignore
 
 
-def get_merge_base(git_repo_client: Repository, pull: PullRequest | PullRequestContext) -> str:
+def get_merge_base(git_repo_client: Any, pull: PullRequest | PullRequestContext) -> str:
     """Gets the merge base of a pull request using a remote GitHub API client.
 
     Args:
@@ -97,7 +96,7 @@ class CodegenPR:
     _op: RepoOperator
 
     # =====[ Computed ]=====
-    _modified_file_ranges: dict[str, list[tuple[int, int]]] = None
+    _modified_file_ranges: dict[str, list[tuple[int, int]]] | None = None
 
     def __init__(self, op: RepoOperator, codebase: "Codebase", pr: PullRequest):
         self._op = op
@@ -130,7 +129,7 @@ def is_modified(self, editable: "Editable") -> bool:
     def modified_symbols(self) -> list[str]:
         # Import SourceFile locally to avoid circular dependencies
         try:
-            from codegen.sdk.core.file import SourceFile
+            from codegen.sdk.core.file import SourceFile  # type: ignore
         except ImportError:
             # If SDK is not available, return empty list
             return []