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

refactor(template): isolate and add tests (#2069)

Browse source 
* refactor(template): isolate and add tests

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>

---------

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
Signed-off-by: Dave <dave@gray101.com>
Co-authored-by: Dave <dave@gray101.com>
This commit is contained in:
Ettore Di Giacinto 2024-04-19 04:40:18 +02:00 • committed by GitHub
parent 852316c5a6
commit 27ec84827c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 218 additions and 95 deletions
Download patch file Download diff file Expand all files Collapse all files

7
pkg/model/loader_test.go
View file

@ -92,10 +92,13 @@ var testMatch map[string]map[string]interface{} = map[string]map[string]interfac
var _ = Describe("Templates", func() {
Context("chat message", func() {
modelLoader := NewModelLoader("")
var modelLoader *ModelLoader
BeforeEach(func() {
modelLoader = NewModelLoader("")
})
for key := range testMatch {
foo := testMatch[key]
It("renders correctly "+key, func() {
It("renders correctly `"+key+"`", func() {
templated, err := modelLoader.EvaluateTemplateForChatMessage(foo["template"].(string), foo["data"].(model.ChatMessageTemplateData))
Expect(err).ToNot(HaveOccurred())
Expect(templated).To(Equal(foo["expected"]), templated)