chore: make tests for ChatMessage retriever and writer compatible with the new API
#986
| Job | Run time |
|---|---|
| 2m 4s | |
| 1m 13s | |
| 4m 10s | |
| 1m 12s | |
| 3m 28s | |
| 4m 59s | |
| 1m 27s | |
| 18m 33s |