feat(types): add strict typing to config, logging_utils, and models modules

Author: NikkeTryHard

.gitignore

@@ -260,3 +260,5 @@ browser_utils/generated_*.js
 # Docker 环境的实际配置文件（保留示例文件）
 docker/.env
 docker/my_*.json
+
+monkeytype.sqlite3

config/constants.py

@@ -42,7 +42,7 @@
 try:
     DEFAULT_STOP_SEQUENCES = json.loads(os.environ.get("DEFAULT_STOP_SEQUENCES", "[]"))
 except (json.JSONDecodeError, TypeError):
-    DEFAULT_STOP_SEQUENCES = []  # 回退到默认值 (空列表)
+    DEFAULT_STOP_SEQUENCES = []  # type: ignore[reportConstantRedefinition]  # 回退到默认值 (空列表)
 
 # --- URL模式 ---
 AI_STUDIO_URL_PATTERN = os.environ.get("AI_STUDIO_URL_PATTERN", "aistudio.google.com/")

models/exceptions.py

@@ -17,7 +17,7 @@
 """
 
 import time
-from typing import Any, Optional
+from typing import Any, Dict, List, Optional
 
 from fastapi import HTTPException
 
@@ -63,7 +63,7 @@ def to_http_exception(self) -> HTTPException:
         Returns:
             HTTPException with appropriate status code and headers
         """
-        headers = {}
+        headers: Dict[str, str] = {}
         if self.retry_after is not None:
             headers["Retry-After"] = str(self.retry_after)
 
@@ -165,7 +165,7 @@ class InvalidModelError(ModelError):
     """Requested model does not exist or is not available."""
 
     def __init__(
-        self, model_id: str, available_models: Optional[list] = None, **kwargs: Any
+        self, model_id: str, available_models: Optional[List[str]] = None, **kwargs: Any
     ):
         message = f"Invalid model '{model_id}'"
         if available_models:

models/logging.py

@@ -3,18 +3,18 @@
 import json
 import logging
 import sys
-from typing import Dict
+from typing import Dict, List
 
 from fastapi import WebSocket, WebSocketDisconnect
 
 
 class StreamToLogger:
-    def __init__(self, logger_instance, log_level=logging.INFO):
+    def __init__(self, logger_instance: logging.Logger, log_level: int = logging.INFO):
         self.logger = logger_instance
         self.log_level = log_level
         self.linebuf = ""
 
-    def write(self, buf):
+    def write(self, buf: str):
         try:
             temp_linebuf = self.linebuf + buf
             self.linebuf = ""
@@ -70,7 +70,7 @@ def disconnect(self, client_id: str):
     async def broadcast(self, message: str):
         if not self.active_connections:
             return
-        disconnected_clients = []
+        disconnected_clients: List[str] = []
         active_conns_copy = list(self.active_connections.items())
         logger = logging.getLogger("AIStudioProxyServer")
         for client_id, connection in active_conns_copy:

monkeytype_config.py

@@ -0,0 +1,82 @@
+"""MonkeyType configuration for AIStudioProxyAPI.
+
+This config:
+- Filters to only trace project modules (exclude tests, external libs)
+- Enables TypedDict generation for config dicts
+- Applies type rewriters to clean up messy unions
+- Increases query limit for comprehensive coverage
+"""
+
+from monkeytype.config import DefaultConfig
+from monkeytype.typing import (
+    ChainedRewriter,
+    RemoveEmptyContainers,
+    RewriteConfigDict,
+    RewriteLargeUnion,
+)
+
+
+class AIStudioProxyConfig(DefaultConfig):
+    """Custom MonkeyType configuration for this project."""
+
+    def code_filter(self):
+        """Only trace project modules, exclude tests and external libraries."""
+
+        def should_trace(code):
+            # Normalize Windows path separators
+            filename = code.co_filename.replace("\\", "/")
+
+            # Project modules to trace
+            project_modules = [
+                "api_utils",
+                "browser_utils",
+                "stream",
+                "config",
+                "models",
+                "launcher",
+                "logging_utils",
+            ]
+
+            # Check if file is in any project module
+            for module in project_modules:
+                if f"/{module}/" in filename or filename.endswith(f"/{module}.py"):
+                    # Exclude test files
+                    if "/tests/" not in filename and "/test_" not in filename:
+                        return True
+
+            return False
+
+        return should_trace
+
+    def type_rewriter(self):
+        """Clean up generated types with chained rewriters."""
+        return ChainedRewriter(
+            [
+                RemoveEmptyContainers(),  # Union[List[Any], List[int]] -> List[int]
+                RewriteConfigDict(),  # Union[Dict[K,V1], Dict[K,V2]] -> Dict[K, Union[V1,V2]]
+                RewriteLargeUnion(
+                    max_union_len=3
+                ),  # Large unions -> Any (strict: max 3 elements)
+            ]
+        )
+
+    def max_typed_dict_size(self) -> int:
+        """Enable TypedDict generation for dictionaries.
+
+        Since 19.11.2, TypedDict generation is disabled by default.
+        This enables it for dicts with up to 50 keys, which is critical
+        for config.settings and similar modules.
+        """
+        return 50
+
+    def query_limit(self) -> int:
+        """Increase query limit for comprehensive type inference.
+
+        Default is 2000. We increase to 5000 to capture more traces
+        and improve type accuracy, especially for polymorphic functions.
+        """
+        return 5000
+
+
+# MonkeyType will automatically find and use this CONFIG instance
+CONFIG = AIStudioProxyConfig()

pyproject.toml

@@ -48,7 +48,9 @@ build-backend = "poetry.core.masonry.api"
 asyncio_mode = "auto"
 testpaths = ["tests"]
 python_files = "test_*.py"
-addopts = "-v --cov=api_utils --cov=browser_utils --cov=stream --cov=config --cov=models --cov-report=term-missing"
+# Coverage disabled by default for MonkeyType compatibility (uses same sys.setprofile hook)
+# To re-enable coverage: pytest --cov=api_utils --cov=browser_utils --cov=stream --cov=config --cov=models --cov-report=term-missing
+addopts = "-v"
 timeout = 10
 env = [
     "LAUNCH_MODE=test",

pytest.ini

@@ -2,7 +2,9 @@
 asyncio_mode = auto
 testpaths = tests
 python_files = test_*.py
-addopts = -v --cov=api_utils --cov=browser_utils --cov=stream --cov=config --cov=models --cov-report=term-missing
+# Coverage disabled by default for MonkeyType compatibility (uses same sys.setprofile hook)
+# To re-enable coverage: pytest --cov=api_utils --cov=browser_utils --cov=stream --cov=config --cov=models --cov-report=term-missing
+addopts = -v
 timeout = 30
 markers =
     integration: Integration tests that use real components (locks, queues, state) instead of mocks