fix return type

airbytehq · Dec 10, 2024 · ec5398b · ec5398b
1 parent 1bdcab1
commit ec5398b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/airbyte/destinations/_translate_cache_to_dest.py b/airbyte/destinations/_translate_cache_to_dest.py
@@ -32,12 +32,14 @@
 SNOWFLAKE_PASSWORD_SECRET_NAME = "SNOWFLAKE_PASSWORD"
 
 
-DestinationConfiguration = Any
+DestinationConfiguration = (
+    DestinationBigquery | DestinationDuckdb | DestinationPostgres | DestinationSnowflake
+)
 
 
 def cache_to_destination_configuration(
     cache: CacheBase,
-) -> dict[str, str]:
+) -> DestinationConfiguration:
     """Get the destination configuration from the cache."""
     conversion_fn_map: dict[str, Callable[[Any], DestinationConfiguration]] = {
         "BigQueryCache": bigquery_cache_to_destination_configuration,