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"); }