tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

Agent loop defensive (#3554)

Browse Source 
Co-authored-by: Douwe Osinga <douwe@squareup.com>
This commit is contained in:
Douwe Osinga
2025-07-22 18:21:47 +02:00
committed by GitHub
parent b3cd03ef61
commit 9f356e7009
19 changed files with 349 additions and 773 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/tests/agent.rs
+7 -5
View File
@@ -129,7 +129,7 @@ async fn run_truncate_test(
),
];
let reply_stream = agent.reply(&messages, None).await?;
let reply_stream = agent.reply(&messages, None, None).await?;
tokio::pin!(reply_stream);
let mut responses = Vec::new();
@@ -619,7 +619,7 @@ mod final_output_tool_tests {
);
// Simulate the reply stream continuing after the final output tool call.
let reply_stream = agent.reply(&vec![], None).await?;
let reply_stream = agent.reply(&vec![], None, None).await?;
tokio::pin!(reply_stream);
let mut responses = Vec::new();
@@ -716,7 +716,7 @@ mod final_output_tool_tests {
agent.add_final_output_tool(response).await;
// Simulate the reply stream being called.
let reply_stream = agent.reply(&vec![], None).await?;
let reply_stream = agent.reply(&vec![], None, None).await?;
tokio::pin!(reply_stream);
let mut responses = Vec::new();
@@ -850,7 +850,9 @@ mod retry_tests {
let initial_messages = vec![Message::user().with_text("Complete this task")];
let reply_stream = agent.reply(&initial_messages, Some(session_config)).await?;
let reply_stream = agent
.reply(&initial_messages, Some(session_config), None)
.await?;
tokio::pin!(reply_stream);
let mut responses = Vec::new();
@@ -1013,7 +1015,7 @@ mod max_turns_tests {
};
let messages = vec![Message::user().with_text("Hello")];
let reply_stream = agent.reply(&messages, Some(session_config)).await?;
let reply_stream = agent.reply(&messages, Some(session_config), None).await?;
tokio::pin!(reply_stream);
let mut responses = Vec::new();