Skip to content

Commit

Permalink
Docs: Expose config classes in cache modules (#315)
Browse files Browse the repository at this point in the history
  • Loading branch information
aaronsteers authored Jul 30, 2024
1 parent d42cb24 commit aee6273
Show file tree
Hide file tree
Showing 5 changed files with 35 additions and 0 deletions.
7 changes: 7 additions & 0 deletions airbyte/caches/bigquery.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,3 +47,10 @@ def get_arrow_dataset(
"BigQuery doesn't currently support to_arrow"
"Please consider using a different cache implementation for these functionalities."
)


# Expose the Cache class and also the Config class.
__all__ = [
"BigQueryCache",
"BigQueryConfig",
]
7 changes: 7 additions & 0 deletions airbyte/caches/duckdb.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,3 +39,10 @@ class DuckDBCache(DuckDBConfig, CacheBase):
"""A DuckDB cache."""

_sql_processor_class: type[DuckDBSqlProcessor] = PrivateAttr(default=DuckDBSqlProcessor)


# Expose the Cache class and also the Config class.
__all__ = [
"DuckDBCache",
"DuckDBConfig",
]
7 changes: 7 additions & 0 deletions airbyte/caches/motherduck.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,3 +61,10 @@ class MotherDuckCache(MotherDuckConfig, DuckDBCache):
"""Cache that uses MotherDuck for external persistent storage."""

_sql_processor_class: type[MotherDuckSqlProcessor] = PrivateAttr(default=MotherDuckSqlProcessor)


# Expose the Cache class and also the Config class.
__all__ = [
"MotherDuckCache",
"MotherDuckConfig",
]
7 changes: 7 additions & 0 deletions airbyte/caches/postgres.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,3 +32,10 @@ class PostgresCache(PostgresConfig, CacheBase):
"""

_sql_processor_class = PrivateAttr(default=PostgresSqlProcessor)


# Expose the Cache class and also the Config class.
__all__ = [
"PostgresCache",
"PostgresConfig",
]
7 changes: 7 additions & 0 deletions airbyte/caches/snowflake.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,3 +34,10 @@ class SnowflakeCache(SnowflakeConfig, CacheBase):
dedupe_mode: RecordDedupeMode = RecordDedupeMode.APPEND

_sql_processor_class = PrivateAttr(default=SnowflakeSqlProcessor)


# Expose the Cache class and also the Config class.
__all__ = [
"SnowflakeCache",
"SnowflakeConfig",
]

0 comments on commit aee6273

Please sign in to comment.