5698d0b832
This saves time when testing on CPU which is the only sensible thing to do on GitHub CI for PRs. For releases or once the commit is merged we could use an external runner with GPU or just wait. Signed-off-by: Richard Palethorpe <io@richiejp.com>
111 lines
3.3 KiB
Go
111 lines
3.3 KiB
Go
package agent_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/mudler/LocalAGI/pkg/llm"
|
|
"github.com/sashabaranov/go-openai"
|
|
|
|
. "github.com/mudler/LocalAGI/core/agent"
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
|
|
)
|
|
|
|
var _ = Describe("Agent test", func() {
|
|
Context("identity", func() {
|
|
var agent *Agent
|
|
|
|
// BeforeEach(func() {
|
|
// Eventually(func() error {
|
|
// // test apiURL is working and available
|
|
// _, err := http.Get(apiURL + "/readyz")
|
|
// return err
|
|
// }, "10m", "10s").ShouldNot(HaveOccurred())
|
|
// })
|
|
|
|
It("generates all the fields with random data", func() {
|
|
var llmClient llm.LLMClient
|
|
if useRealLocalAI {
|
|
llmClient = llm.NewClient(apiKey, apiURL, testModel)
|
|
} else {
|
|
llmClient = &llm.MockClient{
|
|
CreateChatCompletionFunc: func(ctx context.Context, req openai.ChatCompletionRequest) (openai.ChatCompletionResponse, error) {
|
|
return openai.ChatCompletionResponse{
|
|
Choices: []openai.ChatCompletionChoice{{
|
|
Message: openai.ChatCompletionMessage{
|
|
ToolCalls: []openai.ToolCall{{
|
|
ID: "tool_call_id_1",
|
|
Type: "function",
|
|
Function: openai.FunctionCall{
|
|
Name: "generate_identity",
|
|
Arguments: `{"name":"John Doe","age":"42","job_occupation":"Engineer","hobbies":["reading","hiking"],"favorites_music_genres":["Jazz"]}`,
|
|
},
|
|
}},
|
|
},
|
|
}},
|
|
}, nil
|
|
},
|
|
}
|
|
}
|
|
var err error
|
|
agent, err = New(
|
|
WithLLMClient(llmClient),
|
|
WithModel(testModel),
|
|
WithTimeout("10m"),
|
|
WithRandomIdentity(),
|
|
)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
By("generating random identity")
|
|
Expect(agent.Character.Name).ToNot(BeEmpty())
|
|
Expect(agent.Character.Age).ToNot(BeZero())
|
|
Expect(agent.Character.Occupation).ToNot(BeEmpty())
|
|
Expect(agent.Character.Hobbies).ToNot(BeEmpty())
|
|
Expect(agent.Character.MusicTaste).ToNot(BeEmpty())
|
|
})
|
|
It("detect an invalid character", func() {
|
|
mock := &llm.MockClient{
|
|
CreateChatCompletionFunc: func(ctx context.Context, req openai.ChatCompletionRequest) (openai.ChatCompletionResponse, error) {
|
|
return openai.ChatCompletionResponse{}, fmt.Errorf("invalid character")
|
|
},
|
|
}
|
|
var err error
|
|
agent, err = New(
|
|
WithLLMClient(mock),
|
|
WithRandomIdentity(),
|
|
)
|
|
Expect(err).To(HaveOccurred())
|
|
})
|
|
It("generates all the fields", func() {
|
|
mock := &llm.MockClient{
|
|
CreateChatCompletionFunc: func(ctx context.Context, req openai.ChatCompletionRequest) (openai.ChatCompletionResponse, error) {
|
|
return openai.ChatCompletionResponse{
|
|
Choices: []openai.ChatCompletionChoice{{
|
|
Message: openai.ChatCompletionMessage{
|
|
ToolCalls: []openai.ToolCall{{
|
|
ID: "tool_call_id_2",
|
|
Type: "function",
|
|
Function: openai.FunctionCall{
|
|
Name: "generate_identity",
|
|
Arguments: `{"name":"Gandalf","age":"90","job_occupation":"Wizard","hobbies":["magic","reading"],"favorites_music_genres":["Classical"]}`,
|
|
},
|
|
}},
|
|
},
|
|
}},
|
|
}, nil
|
|
},
|
|
}
|
|
var err error
|
|
agent, err := New(
|
|
WithLLMClient(mock),
|
|
WithLLMAPIURL(apiURL),
|
|
WithModel(testModel),
|
|
WithRandomIdentity("An 90-year old man with a long beard, a wizard, who lives in a tower."),
|
|
)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(agent.Character.Name).ToNot(BeEmpty())
|
|
})
|
|
})
|
|
})
|