init

service/gpt.go

@@ -0,0 +1,79 @@
+package service
+
+import (
+	"context"
+	"fmt"
+
+	gogpt "github.com/sashabaranov/go-gpt3"
+)
+
+// 停顿符，用来阻止 GPT-3 补充联想内容
+var stop = "*#06#"
+
+// openai key
+var apiKey = "apiKey"
+
+// openai orgId
+var orgId = "orgId"
+
+type ChatGPT struct {
+	client *gogpt.Client
+	ctx    context.Context
+	userId string
+}
+
+func Ask(question string) (string, error) {
+	k, orgId := apiKey, orgId
+	chat := NewGPT(k, orgId)
+	defer chat.Close()
+	answer, err := chat.Chat(question)
+	if err != nil {
+		fmt.Print(err.Error())
+	}
+	return answer, err
+}
+
+func (c *ChatGPT) Chat(question string) (answer string, err error) {
+	q := question + stop
+	req := gogpt.CompletionRequest{
+		Model:            gogpt.GPT3TextDavinci003,
+		MaxTokens:        1200,
+		Prompt:           q,
+		Temperature:      0,
+		TopP:             0.99,
+		N:                1,
+		Stop:             []string{stop},
+		FrequencyPenalty: 0,
+		PresencePenalty:  0.5,
+		User:             c.userId,
+	}
+	resp, err := c.client.CreateCompletion(c.ctx, req)
+	if err != nil {
+		return "", err
+	}
+	answer = resp.Choices[0].Text
+	for len(answer) > 0 {
+		if answer[0] == '\n' {
+			answer = answer[1:]
+		} else {
+			break
+		}
+	}
+	return resp.Choices[0].Text, err
+}
+
+func NewGPT(ApiKey, UserId string) *ChatGPT {
+	ctx, cancel := context.WithCancel(context.Background())
+	go func() {
+		<-ctx.Done()
+		cancel()
+	}()
+	return &ChatGPT{
+		client: gogpt.NewClient(ApiKey),
+		ctx:    ctx,
+		userId: UserId,
+	}
+}
+func (c *ChatGPT) Close() {
+	c.ctx.Done()
+}