diff --git a/custom_components/reflex_ag_grid/wrapper.py b/custom_components/reflex_ag_grid/wrapper.py index 7ec1f05..df19267 100644 --- a/custom_components/reflex_ag_grid/wrapper.py +++ b/custom_components/reflex_ag_grid/wrapper.py @@ -417,10 +417,19 @@ def _delete_button(cls) -> rx.Component: on_click=cls.delete_selected, ) + @classmethod + def _refresh_button(cls) -> rx.Component: + """Create the refresh button.""" + return rx.icon_button( + "refresh-cw", + on_click=cls._grid_component.api.refreshInfiniteCache, + ) + @classmethod def _top_toolbar(cls) -> rx.Component: """Create the top toolbar.""" return rx.hstack( + cls._refresh_button(), cls._delete_button(), cls._add_dialog(), justify="end",