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

test: fix test_max_turns_limit slow execution and wrong message type (#6221)

Browse Source 
Signed-off-by: Adrian Cole <adrian@tetrate.io>
This commit is contained in:
Adrian Cole
2025-12-22 08:06:25 +08:00
committed by GitHub
parent 5bfc29cb86
commit 02a3a1ead1
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/tests/agent.rs
+10 -8
View File
@@ -387,7 +387,7 @@ mod tests {
let session_config = SessionConfig {
id: session.id,
schedule_id: None,
max_turns: None,
max_turns: Some(1),
retry_config: None,
};
@@ -398,16 +398,18 @@ mod tests {
while let Some(response_result) = reply_stream.next().await {
match response_result {
Ok(AgentEvent::Message(response)) => {
if let Some(MessageContent::ToolConfirmationRequest(ref req)) =
if let Some(MessageContent::ActionRequired(action)) =
response.content.first()
{
agent.handle_confirmation(
req.id.clone(),
goose::permission::PermissionConfirmation {
principal_type: goose::permission::permission_confirmation::PrincipalType::Tool,
permission: goose::permission::Permission::AllowOnce,
if let goose::conversation::message::ActionRequiredData::ToolConfirmation { id, .. } = &action.data {
agent.handle_confirmation(
id.clone(),
goose::permission::PermissionConfirmation {
principal_type: goose::permission::permission_confirmation::PrincipalType::Tool,
permission: goose::permission::Permission::AllowOnce,
}
).await;
}
).await;
}
responses.push(response);
}