update type hints

Author: aaronsteers

airbyte_cdk/legacy/sources/declarative/declarative_stream.py

@@ -202,7 +202,10 @@ def get_cursor(self) -> Optional[Cursor]:
             return self.retriever.cursor
         return None
 
-    def fetch_record(self, pk_value: Any) -> Optional[Mapping[str, Any]]:
+    def fetch_record(
+        self,
+        pk_value: Any | Mapping[str, Any],
+    ) -> Optional[Mapping[str, Any]]:
         """
         Fetch a single record by primary key value.
 

airbyte_cdk/sources/abstract_source.py

@@ -198,9 +198,9 @@ def read(
                         logger.info(timer.report())
 
         if len(stream_name_to_exception) > 0:
-            error_message = generate_failed_streams_error_message(
-                {key: [value] for key, value in stream_name_to_exception.items()}
-            )
+            error_message = generate_failed_streams_error_message({
+                key: [value] for key, value in stream_name_to_exception.items()
+            })
             logger.info(error_message)
             # We still raise at least one exception when a stream raises an exception because the platform currently relies
             # on a non-zero exit code to determine if a sync attempt has failed. We also raise the exception as a config_error
@@ -326,7 +326,10 @@ def stop_sync_on_stream_failure(self) -> bool:
         return False
 
     def fetch_record(
-        self, stream_name: str, pk_value: Any, config: Mapping[str, Any]
+        self,
+        stream_name: str,
+        pk_value: Any | Mapping[str, Any],
+        config: Mapping[str, Any],
     ) -> Optional[Mapping[str, Any]]:
         """
         Fetch a single record from a stream by primary key.