Merge branch 'improve_typing'

Author: Agent-Hellboy

README.md

@@ -2,6 +2,8 @@
 
 <div align="center">
 
+<img src="icon.png" alt="MCP Server Fuzzer Icon" width="100" height="100">
+
 **A comprehensive super-aggressive CLI-based fuzzing tool for MCP servers**
 
 *Multi-protocol support • Two-phase fuzzing • Built-in safety • Rich reporting • async runtime and async fuzzing of mcp tools*

icon.png

@@ -216,7 +216,7 @@ def check_response_schema_conformity(response: Any, schema: dict[str, Any]) -> b
 def verify_response_invariants(
     response: Any,
     expected_error_codes: list[int] | None = None,
-    schema: dict[str, Any | None] = None,
+    schema: dict[str, Any] | None = None,
 ) -> bool:
     """
     Verify all invariants for a response.
@@ -248,7 +248,7 @@ def verify_response_invariants(
 async def verify_batch_responses(
     responses: list[Any],
     expected_error_codes: list[int] | None = None,
-    schema: dict[str, Any | None] = None,
+    schema: dict[str, Any] | None = None,
 ) -> dict[int, bool | str]:
     """
     Verify invariants for a batch of responses asynchronously.

mcp_fuzzer/fuzz_engine/invariants.py

@@ -12,7 +12,7 @@
 class TransportProtocol(ABC):
     @abstractmethod
     async def send_request(
-        self, method: str, params: dict[str, Any | None] = None
+        self, method: str, params: dict[str, Any] | None = None
     ) -> Any:
         pass
 
@@ -22,7 +22,7 @@ async def send_raw(self, payload: dict[str, Any]) -> Any:
 
     @abstractmethod
     async def send_notification(
-        self, method: str, params: dict[str, Any | None] = None
+        self, method: str, params: dict[str, Any] | None = None
     ) -> None:
         pass
 

mcp_fuzzer/transport/base.py

@@ -26,7 +26,7 @@ def register(
         name: str,
         transport_class: Type[TransportProtocol],
         description: str = "",
-        config_schema: dict[str, Any | None] = None,
+        config_schema: dict[str, Any] | None = None,
         factory_function: Callable | None = None,
     ) -> None:
         """Register a custom transport.
@@ -155,7 +155,7 @@ def register_custom_transport(
     name: str,
     transport_class: Type[TransportProtocol],
     description: str = "",
-    config_schema: dict[str, Any | None] = None,
+    config_schema: dict[str, Any] | None = None,
     factory_function: Callable | None = None,
 ) -> None:
     """Register a custom transport with the global registry.

mcp_fuzzer/transport/custom.py

@@ -98,7 +98,7 @@ def __init__(self, *args, **kwargs):
     def _create_jsonrpc_request(
         self,
         method: str,
-        params: dict[str, Any | None] = None,
+        params: dict[str, Any] | None = None,
         request_id: str | int | None = None,
     ) -> JSONRPCRequest:
         """Create a JSON-RPC request payload.
@@ -129,7 +129,7 @@ def _create_jsonrpc_request(
         return payload
 
     def _create_jsonrpc_notification(
-        self, method: str, params: dict[str, Any | None] = None
+        self, method: str, params: dict[str, Any] | None = None
     ) -> JSONRPCNotification:
         """Create a JSON-RPC notification payload.
 

mcp_fuzzer/transport/mixins.py

@@ -157,7 +157,7 @@ def _extract_content_type(self, response: httpx.Response) -> str:
         return response.headers.get(CONTENT_TYPE, "").lower()
 
     async def send_request(
-        self, method: str, params: dict[str, Any | None] = None
+        self, method: str, params: dict[str, Any] | None = None
     ) -> Any:
         request_id = str(asyncio.get_running_loop().time())
         payload = {
@@ -276,7 +276,7 @@ async def send_raw(self, payload: dict[str, Any]) -> Any:
             raise Exception(f"Unexpected content type: {ct}")
 
     async def send_notification(
-        self, method: str, params: dict[str, Any | None] = None
+        self, method: str, params: dict[str, Any] | None = None
     ) -> None:
         payload = {"jsonrpc": "2.0", "method": method, "params": params or {}}
         headers = self._prepare_headers()