mirror of
https://github.com/mudler/LocalAI.git
synced 2026-05-03 16:49:45 -05:00
Ettore Di Giacinto
6d5bde860b
* WIP * wip * wip * Make it compile * Update json.hpp * this shouldn't be private for now * Add logs * Reset auto detected template Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Re-enable grammars * This seems to be broken - https://github.com/ggml-org/llama.cpp/commit/360a9c98e13d35f322b4c5b1309aab0cc90ed82b#diff-a18a8e64e12a01167d8e98fc[…]cccf0d4eed09d76d879L2998-L3207 Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Placeholder * Simplify image loading * use completion type * disable streaming Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * correctly return timings Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Remove some debug logging * Adapt tests Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Keep header * embedding: do not use oai type Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Sync from server.cpp * Use utils and json directly from llama.cpp Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Sync with upstream Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * fix: copy json.hpp from the correct location Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * fix: add httplib * sync llama.cpp Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Embeddiongs: set OAICOMPAT_TYPE_EMBEDDING Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * feat: sync with server.cpp by including it Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * make it darwin-compatible Signed-off-by: Ettore Di Giacinto <mudler@localai.io> --------- Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
90 lines
2.7 KiB
Go
90 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("<__image__>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("<__image__><__image__>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]<__image__><__image__>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]<__image__>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"))
|
|
})
|
|
})
|
|
})
|