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

src/components/documents/location/LocationEditForm.tsx

@@ -1,30 +1,34 @@
 import { AutoSaveTextarea } from "@/components/AutoSaveTextarea";
 import { pb } from "@/lib/pocketbase";
 import type { Location } from "@/lib/types";
-import { useDocument } from "@/context/document/DocumentContext";
+import { useDocumentCache } from "@/context/document/hooks";
 
 /**
  * Renders an editable location form
  */
 export const LocationEditForm = ({ location }: { location: Location }) => {
-  const { dispatch } = useDocument();
+  const { dispatch } = useDocumentCache();
   async function saveLocationName(name: string) {
-    const updated: Location = await pb.collection("documents").update(location.id, {
-      data: {
-        ...location.data,
-        name,
-      },
-    });
+    const updated: Location = await pb
+      .collection("documents")
+      .update(location.id, {
+        data: {
+          ...location.data,
+          name,
+        },
+      });
     dispatch({ type: "setDocument", doc: updated });
   }
 
   async function saveLocationDescription(description: string) {
-    const updated: Location = await pb.collection("documents").update(location.id, {
-      data: {
-        ...location.data,
-        description,
-      },
-    });
+    const updated: Location = await pb
+      .collection("documents")
+      .update(location.id, {
+        data: {
+          ...location.data,
+          description,
+        },
+      });
     dispatch({ type: "setDocument", doc: updated });
   }