From 6e3e2b2e835de9051a877b7fe4aacd3cb506165e Mon Sep 17 00:00:00 2001
From: t8y2 <1156263951@qq.com>
Date: Wed, 24 Jun 2026 02:15:45 +0800
Subject: [PATCH] fix(completion): import duckdb completion helper in schema
 tests

---
 crates/dbx-core/src/schema.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/crates/dbx-core/src/schema.rs b/crates/dbx-core/src/schema.rs
index 0853c8310..aa7bd5721 100644
--- a/crates/dbx-core/src/schema.rs
+++ b/crates/dbx-core/src/schema.rs
@@ -1099,7 +1099,10 @@ mod tests {
         presto_like_information_schema_tables_sql, presto_like_tables_from_query_result,
     };
     #[cfg(feature = "duckdb-bundled")]
-    use super::{duckdb_attach_database, duckdb_list_databases, duckdb_query_tables_in_database};
+    use super::{
+        duckdb_attach_database, duckdb_completion_assistant_search, duckdb_list_databases,
+        duckdb_query_tables_in_database,
+    };
     use crate::models::connection::{ConnectionConfig, DatabaseType};
 
     fn test_column(name: &str, comment: Option<&str>, is_primary_key: bool) -> super::db::ColumnInfo {