loop: make multiplechoice tool calls end the turn
Co-Authored-By: sketch <hello@sketch.dev>
Change-ID: s8d507faf9c095824sk
diff --git a/loop/mocks.go b/loop/mocks.go
index 014cd46..016c021 100644
--- a/loop/mocks.go
+++ b/loop/mocks.go
@@ -149,7 +149,7 @@
return exp.result[0].(*llm.Response), retErr
}
-func (m *MockConvo) ToolResultContents(ctx context.Context, resp *llm.Response) ([]llm.Content, error) {
+func (m *MockConvo) ToolResultContents(ctx context.Context, resp *llm.Response) ([]llm.Content, bool, error) {
m.recordCall("ToolResultContents", resp)
exp, ok := m.findMatchingExpectation("ToolResultContents", resp)
if !ok {
@@ -163,7 +163,7 @@
retErr = err
}
- return exp.result[0].([]llm.Content), retErr
+ return exp.result[0].([]llm.Content), false, retErr
}
func (m *MockConvo) ToolResultCancelContents(resp *llm.Response) ([]llm.Content, error) {