Use in sample taps

meltano · Sep 6, 2024 · 9e46c44 · 9e46c44
1 parent 3169b58
commit 9e46c44
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/samples/sample_tap_countries/countries_tap.py b/samples/sample_tap_countries/countries_tap.py
@@ -13,10 +13,11 @@
     CountriesStream,
 )
 from singer_sdk import Stream, Tap
+from singer_sdk._singerlib.encoding._msgspec import MsgSpecWriter  # noqa: PLC2701
 from singer_sdk.typing import PropertiesList
 
 
-class SampleTapCountries(Tap):
+class SampleTapCountries(MsgSpecWriter, Tap):
     """Sample tap for Countries GraphQL API."""
 
     name: str = "sample-tap-countries"

diff --git a/samples/sample_tap_sqlite/__init__.py b/samples/sample_tap_sqlite/__init__.py
@@ -6,6 +6,7 @@
 
 from singer_sdk import SQLConnector, SQLStream, SQLTap
 from singer_sdk import typing as th
+from singer_sdk._singerlib.encoding._msgspec import MsgSpecWriter  # noqa: PLC2701
 
 DB_PATH_CONFIG = "path_to_db"
 
@@ -39,7 +40,7 @@ class SQLiteStream(SQLStream):
     STATE_MSG_FREQUENCY = 10
 
 
-class SQLiteTap(SQLTap):
+class SQLiteTap(MsgSpecWriter, SQLTap):
     """The Tap class for SQLite."""
 
     name = "tap-sqlite-sample"