Airbyte CDK: fix mypy

Author: artem1205

airbyte_cdk/sources/declarative/decoders/json_decoder.py

@@ -5,7 +5,7 @@
 import logging
 from dataclasses import InitVar, dataclass
 from gzip import decompress
-from typing import Any, Generator, Mapping
+from typing import Any, Generator, Mapping, MutableMapping
 
 import requests
 from airbyte_cdk.sources.declarative.decoders.decoder import Decoder
@@ -25,7 +25,7 @@ class JsonDecoder(Decoder):
     def is_stream_response(self) -> bool:
         return False
 
-    def decode(self, response: requests.Response) -> Generator[Mapping[str, Any], None, None]:
+    def decode(self, response: requests.Response) -> Generator[MutableMapping[str, Any], None, None]:
         """
         Given the response is an empty string or an emtpy list, the function will return a generator with an empty mapping.
         """
@@ -41,7 +41,7 @@ def decode(self, response: requests.Response) -> Generator[Mapping[str, Any], No
     @staticmethod
     def parse_body_json(
         body_json: Mapping[str, Any] | list,
-    ) -> Generator[Mapping[str, Any], None, None]:
+    ) -> Generator[MutableMapping[str, Any], None, None]:
         if not isinstance(body_json, list):
             body_json = [body_json]
         if len(body_json) == 0:
@@ -61,7 +61,7 @@ class IterableDecoder(Decoder):
     def is_stream_response(self) -> bool:
         return True
 
-    def decode(self, response: requests.Response) -> Generator[Mapping[str, Any], None, None]:
+    def decode(self, response: requests.Response) -> Generator[MutableMapping[str, Any], None, None]:
         for line in response.iter_lines():
             yield {"record": line.decode()}
 
@@ -77,7 +77,7 @@ class JsonlDecoder(Decoder):
     def is_stream_response(self) -> bool:
         return True
 
-    def decode(self, response: requests.Response) -> Generator[Mapping[str, Any], None, None]:
+    def decode(self, response: requests.Response) -> Generator[MutableMapping[str, Any], None, None]:
         # TODO???: set delimiter? usually it is `\n` but maybe it would be useful to set optional?
         #  https://github.com/airbytehq/airbyte-internal-issues/issues/8436
         for record in response.iter_lines():
@@ -88,6 +88,6 @@ def decode(self, response: requests.Response) -> Generator[Mapping[str, Any], No
 class GzipJsonDecoder(JsonDecoder):
     encoding: str = "utf-8"
 
-    def decode(self, response: requests.Response) -> Generator[Mapping[str, Any], None, None]:
+    def decode(self, response: requests.Response) -> Generator[MutableMapping[str, Any], None, None]:
         raw_string = decompress(response.content).decode(encoding=self.encoding)
         yield from self.parse_body_json(orjson.loads(raw_string))