2023-06-24 08:18:17 +02:00
|
|
|
package gallery_test
|
2023-05-20 09:06:30 +02:00
|
|
|
|
|
|
|
import (
|
2023-06-24 08:18:17 +02:00
|
|
|
. "github.com/go-skynet/LocalAI/pkg/gallery"
|
2023-05-20 09:06:30 +02:00
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
|
|
. "github.com/onsi/gomega"
|
|
|
|
)
|
|
|
|
|
|
|
|
var _ = Describe("Gallery API tests", func() {
|
|
|
|
Context("requests", func() {
|
|
|
|
It("parses github with a branch", func() {
|
2023-06-24 08:18:17 +02:00
|
|
|
req := GalleryModel{URL: "github:go-skynet/model-gallery/gpt4all-j.yaml@main"}
|
2024-01-05 12:04:46 -05:00
|
|
|
e, err := GetGalleryConfigFromURL(req.URL)
|
2023-06-24 08:18:17 +02:00
|
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(e.Name).To(Equal("gpt4all-j"))
|
|
|
|
})
|
2023-05-20 09:06:30 +02:00
|
|
|
})
|
|
|
|
})
|