feat: implement message type/name persistence, rehydration and calendar date fixes

app/actions.tsx

@@ -661,7 +661,7 @@ export const AI = createAI<AIState, UIState>({
       userId: actualUserId,
       path,
       title,
-      messages: updatedMessages
+      messages: updatedMessages.filter(msg => !(msg.content === 'end' && msg.type === 'end'))
     }
     await saveChat(chat, actualUserId)
   }
@@ -710,6 +710,8 @@ export const getUIStateFromAIState = (aiState: AIState): UIState => {
                 id,
                 component: <CopilotDisplay content={content as string} />
               }
+            default:
+              return null
           }
           break
         case 'assistant':
@@ -770,6 +772,8 @@ export const getUIStateFromAIState = (aiState: AIState): UIState => {
                 )
               }
             }
+            default:
+              return null
           }
           break
         case 'tool':
@@ -868,5 +872,5 @@ export const getUIStateFromAIState = (aiState: AIState): UIState => {
           }
       }
     })
-    .filter(message => message !== null) as UIState
+    .filter(message => message !== null && message !== undefined) as UIState
 }

app/search/[id]/page.tsx

@@ -18,7 +18,7 @@ export async function generateMetadata({ params }: SearchPageProps) {
   // TODO: Metadata generation might need authenticated user if chats are private
   // For now, assuming getChat can be called or it handles anon access for metadata appropriately
   const userId = await getCurrentUserIdOnServer(); // Attempt to get user for metadata
-  const chat = await getChat(id, userId || 'anonymous'); // Pass userId or 'anonymous' if none
+  const chat = await getChat(id, userId || undefined); // Pass userId or undefined if none (allows public fallback)
   return {
     title: chat?.title?.toString().slice(0, 50) || 'Search',
   };
@@ -48,14 +48,11 @@ export default async function SearchPage({ params }: SearchPageProps) {
   const initialMessages: AIMessage[] = dbMessages.map((dbMsg): AIMessage => {
     return {
       id: dbMsg.id,
-      role: dbMsg.role as AIMessage['role'], // Cast role, ensure AIMessage['role'] includes all dbMsg.role possibilities
+      role: dbMsg.role as AIMessage['role'],
       content: dbMsg.content,
       createdAt: dbMsg.createdAt ? new Date(dbMsg.createdAt) : undefined,
-      // 'type' and 'name' are not in the basic Drizzle 'messages' schema.
-      // These would be undefined unless specific logic is added to derive them.
-      // For instance, if a message with role 'tool' should have a 'name',
-      // or if some messages have a specific 'type' based on content or other flags.
-      // This mapping assumes standard user/assistant messages primarily.
+      type: dbMsg.type ? (dbMsg.type as AIMessage['type']) : undefined,
+      name: dbMsg.name || undefined,
     };
   });
 

drizzle/migrations/0009_add_message_type_name.sql

@@ -0,0 +1,3 @@
+ALTER TABLE "messages" ADD COLUMN "type" text;
+--> statement-breakpoint
+ALTER TABLE "messages" ADD COLUMN "name" text;