mirror of
https://github.com/mudler/LocalAI.git
synced 2026-01-05 18:19:53 -06:00
b3c2a3c257
* migrate core/system to pkg/system - it has no dependencies FROM core, and IS USED in pkg Signed-off-by: Dave Lee <dave@gray101.com> * move pkg/templates up to core/templates -- nothing in pkg references it, but it does reference core. Signed-off-by: Dave Lee <dave@gray101.com> * remove extra check, len of nil is 0 Signed-off-by: Dave Lee <dave@gray101.com> * move pkg/startup to core/startup -- it does have important and unfixable dependencies on core Signed-off-by: Dave Lee <dave@gray101.com> --------- Signed-off-by: Dave Lee <dave@gray101.com>
90 lines
2.7 KiB
Go
90 lines
2.7 KiB
Go
package templates_test
|
|
|
|
import (
|
|
. "github.com/mudler/LocalAI/core/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"))
|
|
})
|
|
})
|
|
})
|