Skip to content

Commit 796bb34

Browse files
fix(file-based): override primary_key in PermissionsFileBasedStream to avoid invalid parser-defined key (#903)
Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
1 parent 201bdb8 commit 796bb34

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

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

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -155,10 +155,7 @@ def get_json_schema(self) -> Mapping[str, Any]:
155155

156156
@property
157157
def primary_key(self) -> PrimaryKeyType:
158-
return (
159-
self._legacy_stream.config.primary_key
160-
or self.get_parser().get_parser_defined_primary_key(self._legacy_stream.config)
161-
)
158+
return self._legacy_stream.primary_key
162159

163160
def get_parser(self) -> FileTypeParser:
164161
return self._legacy_stream.get_parser()

airbyte_cdk/sources/file_based/stream/permissions_file_based_stream.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
from airbyte_cdk.models import AirbyteLogMessage, AirbyteMessage, Level
99
from airbyte_cdk.models import Type as MessageType
10+
from airbyte_cdk.sources.file_based.config.file_based_stream_config import PrimaryKeyType
1011
from airbyte_cdk.sources.file_based.file_based_stream_permissions_reader import (
1112
AbstractFileBasedStreamPermissionsReader,
1213
)
@@ -35,6 +36,10 @@ def __init__(
3536
super().__init__(**kwargs)
3637
self.stream_permissions_reader = stream_permissions_reader
3738

39+
@property
40+
def primary_key(self) -> PrimaryKeyType:
41+
return self.config.primary_key
42+
3843
def _filter_schema_invalid_properties(
3944
self, configured_catalog_json_schema: Dict[str, Any]
4045
) -> Dict[str, Any]:

0 commit comments

Comments
 (0)