Golang bindings initial working version(#534)

* WIP

* Fix includes

* Try to fix linking issues

* Refinements

* allow to load MPT and llama models too

* cleanup, add example, add README
This commit is contained in:
Ettore Di Giacinto
2023-05-15 18:45:56 +02:00
committed by GitHub
parent 2433902460
commit 3f63cc6b47
11 changed files with 801 additions and 0 deletions

View File

@@ -0,0 +1,27 @@
package gpt4all_test
import (
. "github.com/nomic-ai/gpt4all/gpt4all-bindings/golang"
. "github.com/onsi/ginkgo/v2"
. "github.com/onsi/gomega"
)
var _ = Describe("LLama binding", func() {
Context("Declaration", func() {
It("fails with no model", func() {
model, err := New("not-existing")
Expect(err).To(HaveOccurred())
Expect(model).To(BeNil())
})
It("fails with no model", func() {
model, err := New("not-existing", SetModelType(MPTType))
Expect(err).To(HaveOccurred())
Expect(model).To(BeNil())
})
It("fails with no model", func() {
model, err := New("not-existing", SetModelType(LLaMAType))
Expect(err).To(HaveOccurred())
Expect(model).To(BeNil())
})
})
})