make get_stream_slice public

airbytehq · Dec 11, 2022 · 9170fe5 · 9170fe5
1 parent c18748e
commit 9170fe5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/airbyte-cdk/python/airbyte_cdk/sources/utils/stream_helpers.py b/airbyte-cdk/python/airbyte_cdk/sources/utils/stream_helpers.py
@@ -18,12 +18,12 @@ def get_first_record(self, stream: Stream) -> StreamData:
         :return: StreamData containing the first record in the stream
         """
         # Some streams need a stream slice to read records (e.g. if they have a SubstreamSlicer)
-        stream_slice = self._get_stream_slice(stream)
+        stream_slice = self.get_stream_slice(stream)
         records = stream.read_records(sync_mode=SyncMode.full_refresh, stream_slice=stream_slice)
         next(records)
 
     @staticmethod
-    def _get_stream_slice(stream: Stream) -> Optional[Mapping[str, Any]]:
+    def get_stream_slice(stream: Stream) -> Optional[Mapping[str, Any]]:
         """
         Gets the first stream_slice from a given stream's stream_slices.