mirror of
https://github.com/mudler/LocalAI.git
synced 2025-05-22 19:44:59 +00:00
feat(templates): use a single template for multimodals messages (#3892)
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
parent
a1d6cc93a8
commit
ccc7cb0287
4 changed files with 140 additions and 29 deletions
|
@ -11,7 +11,77 @@ import (
|
|||
var _ = Describe("EvaluateTemplate", func() {
|
||||
Context("templating simple strings for multimodal chat", func() {
|
||||
It("should template messages correctly", func() {
|
||||
result, err := TemplateMultiModal("[img-{{.ID}}]{{.Text}}", 1, "bar")
|
||||
result, err := TemplateMultiModal("", MultiModalOptions{
|
||||
TotalImages: 1,
|
||||
TotalAudios: 0,
|
||||
TotalVideos: 0,
|
||||
ImagesInMessage: 1,
|
||||
AudiosInMessage: 0,
|
||||
VideosInMessage: 0,
|
||||
}, "bar")
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(result).To(Equal("[img-0]bar"))
|
||||
})
|
||||
|
||||
It("should handle messages with more images correctly", func() {
|
||||
result, err := TemplateMultiModal("", MultiModalOptions{
|
||||
TotalImages: 2,
|
||||
TotalAudios: 0,
|
||||
TotalVideos: 0,
|
||||
ImagesInMessage: 2,
|
||||
AudiosInMessage: 0,
|
||||
VideosInMessage: 0,
|
||||
}, "bar")
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(result).To(Equal("[img-0][img-1]bar"))
|
||||
})
|
||||
It("should handle messages with more images correctly", func() {
|
||||
result, err := TemplateMultiModal("", MultiModalOptions{
|
||||
TotalImages: 4,
|
||||
TotalAudios: 1,
|
||||
TotalVideos: 0,
|
||||
ImagesInMessage: 2,
|
||||
AudiosInMessage: 1,
|
||||
VideosInMessage: 0,
|
||||
}, "bar")
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(result).To(Equal("[audio-0][img-2][img-3]bar"))
|
||||
})
|
||||
It("should handle messages with more images correctly", func() {
|
||||
result, err := TemplateMultiModal("", MultiModalOptions{
|
||||
TotalImages: 3,
|
||||
TotalAudios: 1,
|
||||
TotalVideos: 0,
|
||||
ImagesInMessage: 1,
|
||||
AudiosInMessage: 1,
|
||||
VideosInMessage: 0,
|
||||
}, "bar")
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(result).To(Equal("[audio-0][img-2]bar"))
|
||||
})
|
||||
It("should handle messages with more images correctly", func() {
|
||||
result, err := TemplateMultiModal("", MultiModalOptions{
|
||||
TotalImages: 0,
|
||||
TotalAudios: 0,
|
||||
TotalVideos: 0,
|
||||
ImagesInMessage: 0,
|
||||
AudiosInMessage: 0,
|
||||
VideosInMessage: 0,
|
||||
}, "bar")
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(result).To(Equal("bar"))
|
||||
})
|
||||
})
|
||||
Context("templating with custom defaults", func() {
|
||||
It("should handle messages with more images correctly", func() {
|
||||
result, err := TemplateMultiModal("{{ range .Audio }}[audio-{{ add1 .ID}}]{{end}}{{ range .Images }}[img-{{ add1 .ID}}]{{end}}{{ range .Video }}[vid-{{ add1 .ID}}]{{end}}{{.Text}}", MultiModalOptions{
|
||||
TotalImages: 1,
|
||||
TotalAudios: 0,
|
||||
TotalVideos: 0,
|
||||
ImagesInMessage: 1,
|
||||
AudiosInMessage: 0,
|
||||
VideosInMessage: 0,
|
||||
}, "bar")
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(result).To(Equal("[img-1]bar"))
|
||||
})
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue