mirrors2/opencatd-open
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

up

Browse Source
This commit is contained in:
Sakurasan
2023-04-25 22:27:19 +08:00
parent 97a13a5f79
commit 56b56c703d
1 changed files with 4 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
router/router.go
View File

@@ -338,8 +338,8 @@ func HandleProy(c *gin.Context) {
if c.Request.URL.Path == "/v1/chat/completions" && localuser { if c.Request.URL.Path == "/v1/chat/completions" && localuser {
if err := c.BindJSON(&chatreq); err != nil { if err := c.BindJSON(&chatreq); err != nil {
c.AbortWithError(http.StatusBadRequest, err)
return return
// c.AbortWithError(http.StatusBadRequest, err)
} }
chatlog.Model = chatreq.Model chatlog.Model = chatreq.Model
for _, m := range chatreq.Messages { for _, m := range chatreq.Messages {
@@ -350,9 +350,10 @@ func HandleProy(c *gin.Context) {
isStream = chatreq.Stream isStream = chatreq.Stream
chatlog.UserID, _ = store.GetUserID(auth[7:]) chatlog.UserID, _ = store.GetUserID(auth[7:])
} }
var body bytes.Buffer
json.NewEncoder(&body).Encode(chatreq)
// 创建 API 请求 // 创建 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 { if err != nil {
log.Println(err) log.Println(err)
c.JSON(http.StatusOK, gin.H{"error": err.Error()}) c.JSON(http.StatusOK, gin.H{"error": err.Error()})
@@ -499,18 +500,6 @@ func HandleUsage(c *gin.Context) {
fromStr := c.Query("from") fromStr := c.Query("from")
toStr := c.Query("to") 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) usage, err := store.QueryUsage(fromStr, toStr)
if err != nil { if err != nil {
c.JSON(http.StatusForbidden, gin.H{"error": err.Error()}) c.JSON(http.StatusForbidden, gin.H{"error": err.Error()})