typescript - Typování metody `set` ze Zustand store s middleware Immer při zachování automatického dokončování

Mám problém s typováním metody set z uložiště Zustand při použití middleware Immer. Potřebuji ji typovat, protože některé vlastnosti obsahují funkce, které budou aktualizovat stav, takže potřebují funkci set:

export const useStore = create<Store>()(
     immer((set, get) => ({
         func: beginDraggingVertex(set, get, vertexID) 
     }))
)     

....

function beginDraggingVertex(set: Setter, get: Getter, vertexID: string) {
  ...ommited
  set((store) => 
   // here is where I do want to have the autocomplete working
   store.whateverProperty = 'hi'
  )

}

Pokud odstraním mezivrstvu Immer ze store, když najedu myší na metodu set, opravdu vidím její typ, takže mohu později zadat typy u deklarace beginDraggingVertex argumentů. Ale s aplikovaným Immer se metoda set změní a pokud změním typ set na nový typ, ztratím automatické dokončování.

 //`set` type with Immer applied, with this one I loose autocomplete wihtin:
 type Setter = (
  nextStateOrUpdater:
    | (Store & ActionsObject)
    | Partial<Store & ActionsObject>
    | ((state: WritableDraft<Store & ActionsObject>) => void),
  shouldReplace?: boolean | undefined
) => void

//`set` type without Immer:
 export type Setter = (
  partial:
    | (Store & ActionsObject)
    | Partial<Store & ActionsObject>
    | ((state: Store & ActionsObject) => (Store & ActionsObject) | Partial<Store & ActionsObject>),
  replace?: boolean | undefined
) => void

Pokud použiji Immer a potom dám set starý typ z doby, kdy nebyl aplikován, samozřejmě získám automatické doplňování, ale dostávám tyto chyby i v callbacku set:

I was expecting (Store & ActionsObject) | Partial<Store & ActionsObject> | ((state: Store & ActionsObject) => (Store & ActionsObject) | Partial<...>), but you passed (store: Store & ActionsObject) => void

Opravdu si vážím jakékoliv pomoci, kterou byste mohl poskytnout.