diff --git a/api/libs/agents/src/tools/categories/file_tools/search_data_catalog.rs b/api/libs/agents/src/tools/categories/file_tools/search_data_catalog.rs
index 463e9aa9c..7c075a607 100644
--- a/api/libs/agents/src/tools/categories/file_tools/search_data_catalog.rs
+++ b/api/libs/agents/src/tools/categories/file_tools/search_data_catalog.rs
@@ -1364,7 +1364,7 @@ async fn search_and_update_yml(
                         let dimension_result = stored_values::search::search_values_by_embedding_with_filters(
                             data_source_id_clone,
                             &embedding_clone,
-                            10,
+                            20,
                             Some(database_name),
                             Some(schema_name),
                             Some(&dimension_clone.model_name),