diff --git a/src/lib/conversation-schema/message/AssistantMessageSchema.ts b/src/lib/conversation-schema/message/AssistantMessageSchema.ts
index 39105a1..847bde4 100644
--- a/src/lib/conversation-schema/message/AssistantMessageSchema.ts
+++ b/src/lib/conversation-schema/message/AssistantMessageSchema.ts
@@ -18,6 +18,7 @@ export type AssistantMessageContent = z.infer<
 export const AssistantMessageSchema = z
   .object({
     id: z.string(),
+    container: z.null().optional(),
     type: z.literal("message"),
     role: z.literal("assistant"),
     model: z.string(),