atomWithImmer not resolving the types hint properly

[khanakia]

Summary

I have this code and when i mouse hover to immerSetup.screen in vscode it shows any

How do I make it typesafe ?

type Setup = {
  screen: string;
};

const defaultSetup = {
  screen: "step1",
};

// Original atom.
const setupImmerAtom = atomWithImmer<Setup>(defaultSetup);

const setScreen = (screen: Screen) => {
  const [setupAtom, setAtomSetup] = useAtom(setupImmerAtom);
  setAtomSetup((immerSetup) => {
    immerSetup.screen = screen;
    return immerSetup;
  });
};

On debugging further, I found out this

// actual type definition /node_modules/jotai-immer/dist/atomWithImmer.d.ts
export declare function atomWithImmer<Value>(initialValue: Value): WritableAtom<Value, [Value | ((draft: Draft<Value>) => void)], void>;

// resolving as this
atomWithImmer<Setup>(initialValue: Setup): WritableAtom<Setup, [Setup | ((draft: Draft<Value>) => void)], void>

// IT should resolve as this `Draft<Value>` to `Draft<Setup>`
atomWithImmer<Setup>(initialValue: Setup): WritableAtom<Setup, [Setup | ((draft: Draft<Setup>) => void)], void>

By setting the generic type explicitly in atomWithImmer<Setup>, TypeScript should correctly infer Draft<Setup> instead of Draft<Value>.

Link to reproduction

No need just a typescript hint error

Check List

Please do not ask questions in issues.

• I've already opened a discussion before opening this issue, or already discussed in other media.

Please include a minimal reproduction.

• I've added a link to a typescript playground or codesandbox with a minimal reproduction.