Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix VertexAiMultimodalEmbeddingModelIT test by separating documents

Browse Source 
The Document class requires exactly one of text or media to be specified.
Updated textImageAndVideoEmbedding test to create separate Document
instances for text, image, and video content instead of combining them
in a single document.
This commit is contained in:
Mark Pollack
2024-12-21 21:15:46 -05:00
parent 92cba5d759
commit 3ffcfdc215
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
models/spring-ai-vertex-ai-embedding/src/test/java/org/springframework/ai/vertexai/embedding/multimodal/VertexAiMultimodalEmbeddingModelIT.java
View File

@@ -189,9 +189,7 @@ class VertexAiMultimodalEmbeddingModelIT {
@Test
void textImageAndVideoEmbedding() {
var textDocument = Document.builder()
.text("Hello World")
.build();
var textDocument = Document.builder().text("Hello World").build();
var imageDocument = Document.builder()
.media(new Media(MimeTypeUtils.IMAGE_PNG, new ClassPathResource("/test.image.png")))
@@ -201,7 +199,8 @@ class VertexAiMultimodalEmbeddingModelIT {
.media(new Media(new MimeType("video", "mp4"), new ClassPathResource("/test.video.mp4")))
.build();
DocumentEmbeddingRequest embeddingRequest = new DocumentEmbeddingRequest(List.of(textDocument, imageDocument, videoDocument));
DocumentEmbeddingRequest embeddingRequest = new DocumentEmbeddingRequest(
List.of(textDocument, imageDocument, videoDocument));
EmbeddingResponse embeddingResponse = this.multiModelEmbeddingModel.call(embeddingRequest);
assertThat(embeddingResponse.getResults()).hasSize(3);