Fix type checking errors in MCP tests and related modules

- Fixed return type annotations in repo_operator.py methods
- Added proper None handling in repo_config.py
- Added try/except for SDK imports in language.py with fallback extensions
- Fixed Repository import path in pr_review.py
- Fixed type annotations in docs decorators
- Added type ignore comments for expected TypeError tests
- Reduced type checking errors from 67 to 57 diagnostics

Author: codegen-sh[bot]

src/codegen/git/repo_operator/repo_operator.py

@@ -566,7 +566,7 @@ def emptydir(self, path: str) -> None:
                 if os.path.isfile(file_path):
                     os.remove(file_path)
 
-    def get_file(self, path: str) -> str:
+    def get_file(self, path: str) -> str | None:
         """Returns the contents of a file"""
         file_path = self.abspath(path)
         try:
@@ -621,7 +621,7 @@ def get_filepaths_for_repo(self, ignore_list):
                 decoded_filepath, _ = codecs.escape_decode(raw_filepath)
 
                 # Step 4: Decode those bytes as UTF-8 to get the actual Unicode text
-                filepath = decoded_filepath.decode("utf-8")
+                filepath = decoded_filepath.decode("utf-8")  # type: ignore[union-attr]
 
                 # Step 5: Replace the original filepath with the decoded filepath
                 filepaths[i] = filepath
@@ -754,7 +754,7 @@ def get_pr_data(self, pr_number: int) -> dict:
         """Returns the data associated with a PR"""
         return self.remote_git_repo.get_pr_data(pr_number)
 
-    def create_pr_comment(self, pr_number: int, body: str) -> IssueComment:
+    def create_pr_comment(self, pr_number: int, body: str) -> IssueComment | None:
         """Create a general comment on a pull request.
 
         Args:
@@ -765,6 +765,7 @@ def create_pr_comment(self, pr_number: int, body: str) -> IssueComment:
         if pr:
             comment = self.remote_git_repo.create_issue_comment(pr, body)
             return comment
+        return None
 
     def create_pr_review_comment(
         self,

src/codegen/git/schemas/repo_config.py

@@ -31,11 +31,13 @@ class RepoConfig(BaseModel):
     @classmethod
     def from_envs(cls) -> "RepoConfig":
         default_repo_config = RepositoryConfig()
+        path = default_repo_config.path or os.getcwd()
+        language = default_repo_config.language or "python"
         return RepoConfig(
             name=default_repo_config.name,
             full_name=default_repo_config.full_name,
-            base_dir=os.path.dirname(default_repo_config.path),
-            language=ProgrammingLanguage(default_repo_config.language.upper()),
+            base_dir=os.path.dirname(path),
+            language=ProgrammingLanguage(language.upper()),
         )
 
     @classmethod

src/codegen/git/utils/language.py

@@ -46,13 +46,20 @@ def _determine_language_by_file_count(folder_path: str) -> ProgrammingLanguage:
         ProgrammingLanguage: The dominant programming language, or OTHER if no matching files found
         or if less than MIN_LANGUAGE_RATIO of files match the dominant language
     """
-    from codegen.sdk.python import PyFile
-    from codegen.sdk.typescript.file import TSFile
-
-    EXTENSIONS = {
-        ProgrammingLanguage.PYTHON: PyFile.get_extensions(),
-        ProgrammingLanguage.TYPESCRIPT: TSFile.get_extensions(),
-    }
+    try:
+        from codegen.sdk.python import PyFile
+        from codegen.sdk.typescript.file import TSFile
+        
+        EXTENSIONS = {
+            ProgrammingLanguage.PYTHON: PyFile.get_extensions(),
+            ProgrammingLanguage.TYPESCRIPT: TSFile.get_extensions(),
+        }
+    except ImportError:
+        # Fallback to hardcoded extensions if SDK modules are not available
+        EXTENSIONS = {
+            ProgrammingLanguage.PYTHON: [".py", ".pyx", ".pyi"],
+            ProgrammingLanguage.TYPESCRIPT: [".ts", ".tsx", ".js", ".jsx"],
+        }
 
     folder = Path(folder_path)
     if not folder.exists() or not folder.is_dir():

src/codegen/git/utils/pr_review.py

@@ -1,7 +1,7 @@
 from typing import TYPE_CHECKING
 
-from github import Repository
 from github.PullRequest import PullRequest
+from github.Repository import Repository
 from unidiff import PatchSet
 
 from codegen.git.models.pull_request_context import PullRequestContext
@@ -97,7 +97,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

src/codegen/shared/decorators/docs.py

@@ -2,14 +2,14 @@
 import inspect
 from collections.abc import Callable
 from dataclasses import dataclass
-from typing import TypeVar
+from typing import Any, TypeVar
 
 
 @dataclass
 class DocumentedObject:
     name: str
     module: str
-    object: any
+    object: Any
 
     def __lt__(self, other):
         return self.module < other.module

tests/unit/codegen/agents/test_agent.py

@@ -73,7 +73,7 @@ def test_agent_init_requires_token(self):
         """Test that Agent initialization requires a token parameter."""
         # This should raise a TypeError because token is a required parameter
         with pytest.raises(TypeError, match="missing 1 required positional argument: 'token'"):
-            Agent()  # Missing required token parameter
+            Agent()  # Missing required token parameter  # type: ignore[call-arg]
 
     def test_agent_init_with_none_token(self):
         """Test Agent initialization with None token."""

tests/unit/codegen/agents/test_usage_demo.py

@@ -39,7 +39,7 @@ def test_error_handling_no_token():
     with pytest.raises(TypeError, match="missing 1 required positional argument: 'token'"):
         from codegen.agents.agent import Agent
 
-        Agent()  # No token provided
+        Agent()  # No token provided  # type: ignore[call-arg]
 
 
 def test_basic_initialization_variations():