Skip to content

Commit 8080867

Browse files
committed
file-mode-api: remove file reference of the FileBasedStreamFacade
1 parent 2be82e9 commit 8080867

File tree

1 file changed

+2
-11
lines changed
  • airbyte_cdk/sources/file_based/stream/concurrent

1 file changed

+2
-11
lines changed

airbyte_cdk/sources/file_based/stream/concurrent/adapters.py

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
import copy
66
import logging
7-
from functools import cache, lru_cache
7+
from functools import lru_cache
88
from typing import TYPE_CHECKING, Any, Iterable, List, Mapping, MutableMapping, Optional, Union
99

1010
from typing_extensions import deprecated
@@ -258,12 +258,7 @@ def read(self) -> Iterable[Record]:
258258
and record_data.record is not None
259259
):
260260
# `AirbyteMessage`s of type `Record` should also be yielded so they are enqueued
261-
# If stream is flagged for file_transfer the record should data in file key
262-
record_message_data = (
263-
record_data.record.file
264-
if self._use_file_transfer()
265-
else record_data.record.data
266-
)
261+
record_message_data = record_data.record.data
267262
if not record_message_data:
268263
raise ExceptionWithDisplayMessage("A record without data was found")
269264
else:
@@ -305,10 +300,6 @@ def __hash__(self) -> int:
305300
def stream_name(self) -> str:
306301
return self._stream.name
307302

308-
@cache
309-
def _use_file_transfer(self) -> bool:
310-
return hasattr(self._stream, "use_file_transfer") and self._stream.use_file_transfer
311-
312303
def __repr__(self) -> str:
313304
return f"FileBasedStreamPartition({self._stream.name}, {self._slice})"
314305

0 commit comments

Comments
 (0)