Fix truncation of bash output (#28374)
Release Notes: - Fixed a regression that caused the bash tool to not include all of the output. --------- Co-authored-by: Agus Zubiaga <hi@aguz.me>
This commit is contained in:
parent
64cea2f1f1
commit
d0632a5332
1 changed files with 15 additions and 1 deletions
|
@ -156,7 +156,21 @@ impl Tool for BashTool {
|
|||
|
||||
// Read one more byte to determine whether the output was truncated
|
||||
let mut buffer = vec![0; LIMIT + 1];
|
||||
let bytes_read = reader.read(&mut buffer).await?;
|
||||
let mut bytes_read = 0;
|
||||
|
||||
// Read until we reach the limit
|
||||
loop {
|
||||
let read = reader.read(&mut buffer).await?;
|
||||
if read == 0 {
|
||||
break;
|
||||
}
|
||||
|
||||
bytes_read += read;
|
||||
if bytes_read > LIMIT {
|
||||
bytes_read = LIMIT + 1;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// Repeatedly fill the output reader's buffer without copying it.
|
||||
loop {
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue