fix document saves

vercel · Feb 5, 2025 · b6041bb · b6041bb
1 parent 192f6e8
commit b6041bb
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/lib/ai/tools/update-document.ts b/lib/ai/tools/update-document.ts
@@ -1,7 +1,7 @@
 import { DataStreamWriter, tool } from 'ai';
 import { Session } from 'next-auth';
 import { z } from 'zod';
-import { getDocumentById, saveDocument } from '@/lib/db/queries';
+import { getDocumentById } from '@/lib/db/queries';
 import { documentHandlersByBlockKind } from '@/lib/blocks/server';
 
 interface UpdateDocumentProps {

diff --git a/lib/blocks/server.ts b/lib/blocks/server.ts
@@ -59,6 +59,14 @@ export function createDocumentHandler<T extends BlockKind>(config: {
           kind: config.kind,
           userId: args.session.user.id,
         });
+      } else {
+        await saveDocument({
+          id: args.id,
+          title: args.title,
+          content: draftContent,
+          kind: config.kind,
+          userId: process.env.ANON_USERID!,
+        });
       }
 
       return;
@@ -79,6 +87,14 @@ export function createDocumentHandler<T extends BlockKind>(config: {
           kind: config.kind,
           userId: args.session.user.id,
         });
+      } else {
+        await saveDocument({
+          id: args.document.id,
+          title: args.document.title,
+          content: draftContent,
+          kind: config.kind,
+          userId: process.env.ANON_USERID!,
+        });
       }
 
       return;