ChatGPT/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-10-16 16:13:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: save response time & test time (#59)

Browse Source
This commit is contained in:
JustSong
2023-05-15 11:35:38 +08:00
parent 443a22b75d
commit 225176aae9
3 changed files with 46 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
controller/channel.go
View File

@@ -89,7 +89,6 @@ func AddChannel(c *gin.Context) {
return
}
channel.CreatedTime = common.GetTimestamp()
channel.AccessedTime = common.GetTimestamp()
keys := strings.Split(channel.Key, "\n")
channels := make([]model.Channel, 0)
for _, key := range keys {
@@ -236,7 +235,9 @@ func TestChannel(c *gin.Context) {
tik := time.Now()
err = testChannel(channel, chatRequest)
tok := time.Now()
consumedTime := float64(tok.Sub(tik).Milliseconds()) / 1000.0
milliseconds := tok.Sub(tik).Milliseconds()
go channel.UpdateResponseTime(milliseconds)
consumedTime := float64(milliseconds) / 1000.0
if err != nil {
c.JSON(http.StatusOK, gin.H{
"success": false,