From f922195d36c5cdec641c3471337cc70b161b128b Mon Sep 17 00:00:00 2001
From: Mark Pollack <mark.pollack@broadcom.com>
Date: Tue, 3 Dec 2024 09:37:28 -0500
Subject: [PATCH] fix document builder tests

---
 .../src/main/java/org/springframework/ai/document/Document.java  | 1 +
 1 file changed, 1 insertion(+)

diff --git a/spring-ai-core/src/main/java/org/springframework/ai/document/Document.java b/spring-ai-core/src/main/java/org/springframework/ai/document/Document.java
index 313b81022..df294c0b0 100644
--- a/spring-ai-core/src/main/java/org/springframework/ai/document/Document.java
+++ b/spring-ai-core/src/main/java/org/springframework/ai/document/Document.java
@@ -291,6 +291,7 @@ public class Document implements MediaContent {
 		}
 
 		public Builder media(List<Media> media) {
+			Assert.notNull(media, "media cannot be null");
 			this.media.addAll(media);
 			return this;
 		}