mirror of
https://github.com/mudler/LocalAI.git
synced 2024-06-07 19:40:48 +00:00
31 lines
1.1 KiB
Go
31 lines
1.1 KiB
Go
|
package api_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
. "github.com/go-skynet/LocalAI/api"
|
||
|
|
. "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() {
|
||
|
|
req := ApplyGalleryModelRequest{URL: "github:go-skynet/model-gallery/gpt4all-j.yaml@main"}
|
||
|
|
str, err := req.DecodeURL()
|
||
|
|
Expect(err).ToNot(HaveOccurred())
|
||
|
|
Expect(str).To(Equal("https://raw.githubusercontent.com/go-skynet/model-gallery/main/gpt4all-j.yaml"))
|
||
|
|
})
|
||
|
|
It("parses github without a branch", func() {
|
||
|
|
req := ApplyGalleryModelRequest{URL: "github:go-skynet/model-gallery/gpt4all-j.yaml"}
|
||
|
|
str, err := req.DecodeURL()
|
||
|
|
Expect(err).ToNot(HaveOccurred())
|
||
|
|
Expect(str).To(Equal("https://raw.githubusercontent.com/go-skynet/model-gallery/main/gpt4all-j.yaml"))
|
||
|
|
})
|
||
|
|
It("parses URLS", func() {
|
||
|
|
req := ApplyGalleryModelRequest{URL: "https://raw.githubusercontent.com/go-skynet/model-gallery/main/gpt4all-j.yaml"}
|
||
|
|
str, err := req.DecodeURL()
|
||
|
|
Expect(err).ToNot(HaveOccurred())
|
||
|
|
Expect(str).To(Equal("https://raw.githubusercontent.com/go-skynet/model-gallery/main/gpt4all-j.yaml"))
|
||
|
|
})
|
||
|
|
})
|
||
|
|
})
|