ddl on query

2025-02-12 11:07:59 -07:00 · 2025-02-12 11:07:59 -07:00 · c6d70b62dc
parent 224cbe4d84
commit c6d70b62dc
1 changed files with 3 additions and 2 deletions
--- a/api/src/utils/agents/generate_sql_agent.rs
+++ b/api/src/utils/agents/generate_sql_agent.rs
@ -727,8 +727,9 @@ fn create_dataset_selector_messages(

    for dataset in datasets {
        dataset_schemas.push_str(&format!(
-            "{}",
-            dataset.dataset.yml_file.clone().unwrap_or(dataset.dataset_ddl.clone())
+            "{}\n{}",
+            dataset.dataset.yml_file.clone().unwrap_or("".to_string()),
+            dataset.dataset_ddl.clone(),
        ));
        dataset_schemas.push_str("\n\n");
    }