types: improve types in atom function

StyleShit · Aug 22, 2024 · c04b01c · c04b01c
1 parent 876fa5e
commit c04b01c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/.changeset/polite-terms-dress.md b/.changeset/polite-terms-dress.md
@@ -0,0 +1,5 @@
+---
+'atomsphere': patch
+---
+
+Improve types in `atom` function
diff --git a/src/atom.ts b/src/atom.ts
@@ -11,7 +11,9 @@ type DeriveFn<T> = (get: <P>(atom: ReadableAtom<P>) => P) => T;
 
 export function atom<T>(initialValue: DeriveFn<T>): ReadableAtom<T>;
 export function atom<T>(initialValue: T): WritableAtom<T>;
-export function atom<T>(initialValue: T | DeriveFn<T>) {
+export function atom<T>(
+	initialValue: T | DeriveFn<T>,
+): WritableAtom<T> | ReadableAtom<T> {
 	return typeof initialValue === 'function'
 		? createDerivedAtom(initialValue as DeriveFn<T>)
 		: createAtom(initialValue);