refactor: replace withWorkStore with createWorkStore and run (#72159)
This pull request refactors how work stores are created and used across
various modules in the codebase, replacing the `withWorkStore` function
with a new `createWorkStore` function. This change simplifies the code
and improves readability by separating the creation and usage of work
stores and aligns it with existing stores.