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

completed on files

This commit is contained in:
dal 2025-03-05 16:02:01 -07:00
parent b839e70aa9
commit 518c802fb6
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/libs/handlers/src/chats/post_chat_handler.rs
View File

@ -1168,11 +1168,14 @@ fn transform_assistant_tool_message(
let mut completed_content = file_content.clone(); let mut completed_content = file_content.clone();
completed_content.file.text = Some(complete_text); completed_content.file.text = Some(complete_text);
completed_content.file.text_chunk = None; completed_content.file.text_chunk = None;
// Ensure each file in the collection is marked as completed
completed_content.status = "completed".to_string();
updated_files.insert(file_id.clone(), completed_content); updated_files.insert(file_id.clone(), completed_content);
tracker.clear_chunk(chunk_id); tracker.clear_chunk(chunk_id);
} }
// Mark the overall file container as completed
file.status = "completed".to_string(); file.status = "completed".to_string();
file.files = updated_files; file.files = updated_files;
Some(BusterReasoningMessage::File(file)) Some(BusterReasoningMessage::File(file))
@ -1182,8 +1185,6 @@ fn transform_assistant_tool_message(
let mut updated_files = std::collections::HashMap::new(); let mut updated_files = std::collections::HashMap::new();
for (file_id, file_content) in file.files.iter() { for (file_id, file_content) in file.files.iter() {
// Use the same temporary ID that was generated in the streaming parser
// This ensures consistency with the IDs generated earlier
let chunk_id = format!("{}_{}", file.id, file_content.file_name); let chunk_id = format!("{}_{}", file.id, file_content.file_name);
if let Some(chunk) = &file_content.file.text_chunk { if let Some(chunk) = &file_content.file.text_chunk {
@ -1193,8 +1194,8 @@ fn transform_assistant_tool_message(
let mut updated_content = file_content.clone(); let mut updated_content = file_content.clone();
updated_content.file.text_chunk = Some(delta); updated_content.file.text_chunk = Some(delta);
updated_content.file.text = None; updated_content.file.text = None;
// Keep the original ID which should already be the temp ID
updated_content.id = file_id.clone(); updated_content.id = file_id.clone();
updated_content.status = "loading".to_string();
updated_files.insert(file_id.clone(), updated_content); updated_files.insert(file_id.clone(), updated_content);
has_updates = true; has_updates = true;
} }