mirror of
https://github.com/mudler/cogito.git
synced 2026-07-24 02:45:53 -04:00
Ettore Di Giacinto
2f20fcce01
* Split e2e and unit tests Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Abstract away LLM to an interface Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Update README Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Fixups Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Enhance content improve generation Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Add unit test for reviewer Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Better handling of result status of tool calls Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Fixups Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Add tools test file Signed-off-by: Ettore Di Giacinto <mudler@localai.io> --------- Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
24 lines
534 B
Go
24 lines
534 B
Go
package cogito_test
|
|
|
|
import (
|
|
"context"
|
|
|
|
. "github.com/mudler/cogito"
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("Client test", Label("e2e"), func() {
|
|
Context("A simple pipeline", func() {
|
|
It("should ask to the LLM", func() {
|
|
defaultLLM := NewOpenAILLM(defaultModel, "", apiEndpoint)
|
|
|
|
conv := NewEmptyFragment().AddMessage("user", "Hi!")
|
|
|
|
result, err := defaultLLM.Ask(context.TODO(), conv)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(result.String()).ToNot(BeEmpty())
|
|
})
|
|
})
|
|
})
|