From 8b8ac7b24488e7074e8f0d5075e95adf8454e8c2 Mon Sep 17 00:00:00 2001 From: Chris Sexton <3216719+chrissexton@users.noreply.github.com> Date: Fri, 3 Mar 2023 12:14:06 -0500 Subject: [PATCH] gpt: reset client when prompt changes --- plugins/gpt/chatgpt.go | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/plugins/gpt/chatgpt.go b/plugins/gpt/chatgpt.go index f7ea523..1267350 100644 --- a/plugins/gpt/chatgpt.go +++ b/plugins/gpt/chatgpt.go @@ -6,7 +6,7 @@ import ( ) import "github.com/andrewstuart/openai" -var session *openai.ChatSession +var session openai.ChatSession var client *openai.Client func (p *GPTPlugin) getClient() (*openai.Client, error) { @@ -14,14 +14,11 @@ func (p *GPTPlugin) getClient() (*openai.Client, error) { if token == "" { return nil, fmt.Errorf("no GPT token given") } - if client == nil { - return openai.NewClient(token) - } - return client, nil + return openai.NewClient(token) } func (p *GPTPlugin) chatGPT(request string) (string, error) { - if session == nil { + if client == nil { if err := p.setDefaultPrompt(); err != nil { return "", err } @@ -38,7 +35,6 @@ func (p *GPTPlugin) setPrompt(prompt string) error { if err != nil { return err } - sess := client.NewChatSession(prompt) - session = &sess + session = client.NewChatSession(prompt) return nil }