fixes

Author: YousefED

agent-demo/app/document.tsx

@@ -5,6 +5,7 @@ import {
   getMessageWithToolCallId,
   useChatContext,
 } from "@/components/ChatContext";
+import { Chat } from "@ai-sdk/react";
 import {
   ActionBarPrimitive,
   makeAssistantTool,
@@ -50,6 +51,9 @@ const UndoActionBar = () => {
       <TooltipIconButton
         tooltip="Revert to before this change"
         className="aui-user-action-edit"
+        onClick={() => {
+          window.alert("Not implemented");
+        }}
       >
         <UndoIcon className="size-4" />
       </TooltipIconButton>

agent-demo/components/BlockNoteChat.tsx

@@ -1,13 +1,12 @@
 import { Chat } from "@ai-sdk/react";
 import { BlockNoteEditor } from "@blocknote/core";
-import { ChatInit, UIMessage } from "ai";
-
 import {
   aiDocumentFormats,
   AIRequest,
   buildAIRequest,
   sendMessageWithAIRequest,
 } from "@blocknote/xl-ai";
+import { ChatInit, UIMessage } from "ai";
 
 // maybe better to use a custom transport for this?
 class BlockNoteAISDKChatRaw<
@@ -44,7 +43,7 @@ class BlockNoteAISDKChatRaw<
           aiDocumentFormats.html.defaultDocumentStateBuilder,
       }));
 
-    return sendMessageWithAIRequest(
+    await sendMessageWithAIRequest(
       this as Chat<any>,
       aiRequest,
       message,

agent-demo/components/DocumentList.tsx

@@ -21,7 +21,12 @@ export const DocumentList: FC = () => {
 
 const DocumentListNew: FC = () => {
   return (
-    <ThreadListPrimitive.New asChild>
+    <ThreadListPrimitive.New
+      asChild
+      onClick={() => {
+        window.alert("Not implemented");
+      }}
+    >
       <Button
         className="aui-thread-list-new flex items-center justify-start gap-1 rounded-lg px-2.5 py-2 text-start hover:bg-muted data-active:bg-muted"
         variant="ghost"
@@ -79,7 +84,7 @@ const DocumentListItem: FC = () => {
 const DocumentListItemTitle: FC = () => {
   return (
     <span className="aui-thread-list-item-title text-sm">
-      <ThreadListItemPrimitive.Title fallback="New Chat" />
+      <ThreadListItemPrimitive.Title fallback="Sample Document" />
     </span>
   );
 };

agent-demo/components/assistant-ui/thread-list.tsx

@@ -1,13 +1,13 @@
-import type { FC } from "react";
 import {
   ThreadListItemPrimitive,
   ThreadListPrimitive,
   useAssistantState,
 } from "@assistant-ui/react";
 import { ArchiveIcon, PlusIcon } from "lucide-react";
+import type { FC } from "react";
 
-import { Button } from "@/components/ui/button";
 import { TooltipIconButton } from "@/components/assistant-ui/tooltip-icon-button";
+import { Button } from "@/components/ui/button";
 import { Skeleton } from "@/components/ui/skeleton";
 
 export const ThreadList: FC = () => {
@@ -75,7 +75,7 @@ const ThreadListItem: FC = () => {
 const ThreadListItemTitle: FC = () => {
   return (
     <span className="aui-thread-list-item-title text-sm">
-      <ThreadListItemPrimitive.Title fallback="New Chat" />
+      <ThreadListItemPrimitive.Title fallback="Sample Document" />
     </span>
   );
 };

packages/core/src/extensions/tiptap-extensions/Suggestions/SuggestionMarks.ts

@@ -12,7 +12,7 @@ export const SuggestionAddMark = Mark.create({
   excludes: "deletion modification insertion",
   addAttributes() {
     return {
-      id: { default: null, validate: "number" }, // note: validate is supported in prosemirror but not in tiptap, so this doesn't actually work (considered not critical)
+      id: { default: null, validate: "number|null" }, // note: validate is supported in prosemirror but not in tiptap, so this doesn't actually work (considered not critical)
     };
   },
   extendMarkSchema(extension) {
@@ -57,7 +57,7 @@ export const SuggestionDeleteMark = Mark.create({
   excludes: "insertion modification deletion",
   addAttributes() {
     return {
-      id: { default: null, validate: "number" }, // note: validate is supported in prosemirror but not in tiptap
+      id: { default: null, validate: "number|null" }, // note: validate is supported in prosemirror but not in tiptap
     };
   },
   extendMarkSchema(extension) {