From c15e91a65b153556d7f02d057386a9f7977dddc9 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Fri, 16 May 2025 19:06:45 +0200 Subject: [PATCH] Adapt tests Signed-off-by: Ettore Di Giacinto --- pkg/templates/multimodal_test.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pkg/templates/multimodal_test.go b/pkg/templates/multimodal_test.go index ef8607a7..d0918697 100644 --- a/pkg/templates/multimodal_test.go +++ b/pkg/templates/multimodal_test.go @@ -20,7 +20,7 @@ var _ = Describe("EvaluateTemplate", func() { VideosInMessage: 0, }, "bar") Expect(err).NotTo(HaveOccurred()) - Expect(result).To(Equal("[img-0]bar")) + Expect(result).To(Equal("<__image__>bar")) }) It("should handle messages with more images correctly", func() { @@ -33,7 +33,7 @@ var _ = Describe("EvaluateTemplate", func() { VideosInMessage: 0, }, "bar") Expect(err).NotTo(HaveOccurred()) - Expect(result).To(Equal("[img-0][img-1]bar")) + Expect(result).To(Equal("<__image__><__image__>bar")) }) It("should handle messages with more images correctly", func() { result, err := TemplateMultiModal("", MultiModalOptions{ @@ -45,7 +45,7 @@ var _ = Describe("EvaluateTemplate", func() { VideosInMessage: 0, }, "bar") Expect(err).NotTo(HaveOccurred()) - Expect(result).To(Equal("[audio-0][img-2][img-3]bar")) + Expect(result).To(Equal("[audio-0]<__image__><__image__>bar")) }) It("should handle messages with more images correctly", func() { result, err := TemplateMultiModal("", MultiModalOptions{ @@ -57,7 +57,7 @@ var _ = Describe("EvaluateTemplate", func() { VideosInMessage: 0, }, "bar") Expect(err).NotTo(HaveOccurred()) - Expect(result).To(Equal("[audio-0][img-2]bar")) + Expect(result).To(Equal("[audio-0]<__image__>bar")) }) It("should handle messages with more images correctly", func() { result, err := TemplateMultiModal("", MultiModalOptions{