insertAssistantMessage inserts an assistant message with v1 content into a chat.
( t *testing.T, db database.Store, chatID uuid.UUID, modelID uuid.UUID, text string, )
| 3076 | // insertAssistantMessage inserts an assistant message with v1 content |
| 3077 | // into a chat. |
| 3078 | func insertAssistantMessage( |
| 3079 | t *testing.T, |
| 3080 | db database.Store, |
| 3081 | chatID uuid.UUID, |
| 3082 | modelID uuid.UUID, |
| 3083 | text string, |
| 3084 | ) { |
| 3085 | t.Helper() |
| 3086 | |
| 3087 | parts := []codersdk.ChatMessagePart{codersdk.ChatMessageText(text)} |
| 3088 | data, err := json.Marshal(parts) |
| 3089 | require.NoError(t, err) |
| 3090 | |
| 3091 | _ = dbgen.ChatMessage(t, db, database.ChatMessage{ |
| 3092 | ChatID: chatID, |
| 3093 | CreatedBy: uuid.NullUUID{}, |
| 3094 | ModelConfigID: uuid.NullUUID{UUID: modelID, Valid: true}, |
| 3095 | Role: database.ChatMessageRoleAssistant, |
| 3096 | Content: pqtype.NullRawMessage{RawMessage: data, Valid: true}, |
| 3097 | ContentVersion: chatprompt.ContentVersionV1, |
| 3098 | }) |
| 3099 | } |
| 3100 | |
| 3101 | func insertLinkedChatFile( |
| 3102 | ctx context.Context, |
no test coverage detected