|
23 | 23 | get_origin, |
24 | 24 | get_type_hints, |
25 | 25 | ) |
26 | | -from requests import Response |
| 26 | + |
27 | 27 | from isodate import parse_duration |
28 | 28 | from pydantic.v1 import BaseModel |
| 29 | +from requests import Response |
29 | 30 |
|
30 | 31 | from airbyte_cdk.connector_builder.models import ( |
31 | 32 | LogMessage as ConnectorBuilderLogMessage, |
@@ -2401,13 +2402,15 @@ def create_dynamic_schema_loader( |
2401 | 2402 | stream_slicer=combined_slicers, |
2402 | 2403 | transformations=[], |
2403 | 2404 | use_cache=True, |
2404 | | - log_formatter=(lambda response: format_http_message( |
| 2405 | + log_formatter=( |
| 2406 | + lambda response: format_http_message( |
2405 | 2407 | response, |
2406 | 2408 | f"Schema loader '{name}' request", |
2407 | 2409 | f"Request performed in order to extract schema.", |
2408 | 2410 | name, |
2409 | 2411 | is_auxiliary=True, |
2410 | | - )), |
| 2412 | + ) |
| 2413 | + ), |
2411 | 2414 | ) |
2412 | 2415 | schema_type_identifier = self._create_component_from_model( |
2413 | 2416 | model.schema_type_identifier, config=config, parameters=model.parameters or {} |
@@ -3171,12 +3174,16 @@ def _get_url() -> str: |
3171 | 3174 | config=config, |
3172 | 3175 | maximum_number_of_slices=self._limit_slices_fetched or 5, |
3173 | 3176 | ignore_stream_slicer_parameters_on_paginated_requests=ignore_stream_slicer_parameters_on_paginated_requests, |
3174 | | - log_formatter=(lambda response: format_http_message( |
3175 | | - response, |
3176 | | - f"Stream '{name}' request", |
3177 | | - f"Request performed in order to extract records for stream '{name}'", |
3178 | | - name, |
3179 | | - )) if not log_formatter else log_formatter, |
| 3177 | + log_formatter=( |
| 3178 | + lambda response: format_http_message( |
| 3179 | + response, |
| 3180 | + f"Stream '{name}' request", |
| 3181 | + f"Request performed in order to extract records for stream '{name}'", |
| 3182 | + name, |
| 3183 | + ) |
| 3184 | + ) |
| 3185 | + if not log_formatter |
| 3186 | + else log_formatter, |
3180 | 3187 | parameters=model.parameters or {}, |
3181 | 3188 | ) |
3182 | 3189 | return SimpleRetriever( |
|
0 commit comments