You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
1.2 KiB
37 lines
1.2 KiB
2 years ago
|
package utils_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
. "github.com/go-skynet/LocalAI/pkg/utils"
|
||
|
|
. "github.com/onsi/ginkgo/v2"
|
||
|
|
. "github.com/onsi/gomega"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _ = Describe("Gallery API tests", func() {
|
||
|
|
Context("URI", func() {
|
||
|
|
It("parses github with a branch", func() {
|
||
|
|
Expect(
|
||
|
|
GetURI("github:go-skynet/model-gallery/gpt4all-j.yaml", func(url string, i []byte) error {
|
||
|
|
Expect(url).To(Equal("https://raw.githubusercontent.com/go-skynet/model-gallery/main/gpt4all-j.yaml"))
|
||
|
|
return nil
|
||
|
|
}),
|
||
|
|
).ToNot(HaveOccurred())
|
||
|
|
})
|
||
|
|
It("parses github without a branch", func() {
|
||
|
|
Expect(
|
||
|
|
GetURI("github:go-skynet/model-gallery/gpt4all-j.yaml@main", func(url string, i []byte) error {
|
||
|
|
Expect(url).To(Equal("https://raw.githubusercontent.com/go-skynet/model-gallery/main/gpt4all-j.yaml"))
|
||
|
|
return nil
|
||
|
|
}),
|
||
|
|
).ToNot(HaveOccurred())
|
||
|
|
})
|
||
|
|
It("parses github with urls", func() {
|
||
|
|
Expect(
|
||
|
|
GetURI("https://raw.githubusercontent.com/go-skynet/model-gallery/main/gpt4all-j.yaml", func(url string, i []byte) error {
|
||
|
|
Expect(url).To(Equal("https://raw.githubusercontent.com/go-skynet/model-gallery/main/gpt4all-j.yaml"))
|
||
|
|
return nil
|
||
|
|
}),
|
||
|
|
).ToNot(HaveOccurred())
|
||
|
|
})
|
||
|
|
})
|
||
|
|
})
|