diff --git a/ibis-server/tests/routers/v2/connector/test_gcs_file.py b/ibis-server/tests/routers/v2/connector/test_gcs_file.py index 47d53ce9a..6ca6695cb 100644 --- a/ibis-server/tests/routers/v2/connector/test_gcs_file.py +++ b/ibis-server/tests/routers/v2/connector/test_gcs_file.py @@ -77,6 +77,7 @@ "condition": "Customer.custkey = Orders.custkey", } ], + "dataSource": "gcs_file", } diff --git a/ibis-server/tests/routers/v2/connector/test_local_file.py b/ibis-server/tests/routers/v2/connector/test_local_file.py index 0f0b6c99c..33f98ac4e 100644 --- a/ibis-server/tests/routers/v2/connector/test_local_file.py +++ b/ibis-server/tests/routers/v2/connector/test_local_file.py @@ -72,6 +72,7 @@ "condition": "Customer.custkey = Orders.custkey", } ], + "dataSource": "local_file", } diff --git a/ibis-server/tests/routers/v2/connector/test_minio_file.py b/ibis-server/tests/routers/v2/connector/test_minio_file.py index a0441e76a..eafce84e8 100644 --- a/ibis-server/tests/routers/v2/connector/test_minio_file.py +++ b/ibis-server/tests/routers/v2/connector/test_minio_file.py @@ -75,6 +75,7 @@ "condition": "Customer.custkey = Orders.custkey", } ], + "dataSource": "minio_file", } diff --git a/ibis-server/tests/routers/v2/connector/test_s3_file.py b/ibis-server/tests/routers/v2/connector/test_s3_file.py index 8e855fc9b..0e5ed035c 100644 --- a/ibis-server/tests/routers/v2/connector/test_s3_file.py +++ b/ibis-server/tests/routers/v2/connector/test_s3_file.py @@ -77,6 +77,7 @@ "condition": "Customer.custkey = Orders.custkey", } ], + "dataSource": "s3_file", } diff --git a/wren-core-base/manifest-macro/src/lib.rs b/wren-core-base/manifest-macro/src/lib.rs index 0207b5a1a..65f145879 100644 --- a/wren-core-base/manifest-macro/src/lib.rs +++ b/wren-core-base/manifest-macro/src/lib.rs @@ -94,6 +94,14 @@ pub fn data_source(python_binding: proc_macro::TokenStream) -> proc_macro::Token Datafusion, #[serde(alias = "duckdb")] DuckDB, + #[serde(alias = "local_file")] + LocalFile, + #[serde(alias = "s3_file")] + S3File, + #[serde(alias = "gcs_file")] + GcsFile, + #[serde(alias = "minio_file")] + MinioFile, } }; proc_macro::TokenStream::from(expanded) diff --git a/wren-core-base/src/mdl/manifest.rs b/wren-core-base/src/mdl/manifest.rs index 5118e7097..9264ded1c 100644 --- a/wren-core-base/src/mdl/manifest.rs +++ b/wren-core-base/src/mdl/manifest.rs @@ -102,6 +102,10 @@ impl Display for DataSource { DataSource::Snowflake => write!(f, "SNOWFLAKE"), DataSource::Datafusion => write!(f, "DATAFUSION"), DataSource::DuckDB => write!(f, "DUCKDB"), + DataSource::LocalFile => write!(f, "LOCAL_FILE"), + DataSource::S3File => write!(f, "S3_FILE"), + DataSource::GcsFile => write!(f, "GCS_FILE"), + DataSource::MinioFile => write!(f, "MINIO_FILE"), } } }