dearwolf/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-05-20 02:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 21

Adapt tests

Browse source 
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2025-05-16 19:06:45 +02:00
parent b35483742c
commit c15e91a65b
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
pkg/templates/multimodal_test.go
View file

@ -20,7 +20,7 @@ var _ = Describe("EvaluateTemplate", func() {
VideosInMessage: 0, VideosInMessage: 0,
}, "bar") }, "bar")
Expect(err).NotTo(HaveOccurred()) 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() { It("should handle messages with more images correctly", func() {
@ -33,7 +33,7 @@ var _ = Describe("EvaluateTemplate", func() {
VideosInMessage: 0, VideosInMessage: 0,
}, "bar") }, "bar")
Expect(err).NotTo(HaveOccurred()) 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() { It("should handle messages with more images correctly", func() {
result, err := TemplateMultiModal("", MultiModalOptions{ result, err := TemplateMultiModal("", MultiModalOptions{
@ -45,7 +45,7 @@ var _ = Describe("EvaluateTemplate", func() {
VideosInMessage: 0, VideosInMessage: 0,
}, "bar") }, "bar")
Expect(err).NotTo(HaveOccurred()) 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() { It("should handle messages with more images correctly", func() {
result, err := TemplateMultiModal("", MultiModalOptions{ result, err := TemplateMultiModal("", MultiModalOptions{
@ -57,7 +57,7 @@ var _ = Describe("EvaluateTemplate", func() {
VideosInMessage: 0, VideosInMessage: 0,
}, "bar") }, "bar")
Expect(err).NotTo(HaveOccurred()) 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() { It("should handle messages with more images correctly", func() {
result, err := TemplateMultiModal("", MultiModalOptions{ result, err := TemplateMultiModal("", MultiModalOptions{