RichEditor: Dynamic mergeTags based on other fields value

return $schema
            ->components([
                Section::make('Template Details')
                    ->components([
                        Group::make([
                            TextInput::make('name')
                                ->required(),
                            TextInput::make('subject')
                                ->required(),
                            TextInput::make('description'),
                        ]),
                        ToggleButtons::make('entities')
                            ->live()
                            ->multiple()
                            ->options(EntityEnum::class),
                    ])
                    ->compact()
                    ->columns(2)
                    ->columnSpanFull(),
                RichEditor::make('body')
                    ->toolbarButtons(['mergeTags'])
                    ->mergeTags(function (Get $get): array {
                        $entities = $get('entities') ?? [];
                        $variables = [];
                        foreach ($entities as $entity) {
                            if ($entity->value === EntityEnum::Invoice->value) {
                                $variables = [
                                    'invoice.number' => 'Invoice Number',
                                    'invoice.date' => 'Invoice Date',
                                ];
                            }
                        }

                        return $variables;
                    })
                    ->required()
                    ->columnSpanFull(),
            ]);