From c38181829ad8a3e2615d22b330be49b7bdc3db19 Mon Sep 17 00:00:00 2001
From: Zhang Minghan <zmh13054618081@dingtalk.com>
Date: Sun, 12 Nov 2023 10:37:53 +0800
Subject: [PATCH] update transhipment

---
 connection/database.go  |  2 +-
 manager/transhipment.go | 11 ++++++-----
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/connection/database.go b/connection/database.go
index 1a6516e..418db2b 100644
--- a/connection/database.go
+++ b/connection/database.go
@@ -105,7 +105,7 @@ func CreateConversationTable(db *sql.DB) {
 		  user_id INT,
 		  conversation_id INT,
 		  conversation_name VARCHAR(255),
-		  data TEXT,
+		  data MEDIUMTEXT,
 		  model VARCHAR(255) NOT NULL DEFAULT 'gpt-3.5-turbo-0613',
 		  updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
 		  UNIQUE KEY (user_id, conversation_id)
diff --git a/manager/transhipment.go b/manager/transhipment.go
index 914853e..f3a089e 100644
--- a/manager/transhipment.go
+++ b/manager/transhipment.go
@@ -135,14 +135,15 @@ func sendTranshipmentResponse(c *gin.Context, form TranshipmentForm, id string,
 
 	CollectQuota(c, user, buffer, plan)
 	c.JSON(http.StatusOK, TranshipmentResponse{
-		Id:      id,
+		Id:      fmt.Sprintf("chatcmpl-%s", id),
 		Object:  "chat.completion",
 		Created: created,
 		Model:   form.Model,
 		Choices: []Choice{
 			{
-				Index:   0,
-				Message: globals.Message{Role: "assistant", Content: buffer.ReadWithDefault(defaultMessage)},
+				Index:        0,
+				Message:      globals.Message{Role: "assistant", Content: buffer.ReadWithDefault(defaultMessage)},
+				FinishReason: "stop",
 			},
 		},
 		Usage: Usage{
@@ -156,8 +157,8 @@ func sendTranshipmentResponse(c *gin.Context, form TranshipmentForm, id string,
 
 func getStreamTranshipmentForm(id string, created int64, form TranshipmentForm, data string, buffer *utils.Buffer, end bool) TranshipmentStreamResponse {
 	return TranshipmentStreamResponse{
-		Id:      id,
-		Object:  "chat.completion",
+		Id:      fmt.Sprintf("chatcmpl-%s", id),
+		Object:  "chat.completion.chunk",
 		Created: created,
 		Model:   form.Model,
 		Choices: []ChoiceDelta{