mirror of
https://github.com/mudler/LocalAI.git
synced 2024-12-20 21:23:10 +00:00
c4f958e11b
* start breaking up the giant channel refactor now that it's better understood - easier to merge bites Signed-off-by: Dave Lee <dave@gray101.com> * add concurrency and base64 back in, along with new base64 tests. Signed-off-by: Dave Lee <dave@gray101.com> * Automatic rename of whisper.go's Result to TranscriptResult Signed-off-by: Dave Lee <dave@gray101.com> * remove pkg/concurrency - significant changes coming in split 2 Signed-off-by: Dave Lee <dave@gray101.com> * fix comments Signed-off-by: Dave Lee <dave@gray101.com> * add list_model service as another low-risk service to get it out of the way Signed-off-by: Dave Lee <dave@gray101.com> * split backend config loader into seperate file from the actual config struct. No changes yet, just reduce cognative load with smaller files of logical blocks Signed-off-by: Dave Lee <dave@gray101.com> * rename state.go ==> application.go Signed-off-by: Dave Lee <dave@gray101.com> * fix lost import? Signed-off-by: Dave Lee <dave@gray101.com> --------- Signed-off-by: Dave Lee <dave@gray101.com>
32 lines
1.2 KiB
Go
32 lines
1.2 KiB
Go
package utils_test
|
|
|
|
import (
|
|
. "github.com/go-skynet/LocalAI/pkg/utils"
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("utils/base64 tests", func() {
|
|
It("GetImageURLAsBase64 can strip data url prefixes", func() {
|
|
// This one doesn't actually _care_ that it's base64, so feed "bad" data in this test in order to catch a change in that behavior for informational purposes.
|
|
input := "data:image/jpeg;base64,FOO"
|
|
b64, err := GetImageURLAsBase64(input)
|
|
Expect(err).To(BeNil())
|
|
Expect(b64).To(Equal("FOO"))
|
|
})
|
|
It("GetImageURLAsBase64 returns an error for bogus data", func() {
|
|
input := "FOO"
|
|
b64, err := GetImageURLAsBase64(input)
|
|
Expect(b64).To(Equal(""))
|
|
Expect(err).ToNot(BeNil())
|
|
Expect(err).To(MatchError("not valid string"))
|
|
})
|
|
It("GetImageURLAsBase64 can actually download images and calculates something", func() {
|
|
// This test doesn't actually _check_ the results at this time, which is bad, but there wasn't a test at all before...
|
|
input := "https://upload.wikimedia.org/wikipedia/en/2/29/Wargames.jpg"
|
|
b64, err := GetImageURLAsBase64(input)
|
|
Expect(err).To(BeNil())
|
|
Expect(b64).ToNot(BeNil())
|
|
})
|
|
})
|