diff --git a/internal/agent/grouping.go b/internal/agent/grouping.go index 32892c3..72e3ae5 100644 --- a/internal/agent/grouping.go +++ b/internal/agent/grouping.go @@ -64,11 +64,11 @@ func callGroupingLLM(ctx context.Context, diffs []model.Diff, client llm.LLMClie } }() - table := buildFileMetadataTable(diffs) + fileList := buildFileList(diffs) messages := make([]llm.Message, 0, len(task.Messages)) for _, m := range task.Messages { - content := strings.ReplaceAll(m.Content, "{{file_metadata_table}}", table) + content := strings.ReplaceAll(m.Content, "{{file_list}}", fileList) messages = append(messages, llm.NewTextMessage(m.Role, content)) } @@ -92,7 +92,7 @@ func callGroupingLLM(ctx context.Context, diffs []model.Diff, client llm.LLMClie return groups, usage, err } -func buildFileMetadataTable(diffs []model.Diff) string { +func buildFileList(diffs []model.Diff) string { var sb strings.Builder for _, d := range diffs { status := "MODIFIED" diff --git a/internal/agent/grouping_test.go b/internal/agent/grouping_test.go index 4e0f5af..ee559f8 100644 --- a/internal/agent/grouping_test.go +++ b/internal/agent/grouping_test.go @@ -236,7 +236,7 @@ func TestGroupDiffs_LLMError_Fallback(t *testing.T) { client := &fakeGroupingClient{err: fmt.Errorf("connection refused")} tpl := template.Template{ GroupingTask: &template.LlmConversation{ - Messages: []template.ChatMessage{{Role: "user", Content: "{{file_metadata_table}}"}}, + Messages: []template.ChatMessage{{Role: "user", Content: "{{file_list}}"}}, }, } result := groupDiffs(context.Background(), diffs, client, "fake", tpl, 0) @@ -252,7 +252,7 @@ func TestGroupDiffs_LLMSuccess(t *testing.T) { } tpl := template.Template{ GroupingTask: &template.LlmConversation{ - Messages: []template.ChatMessage{{Role: "user", Content: "{{file_metadata_table}}"}}, + Messages: []template.ChatMessage{{Role: "user", Content: "{{file_list}}"}}, }, } result := groupDiffs(context.Background(), diffs, client, "fake", tpl, 0) @@ -271,7 +271,7 @@ func TestCallGroupingLLM_EmptyResponse(t *testing.T) { diffs := []model.Diff{{NewPath: "a.go"}} client := &fakeGroupingClient{response: ""} task := &template.LlmConversation{ - Messages: []template.ChatMessage{{Role: "user", Content: "{{file_metadata_table}}"}}, + Messages: []template.ChatMessage{{Role: "user", Content: "{{file_list}}"}}, } _, _, err := callGroupingLLM(context.Background(), diffs, client, "fake", task) if err == nil { @@ -284,7 +284,7 @@ func TestBuildFileMetadataTable(t *testing.T) { {NewPath: "a.go", IsNew: true, Insertions: 10, Deletions: 0}, {NewPath: "b.go", IsDeleted: true, Insertions: 0, Deletions: 5}, } - table := buildFileMetadataTable(diffs) + table := buildFileList(diffs) if !contains(table, "ADDED") || !contains(table, "DELETED") { t.Errorf("table missing status:\n%s", table) } diff --git a/internal/config/template/prompts/grouping_task_user.md b/internal/config/template/prompts/grouping_task_user.md index 591b8ed..ff257ac 100644 --- a/internal/config/template/prompts/grouping_task_user.md +++ b/internal/config/template/prompts/grouping_task_user.md @@ -1,6 +1,6 @@ Group the following changed files: -{{file_metadata_table}} +{{file_list}} Respond with a JSON array: [{"label": "short theme description", "files": ["path1", "path2"]}] \ No newline at end of file