使用gpt-3.5-turbo优化模型

2023-03-03 22:12:46 +08:00
parent 39cbed1853
commit dc4ffb96b3
2 changed files with 3 additions and 2 deletions
--- a/chat/service.go
+++ b/chat/service.go
@@ -78,7 +78,8 @@ func (api *Api) GetChatMessage(conn *websocket.Conn, cli *gogpt.Client, mutex *s
 	var err error
 	var strResp string
 	req := gogpt.CompletionRequest{
-		Model:            gogpt.GPT3TextDavinci003,
+		// Model:            gogpt.GPT3TextDavinci003,
+		Model:            "gpt-3.5-turbo",
 		MaxTokens:        api.Config.MaxLength,
 		Temperature:      0.6,
 		Prompt:           requestMsg,
--- a/go.mod
+++ b/go.mod
@@ -7,7 +7,7 @@ require (
 	github.com/gin-gonic/gin v1.8.2
 	github.com/google/uuid v1.3.0
 	github.com/gorilla/websocket v1.5.0
-	github.com/sashabaranov/go-gpt3 v1.1.0
+	github.com/sashabaranov/go-gpt3 v1.3.3
 	github.com/sirupsen/logrus v1.9.0
 	gopkg.in/yaml.v3 v3.0.1
 )