From 94d7b0a4a3437782b782dbeaa76387e4b08511b1 Mon Sep 17 00:00:00 2001 From: Carlos Villela Date: Sat, 13 Jun 2026 21:09:08 -0700 Subject: [PATCH] test(e2e): accept multiturn Kimi tool calls Signed-off-by: Carlos Villela --- test/e2e/test-kimi-inference-compat.sh | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/test/e2e/test-kimi-inference-compat.sh b/test/e2e/test-kimi-inference-compat.sh index d547e7545f..69726c7286 100755 --- a/test/e2e/test-kimi-inference-compat.sh +++ b/test/e2e/test-kimi-inference-compat.sh @@ -678,7 +678,9 @@ assistant_tool_messages = [ and item.get("message", {}).get("role") == "assistant" and any(block.get("type") == "toolCall" for block in item.get("message", {}).get("content", [])) ] -source_calls = assistant_tool_messages[-1].get("content", []) if assistant_tool_messages else [] +source_calls = [] +for message in assistant_tool_messages: + source_calls.extend(message.get("content", [])) source_commands = [block.get("arguments", {}).get("command") for block in source_calls] messages = [item.get("message", {}) for item in session if item.get("type") == "message"] tool_result_indices = [idx for idx, msg in enumerate(messages) if msg.get("role") == "toolResult"]