From be03fe5a47a1c287941514ab0fec14ccc44b707e Mon Sep 17 00:00:00 2001
From: Oleksandr Bazarnov <oleksandr.bazarnov@globallogic.com>
Date: Thu, 20 Feb 2025 16:01:15 +0200
Subject: [PATCH] formatted

---
 airbyte_cdk/connector_builder/test_reader/helpers.py   | 10 +++++-----
 .../connector_builder/test_reader/message_grouper.py   |  2 +-
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/airbyte_cdk/connector_builder/test_reader/helpers.py b/airbyte_cdk/connector_builder/test_reader/helpers.py
index 55052c6ef..fcd36189f 100644
--- a/airbyte_cdk/connector_builder/test_reader/helpers.py
+++ b/airbyte_cdk/connector_builder/test_reader/helpers.py
@@ -600,7 +600,7 @@ def handle_record_message(
 # -------
 
 
-def get_airbyte_cdk_from_message(json_message: Dict[str, JsonType]) -> dict:
+def get_airbyte_cdk_from_message(json_message: Dict[str, JsonType]) -> dict:  # type: ignore
     """
     Retrieves the "airbyte_cdk" dictionary from the provided JSON message.
 
@@ -626,7 +626,7 @@ def get_airbyte_cdk_from_message(json_message: Dict[str, JsonType]) -> dict:
     return airbyte_cdk
 
 
-def get_stream_from_airbyte_cdk(airbyte_cdk: dict) -> dict:
+def get_stream_from_airbyte_cdk(airbyte_cdk: dict) -> dict:  # type: ignore
     """
     Retrieves the "stream" dictionary from the provided "airbyte_cdk" dictionary.
 
@@ -651,14 +651,14 @@ def get_stream_from_airbyte_cdk(airbyte_cdk: dict) -> dict:
     return stream
 
 
-def get_auxiliary_request_title_prefix(stream: dict) -> str:
+def get_auxiliary_request_title_prefix(stream: dict) -> str:  # type: ignore
     """
     Generates a title prefix based on the stream type.
     """
     return "Parent stream: " if stream.get("is_substream", False) else ""
 
 
-def get_http_property_from_message(json_message: Dict[str, JsonType]) -> dict:
+def get_http_property_from_message(json_message: Dict[str, JsonType]) -> dict:  # type: ignore
     """
     Retrieves the "http" dictionary from the provided JSON message.
 
@@ -682,7 +682,7 @@ def get_http_property_from_message(json_message: Dict[str, JsonType]) -> dict:
     return http
 
 
-def get_auxiliary_request_type(stream: dict, http: dict) -> str:
+def get_auxiliary_request_type(stream: dict, http: dict) -> str:  # type: ignore
     """
     Determines the type of the auxiliary request based on the stream and HTTP properties.
     """
diff --git a/airbyte_cdk/connector_builder/test_reader/message_grouper.py b/airbyte_cdk/connector_builder/test_reader/message_grouper.py
index 07d85bf95..e4478a0ad 100644
--- a/airbyte_cdk/connector_builder/test_reader/message_grouper.py
+++ b/airbyte_cdk/connector_builder/test_reader/message_grouper.py
@@ -91,7 +91,7 @@ def get_message_groups(
     current_page_request: Optional[HttpRequest] = None
     current_page_response: Optional[HttpResponse] = None
     latest_state_message: Optional[Dict[str, Any]] = None
-    slice_auxiliary_requests: Optional[List[AuxiliaryRequest]] = []
+    slice_auxiliary_requests: List[AuxiliaryRequest] = []
 
     while records_count < limit and (message := next(messages, None)):
         json_message = airbyte_message_to_json(message)