Skip to content

Commit

Permalink
Update the test. (#197)
Browse files Browse the repository at this point in the history
* I wanted to inspect the output to see if the AI was returning
responses that end with a closing tag "</output>" but it doesn't look
like it.
  • Loading branch information
jlewi authored Aug 19, 2024
1 parent b70f043 commit 266c923
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions app/pkg/agent/agent_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,8 @@ import (
"testing"
"time"

"github.com/sashabaranov/go-openai"

parserv1 "github.com/stateful/runme/v3/pkg/api/gen/proto/go/runme/parser/v1"

"github.com/go-logr/zapr"
Expand Down Expand Up @@ -102,6 +104,9 @@ func Test_Generate(t *testing.T) {
t.Fatalf("Error creating in memory DB; %v", err)
}

cfg.Agent.ModelProvider = api.ModelProviderOpenAI
cfg.Agent.Model = openai.GPT3Dot5Turbo0125

completer, err := oai.NewCompleter(*cfg, client)
if err != nil {
t.Fatalf("Error creating completer; %v", err)
Expand Down

0 comments on commit 266c923

Please sign in to comment.