Function insertAssistantCostMessage

coderd/exp_chats_test.go:264–285

(
	t *testing.T,
	db database.Store,
	chatID uuid.UUID,
	modelConfigID uuid.UUID,
	totalCostMicros int64,
)

}

func insertAssistantCostMessage(
	t *testing.T,
	db database.Store,
	chatID uuid.UUID,
	modelConfigID uuid.UUID,
	totalCostMicros int64,
) {
	t.Helper()

	assistantContent, err := chatprompt.MarshalParts([]codersdk.ChatMessagePart{
		codersdk.ChatMessageText("assistant"),
	})
	require.NoError(t, err)

	_ = dbgen.ChatMessage(t, db, database.ChatMessage{
		ChatID:          chatID,
		ModelConfigID:   uuid.NullUUID{UUID: modelConfigID, Valid: true},
		Role:            database.ChatMessageRoleAssistant,
		Content:         assistantContent,
		TotalCostMicros: sql.NullInt64{Int64: totalCostMicros, Valid: true},
	})
}

func TestPostChats(t *testing.T) {
	t.Parallel()