I think I have a working document cache solution that's actually pretty good.

src/components/documents/scene/SceneEditForm.tsx

@@ -1,15 +1,13 @@
 import { AutoSaveTextarea } from "@/components/AutoSaveTextarea";
 import { pb } from "@/lib/pocketbase";
 import type { Scene } from "@/lib/types";
-import { useDocument } from "@/context/document/DocumentContext";
-import { useQueryClient } from "@tanstack/react-query";
+import { useDocumentCache } from "@/context/document/hooks";
 
 /**
  * Renders an editable scene form
  */
 export const SceneEditForm = ({ scene }: { scene: Scene }) => {
-  const { dispatch } = useDocument();
-  const queryClient = useQueryClient();
+  const { dispatch } = useDocumentCache();
 
   async function saveScene(text: string) {
     const updated: Scene = await pb.collection("documents").update(scene.id, {