|
22 | 22 |
|
23 | 23 | import requests |
24 | 24 | from typing_extensions import deprecated |
25 | | -from airbyte_cdk.sources.http_logger import format_http_message |
| 25 | + |
26 | 26 | from airbyte_cdk.models import AirbyteMessage |
27 | 27 | from airbyte_cdk.sources.declarative.extractors.http_selector import HttpSelector |
28 | 28 | from airbyte_cdk.sources.declarative.incremental import ResumableFullRefreshCursor |
|
41 | 41 | from airbyte_cdk.sources.declarative.requesters.requester import Requester |
42 | 42 | from airbyte_cdk.sources.declarative.retrievers.retriever import Retriever |
43 | 43 | from airbyte_cdk.sources.declarative.stream_slicers.stream_slicer import StreamSlicer |
| 44 | +from airbyte_cdk.sources.http_logger import format_http_message |
44 | 45 | from airbyte_cdk.sources.source import ExperimentalClassWarning |
45 | 46 | from airbyte_cdk.sources.streams.core import StreamData |
46 | 47 | from airbyte_cdk.sources.types import Config, Record, StreamSlice, StreamState |
@@ -655,12 +656,18 @@ class SimpleRetrieverTestReadDecorator(SimpleRetriever): |
655 | 656 |
|
656 | 657 | def __post_init__(self, options: Mapping[str, Any]) -> None: |
657 | 658 | super().__post_init__(options) |
658 | | - self.log_formatter = (lambda response: format_http_message( |
| 659 | + self.log_formatter = ( |
| 660 | + ( |
| 661 | + lambda response: format_http_message( |
659 | 662 | response, |
660 | 663 | f"Stream '{self.name}' request", |
661 | 664 | f"Request performed in order to extract records for stream '{self.name}'", |
662 | 665 | self.name, |
663 | | - )) if not self.log_formatter else self.log_formatter |
| 666 | + ) |
| 667 | + ) |
| 668 | + if not self.log_formatter |
| 669 | + else self.log_formatter |
| 670 | + ) |
664 | 671 |
|
665 | 672 | if self.maximum_number_of_slices and self.maximum_number_of_slices < 1: |
666 | 673 | raise ValueError( |
|
0 commit comments