Ling0925
/
buster
mirror of https://github.com/buster-so/buster.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

return records no matter what

This commit is contained in:
dal 2025-04-02 16:33:13 -06:00
parent 90a11bc175
commit 33eab20edb
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/libs/agents/src/tools/categories/file_tools/common.rs
View File

@ -48,7 +48,7 @@ pub async fn validate_sql(
};
// Try to execute the query using query_engine
let results = match query_engine(&data_source_id, sql, None).await {
let results = match query_engine(&data_source_id, sql, Some(15)).await {
Ok(results) => results,
Err(e) => return Err(anyhow!("SQL validation failed: {}", e)),
};
@ -58,15 +58,17 @@ pub async fn validate_sql(
// Create appropriate message based on number of records
let message = if num_records == 0 {
"No records were found".to_string()
} else if num_records > 13 {
format!("{} records were returned (showing first 13)", num_records)
} else {
format!("{} records were returned", num_records)
};
// Return records only if there are 13 or fewer
// Return at most 13 records
let return_records = if num_records <= 13 {
results
} else {
Vec::new() // Empty vec when more than 13 records
results.into_iter().take(13).collect() // Take first 13 records when more than 13
};
Ok((message, return_records))
@ -1060,7 +1062,7 @@ pub struct FileModificationBatch<T> {
pub modification_results: Vec<ModificationResult>,
}
#[derive(Debug, Serialize)]
#[derive(Debug, Serialize, Deserialize)]
pub struct ModificationResult {
pub file_id: Uuid,
pub file_name: String,