diff --git a/internal/llmloop/loop_execute_more_test.go b/internal/llmloop/loop_execute_more_test.go index a037887..e321403 100644 --- a/internal/llmloop/loop_execute_more_test.go +++ b/internal/llmloop/loop_execute_more_test.go @@ -61,7 +61,7 @@ func codeCommentResponse(reasoning, content string) *llm.ChatResponse { Type: "function", Function: llm.FunctionCall{ Name: tool.CodeComment.Name(), - Arguments: `{"comments":[{"content":"issue","existing_code":"x","path":"file.go"}]}`, + Arguments: `{"comments":[{"content":"issue","existing_code":"x"}]}`, }, }}, }}}, @@ -281,8 +281,8 @@ func TestExecuteToolCall_CodeCommentThinkingBackfill(t *testing.T) { Function: llm.FunctionCall{ Name: tool.CodeComment.Name(), Arguments: `{"comments":[` + - `{"content":"a","existing_code":"x","path":"file.go"},` + - `{"content":"b","existing_code":"y","thinking":"explicit","path":"file.go"}]}`, + `{"content":"a","existing_code":"x"},` + + `{"content":"b","existing_code":"y","thinking":"explicit"}]}`, }, }, nil, "turn reasoning") @@ -314,7 +314,7 @@ func TestExecuteToolCall_CodeCommentNoReasoning(t *testing.T) { cp := r.executeToolCall(context.Background(), "file.go", llm.ToolCall{ Function: llm.FunctionCall{ Name: tool.CodeComment.Name(), - Arguments: `{"comments":[{"content":"a","existing_code":"x","path":"file.go"}]}`, + Arguments: `{"comments":[{"content":"a","existing_code":"x"}]}`, }, }, nil, "")