cws
/
catbase
mirror of https://github.com/velour/catbase.git
1
0
Fork 0
Code Issues Releases Wiki Activity

gpt: reset client when prompt changes

This commit is contained in:
Chris Sexton 2023-03-03 12:14:06 -05:00
parent 92ce4979b4
commit 8b8ac7b244
1 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
plugins/gpt/chatgpt.go
View File

@ -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
}
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
}