Skip to content

Commit

Permalink
Fix VertexAiMultimodalEmbeddingModelIT test by separating documents
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
Mark Pollack committed Dec 22, 2024
1 parent 92cba5d commit 3ffcfdc
Showing 1 changed file with 3 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -189,9 +189,7 @@ void videoEmbedding() {
@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")))
Expand All @@ -201,7 +199,8 @@ void textImageAndVideoEmbedding() {
.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);
Expand Down

0 comments on commit 3ffcfdc

Please sign in to comment.