mirror of
https://github.com/mudler/LocalAI.git
synced 2025-05-30 23:44:59 +00:00
88de2ea01a
* feat(llama.cpp): add support for audio input Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Adapt tests Signed-off-by: Ettore Di Giacinto <mudler@localai.io> --------- Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
89 lines
2.7 KiB
Go
89 lines
2.7 KiB
Go
package templates_test
|
|
|
|
import (
|
|
. "github.com/mudler/LocalAI/pkg/templates" // Update with your module path
|
|
|
|
// Update with your module path
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("EvaluateTemplate", func() {
|
|
Context("templating simple strings for multimodal chat", func() {
|
|
It("should template messages correctly", func() {
|
|
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("<__media__>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("<__media__><__media__>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("<__media__><__media__><__media__>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("<__media__><__media__>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"))
|
|
})
|
|
})
|
|
})
|