SurfSense/surfsense_web/atoms/snapshots/snapshots-query.atoms.ts

import { atomWithQuery } from "jotai-tanstack-query";
import { activeSearchSpaceIdAtom } from "@/atoms/search-spaces/search-space-query.atoms";
import { chatThreadsApiService } from "@/lib/apis/chat-threads-api.service";
import { cacheKeys } from "@/lib/query-client/cache-keys";

export const searchSpaceSnapshotsAtom = atomWithQuery((get) => {
	const searchSpaceId = get(activeSearchSpaceIdAtom);

	return {
		queryKey: cacheKeys.snapshots.bySearchSpace(Number(searchSpaceId) || 0),
		enabled: !!searchSpaceId,
		staleTime: 5 * 60 * 1000,
		queryFn: async () => {
			if (!searchSpaceId) {
				return { snapshots: [] };
			}
			return chatThreadsApiService.listSearchSpaceSnapshots({
				search_space_id: Number(searchSpaceId),
			});
		},
	};
});