|
@@ -8,6 +8,7 @@ import (
|
|
|
"github.com/gogf/gf/v2/net/ghttp"
|
|
|
"github.com/gogf/gf/v2/os/glog"
|
|
|
"github.com/gogf/gf/v2/util/guid"
|
|
|
+ "github.com/google/uuid"
|
|
|
v1 "go-gpt/api/v1"
|
|
|
"go-gpt/gpt"
|
|
|
)
|
|
@@ -32,15 +33,16 @@ func (c *sChat) Chat(ctx context.Context, req *v1.ChatReq) (res *v1.ChatRes, err
|
|
|
var chatRequst gpt.ChatCompletionRequest
|
|
|
var chatMessage []gpt.ChatCompletionRequestMessage
|
|
|
|
|
|
- //re := g.RequestFromCtx(ctx)
|
|
|
- //ConversationID := ""
|
|
|
+ re := g.RequestFromCtx(ctx)
|
|
|
+ ConversationID := res.ConversationID
|
|
|
+ MessageId := res.MessageId
|
|
|
chatRequst.Model = gpt.GPT3Dot5Turbo
|
|
|
chatRequst.MaxTokens = 1024
|
|
|
chatRequst.Temperature = 0
|
|
|
|
|
|
if req.ConversationID == "" || req.MessageId == "" {
|
|
|
- //ConversationID = CreateSessionId(re)
|
|
|
- //chatRequst.ConversationID = ConversationID
|
|
|
+ ConversationID = CreateSessionId(re)
|
|
|
+ MessageId = uuid.NewString()
|
|
|
}
|
|
|
chatMessage = append(chatMessage, gpt.ChatCompletionRequestMessage{
|
|
|
Role: req.Role,
|
|
@@ -54,6 +56,8 @@ func (c *sChat) Chat(ctx context.Context, req *v1.ChatReq) (res *v1.ChatRes, err
|
|
|
}
|
|
|
res = new(v1.ChatRes)
|
|
|
if len(completion.Choices) > 0 {
|
|
|
+ res.ConversationID = ConversationID
|
|
|
+ res.MessageId = MessageId
|
|
|
res.Text = completion.Choices[0].Message.Content
|
|
|
}
|
|
|
|