From c6d70b62dc773d5f6d575a25ed7c49bd32688866 Mon Sep 17 00:00:00 2001
From: dal <dallin@buster.so>
Date: Wed, 12 Feb 2025 11:07:59 -0700
Subject: [PATCH] ddl on query

---
 api/src/utils/agents/generate_sql_agent.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/api/src/utils/agents/generate_sql_agent.rs b/api/src/utils/agents/generate_sql_agent.rs
index 982d5adfd..3f2f677f8 100644
--- 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");
     }