From 56b56c703d6ba2dc2b89f5b9ed65698e87e00111 Mon Sep 17 00:00:00 2001
From: Sakurasan <1173092237@qq.com>
Date: Tue, 25 Apr 2023 22:27:19 +0800
Subject: [PATCH] up

---
 router/router.go | 19 ++++---------------
 1 file changed, 4 insertions(+), 15 deletions(-)

diff --git a/router/router.go b/router/router.go
index c6b457a..63f6e3c 100644
--- a/router/router.go
+++ b/router/router.go
@@ -338,8 +338,8 @@ func HandleProy(c *gin.Context) {
 	if c.Request.URL.Path == "/v1/chat/completions" && localuser {
 
 		if err := c.BindJSON(&chatreq); err != nil {
+			c.AbortWithError(http.StatusBadRequest, err)
 			return
-			// c.AbortWithError(http.StatusBadRequest, err)
 		}
 		chatlog.Model = chatreq.Model
 		for _, m := range chatreq.Messages {
@@ -350,9 +350,10 @@ func HandleProy(c *gin.Context) {
 		isStream = chatreq.Stream
 		chatlog.UserID, _ = store.GetUserID(auth[7:])
 	}
-
+	var body bytes.Buffer
+	json.NewEncoder(&body).Encode(chatreq)
 	// 创建 API 请求
-	req, err := http.NewRequest(c.Request.Method, baseUrl+c.Request.RequestURI, c.Request.Body)
+	req, err := http.NewRequest(c.Request.Method, baseUrl+c.Request.RequestURI, &body)
 	if err != nil {
 		log.Println(err)
 		c.JSON(http.StatusOK, gin.H{"error": err.Error()})
@@ -499,18 +500,6 @@ func HandleUsage(c *gin.Context) {
 	fromStr := c.Query("from")
 	toStr := c.Query("to")
 
-	// from, err := time.Parse("2006-01-02", fromStr)
-	// if err != nil {
-	// 	c.JSON(400, gin.H{"error": "Invalid from date format"})
-	// 	return
-	// }
-
-	// to, err := time.Parse("2006-01-02", toStr)
-	// if err != nil {
-	// 	c.JSON(400, gin.H{"error": "Invalid to date format"})
-	// 	return
-	// }
-
 	usage, err := store.QueryUsage(fromStr, toStr)
 	if err != nil {
 		c.JSON(http.StatusForbidden, gin.H{"error": err.Error()})