loop: make multiplechoice tool calls end the turn
Co-Authored-By: sketch <hello@sketch.dev>
Change-ID: s8d507faf9c095824sk
diff --git a/loop/agent_user_cancel_test.go b/loop/agent_user_cancel_test.go
index ed7df1b..32e6241 100644
--- a/loop/agent_user_cancel_test.go
+++ b/loop/agent_user_cancel_test.go
@@ -104,7 +104,7 @@
// Set up the mock response for tool results
mockConvo.ExpectCall("SendMessage", userMsg).Return(userMsgResponse, nil)
- mockConvo.ExpectCall("ToolResultContents", userMsgResponse).Return(toolUseContents, nil)
+ mockConvo.ExpectCall("ToolResultContents", userMsgResponse).Return(toolUseContents, false, nil)
mockConvo.ExpectCall("SendMessage", toolUseResultsMsg).Return(toolUseResponse, nil)
ctx, cancel := context.WithCancel(context.Background())
@@ -458,7 +458,7 @@
defer cancel()
// Setting up the mock response for tool results
- mockConvo.ExpectCall("ToolResultContents", initialResponse).Return(toolUseContents, nil)
+ mockConvo.ExpectCall("ToolResultContents", initialResponse).Return(toolUseContents, false, nil)
mockConvo.ExpectCall("SendMessage", nil).Return(toolUseResponse, nil)
// mockConvo, as a mock, isn't able to run the loop in conversation.Convo that makes this agent.OnToolResult callback.
// So we "mock" it out here by calling it explicitly, in order to make sure it calls .pushToOutbox with this message.