Revert "fix: typecheck snapshot (#2655)"

This reverts commit 782fd74.

Author: Kitenite

apps/web/client/src/app/project/[id]/_components/right-panel/chat-tab/chat-messages/user-message.tsx

@@ -35,7 +35,7 @@ export const UserMessage = ({ message }: UserMessageProps) => {
     const [isRestoring, setIsRestoring] = useState(false);
 
     const textareaRef = useRef<HTMLTextAreaElement>(null);
-    const commitOid = message.content.metadata?.snapshots?.find((s) => s.type === MessageSnapshotType.GIT)?.oid;
+    const commitOid = message.content.metadata.snapshots.find((s) => s.type === MessageSnapshotType.GIT)?.oid;
 
     useEffect(() => {
         if (isEditing && textareaRef.current) {
@@ -210,7 +210,7 @@ export const UserMessage = ({ message }: UserMessageProps) => {
                 <div className="h-6 relative">
                     <div className="absolute top-1 left-0 right-0 flex flex-row justify-start items-center w-full overflow-auto pr-16">
                         <div className="flex flex-row gap-3 text-micro text-foreground-secondary">
-                            {message.content.metadata?.context?.map((context) => (
+                            {message.content.metadata.context.map((context) => (
                                 <SentContextPill key={nanoid()} context={context} />
                             ))}
                         </div>

apps/web/client/src/components/store/editor/chat/conversation.ts

@@ -153,16 +153,13 @@ export class ConversationManager {
         }
         const userMessage = message as UserChatMessage;
         const newSnapshots = [
-            ...userMessage.content.metadata?.snapshots ?? [],
+            ...userMessage.content.metadata.snapshots,
             {
                 type: MessageSnapshotType.GIT,
                 oid: commit.oid,
                 createdAt: new Date(),
             },
         ];
-        if (!userMessage.content.metadata) {
-            userMessage.content.metadata = {};
-        }
         userMessage.content.metadata.snapshots = newSnapshots;
         await api.chat.message.updateSnapshot.mutate({
             messageId: message.id,
@@ -176,7 +173,7 @@ export class ConversationManager {
             console.error('No conversation found');
             return;
         }
-        const index = this.current.messages.findIndex((m) => m.id === message.id || (m.content.metadata?.vercelId && m.content.metadata.vercelId === message.content.metadata?.vercelId));
+        const index = this.current.messages.findIndex((m) => m.id === message.id || (m.content.metadata.vercelId && m.content.metadata.vercelId === message.content.metadata.vercelId));
         if (index === -1) {
             this.current.messages.push(message);
         } else {

apps/web/client/src/components/store/editor/chat/index.ts

@@ -77,12 +77,8 @@ export class ChatManager {
         const messagesToRemove = this.conversation.current?.messages.filter((m) => m.createdAt >= oldMessage.createdAt);
 
         // Create a new message with the new content
-        const newContext = await this.context.getRefreshedContext(oldMessage.content.metadata?.context ?? []);
-        if (!oldMessage.content.metadata) {
-            oldMessage.content.metadata = {};
-        }
+        const newContext = await this.context.getRefreshedContext(oldMessage.content.metadata.context);
         oldMessage.content.metadata.context = newContext;
-        await this.conversation.addOrReplaceMessage(oldMessage);
         const newMessage = await this.conversation.addUserMessage(newMessageContent, newContext);
         if (!newMessage) {
             console.error('Failed to add user message');

packages/models/src/chat/message/message.ts

@@ -9,15 +9,12 @@ export enum ChatMessageRole {
     ASSISTANT = 'assistant',
 }
 export interface ChatMessageContent extends MastraMessageContentV2 {
-    metadata:
-        | {
-              vercelId?: string;
-              context?: ChatMessageContext[];
-              snapshots?: MessageSnapshot[];
-          }
-        | undefined;
+    metadata: {
+        vercelId?: string;
+        context: ChatMessageContext[];
+        snapshots: MessageSnapshot[];
+    };
 }
-
 interface BaseChatMessage extends MastraMessageV2 {
     role: ChatMessageRole;
     threadId: string;