From 266c923056a187214e84cd0d4ffa920d0023be58 Mon Sep 17 00:00:00 2001 From: Jeremy Lewi Date: Mon, 19 Aug 2024 16:57:21 -0700 Subject: [PATCH] Update the test. (#197) * I wanted to inspect the output to see if the AI was returning responses that end with a closing tag "" but it doesn't look like it. --- app/pkg/agent/agent_test.go | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/app/pkg/agent/agent_test.go b/app/pkg/agent/agent_test.go index 0709afc0..bce04d35 100644 --- a/app/pkg/agent/agent_test.go +++ b/app/pkg/agent/agent_test.go @@ -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" @@ -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)