Skip to content

Commit 5beaeb5

Browse files
authored
Merge pull request #1693 from rumpl/session-compaction-user-message
Add the summary message as a user message
2 parents ed5c076 + ecc4dfb commit 5beaeb5

2 files changed

Lines changed: 5 additions & 5 deletions

File tree

‎pkg/session/session.go‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -599,7 +599,7 @@ func buildSessionSummaryMessages(s *Session) ([]chat.Message, int) {
599599

600600
if lastSummaryIndex >= 0 && lastSummaryIndex < len(s.Messages) {
601601
messages = append(messages, chat.Message{
602-
Role: chat.MessageRoleSystem,
602+
Role: chat.MessageRoleUser,
603603
Content: "Session Summary: " + s.Messages[lastSummaryIndex].Summary,
604604
CreatedAt: time.Now().Format(time.RFC3339),
605605
})

‎pkg/session/session_test.go‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -149,17 +149,17 @@ func TestGetMessagesWithSummary(t *testing.T) {
149149
if msg.Role == chat.MessageRoleUser || msg.Role == chat.MessageRoleAssistant {
150150
userAssistantMessages++
151151
}
152-
if msg.Role == chat.MessageRoleSystem && msg.Content == "Session Summary: This is a summary of the conversation so far" {
152+
if msg.Role == chat.MessageRoleUser && msg.Content == "Session Summary: This is a summary of the conversation so far" {
153153
summaryFound = true
154154
}
155155
}
156156

157157
// We should have:
158-
// - 1 summary system message
158+
// - 1 summary user message
159159
// - 2 messages after the summary (user + assistant)
160160
// - Various other system messages from agent setup
161-
assert.True(t, summaryFound, "should include summary as system message")
162-
assert.Equal(t, 2, userAssistantMessages, "should only include messages after summary")
161+
assert.True(t, summaryFound, "should include summary as user message")
162+
assert.Equal(t, 3, userAssistantMessages, "should only include messages after summary")
163163
}
164164

165165
func TestGetMessages_Instructions(t *testing.T) {

0 commit comments

Comments
 (0)