(t *testing.T)
| 65 | } |
| 66 | |
| 67 | func TestBuildSession_PreservesHistory(t *testing.T) { |
| 68 | t.Parallel() |
| 69 | sess := buildSession([]ChatCompletionMessage{ |
| 70 | {Role: "system", Content: "you are a docker agent"}, |
| 71 | {Role: "user", Content: "hello"}, |
| 72 | {Role: "assistant", Content: "hi there"}, |
| 73 | {Role: "user", Content: "how are you?"}, |
| 74 | }) |
| 75 | require.NotNil(t, sess) |
| 76 | |
| 77 | // GetAllMessages omits system messages. |
| 78 | all := sess.GetAllMessages() |
| 79 | require.Len(t, all, 3) |
| 80 | |
| 81 | roles := make([]chat.MessageRole, len(all)) |
| 82 | for i, m := range all { |
| 83 | roles[i] = m.Message.Role |
| 84 | } |
| 85 | assert.Equal(t, []chat.MessageRole{ |
| 86 | chat.MessageRoleUser, |
| 87 | chat.MessageRoleAssistant, |
| 88 | chat.MessageRoleUser, |
| 89 | }, roles) |
| 90 | |
| 91 | assert.Equal(t, "how are you?", sess.GetLastUserMessageContent()) |
| 92 | assert.Equal(t, "hi there", sess.GetLastAssistantMessageContent()) |
| 93 | } |
| 94 | |
| 95 | func TestBuildSession_PreservesToolMessage(t *testing.T) { |
| 96 | t.Parallel() |
nothing calls this directly
no test coverage detected