mirror of
https://github.com/mudler/cogito.git
synced 2026-07-24 02:45:53 -04:00
Ettore Di Giacinto
24 lines
514 B
Go
24 lines
514 B
Go
package cogito_test
|
|
|
|
import (
|
|
"context"
|
|
|
|
. "github.com/mudler/cogito"
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("Client test", func() {
|
|
Context("A simple pipeline", func() {
|
|
It("should ask to the LLM", func() {
|
|
defaultLLM := NewLLM(defaultModel, "", apiEndpoint)
|
|
|
|
conv := NewEmptyFragment().AddMessage("user", "Hi!")
|
|
|
|
result, err := defaultLLM.Ask(context.TODO(), conv)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(result.String()).ToNot(BeEmpty())
|
|
})
|
|
})
|
|
})
|