Update api/common.go

api/common.go

@@ -6,7 +6,8 @@ import (
 	"os"
 	"strings"
 
-    "fmt"
+	"fmt"
+	"encoding/json"
 	"github.com/linweiyuan/go-chatgpt-api/util/logger"
 	"github.com/gin-gonic/gin"
 	_ "github.com/linweiyuan/go-chatgpt-api/env"
@@ -85,7 +86,8 @@ func HandleConversationResponse(c *gin.Context, resp *http.Response) {
 		}
 
 		line = strings.TrimSpace(line)
-		logger.Info(fmt.Sprintf("HandleConversationResponse: %s", line))
+
+		// logger.Info(fmt.Sprintf("HandleConversationResponse: %s", line))
 		if strings.HasPrefix(line, "event") ||
 			strings.HasPrefix(line, "data: 20") ||
 			line == "" {
@@ -94,6 +96,24 @@ func HandleConversationResponse(c *gin.Context, resp *http.Response) {
 
 		c.Writer.Write([]byte(line + "\n\n"))
 		c.Writer.Flush()
+
+		data := line[6:]
+		var result map[string]interface{}
+		err := json.Unmarshal([]byte(data), &result)
+		if err != nil {
+			continue
+		}
+		message := result["message"].(map[string]interface{})
+		status := message["status"].(string)
+
+		logger.Info(fmt.Sprintf("status的值: %s", status))
+		if(status == "finished_successfully") {
+			finishDetails := message["metadata"].(map[string]interface{})["finish_details"].(map[string]interface{})
+			finishType := finishDetails["type"].(string)
+			if(finishType == "max_tokens") {
+				logger.Info(fmt.Sprintf("finish_details中type的值: %s", finishType))
+			}
+		}
 	}
 }