diff --git a/src/db/index.ts b/src/db/index.ts
index 68b3b02..e4c136c 100644
--- a/src/db/index.ts
+++ b/src/db/index.ts
@@ -301,7 +301,8 @@ export const formatToMessage = (messages: MessageHistory): MessageType[] => {
       message: message.content,
       name: message.name,
       sources: message?.sources || [],
-      images: message.images || []
+      images: message.images || [],
+      generationInfo: message?.generationInfo,
     }
   })
 }