🔖 chore: openai realtime add error log

common/requester/ws_proxy.go

@@ -1,10 +1,12 @@
 package requester
 
 import (
+	"fmt"
 	"time"
 
 	"github.com/gorilla/websocket"
 
+	"one-api/common/logger"
 	"one-api/types"
 )
 
@@ -70,27 +72,21 @@ func (p *WSProxy) transfer(src, dst *websocket.Conn, source MessageSource, close
 		p.done <- struct{}{}
 	}()
 
-	inactivityTimer := time.NewTimer(p.timeout)
-	defer inactivityTimer.Stop()
-
 	for {
 		src.SetReadDeadline(time.Now().Add(p.timeout))
 
 		messageType, message, err := src.ReadMessage()
 		if err != nil {
+			logger.SysError(fmt.Sprintf("source: %d, ReadMessage error: %s", source, err.Error()))
 			return
 		}
 
-		if !inactivityTimer.Stop() {
-			<-inactivityTimer.C
-		}
-		inactivityTimer.Reset(p.timeout)
-
 		if p.handler != nil {
 			shouldContinue, usage, newMessage, err := p.handler(source, messageType, message)
 			if err != nil {
 				errMsg := []byte(err.Error())
 				dst.WriteMessage(websocket.TextMessage, errMsg)
+				logger.SysError(fmt.Sprintf("source: %d, handler error: %s", source, err.Error()))
 				return
 			}
 
@@ -108,13 +104,15 @@ func (p *WSProxy) transfer(src, dst *websocket.Conn, source MessageSource, close
 					dst.WriteMessage(websocket.TextMessage, message)
 					errMsg := []byte(err.Error())
 					dst.WriteMessage(websocket.TextMessage, errMsg)
+					logger.SysError(fmt.Sprintf("source: %d, usageHandler error: %s", source, err.Error()))
 					return
 				}
 			}
 		}
 
 		err = dst.WriteMessage(messageType, message)
 		if err != nil {
+			logger.SysError(fmt.Sprintf("source: %d, WriteMessage error: %s", source, err.Error()))
 			return
 		}
 	}

providers/openai/realtime.go

@@ -6,6 +6,7 @@ import (
 	"net/http"
 	"one-api/common"
 	"one-api/common/config"
+	"one-api/common/logger"
 	"one-api/common/requester"
 	"one-api/types"
 
@@ -58,6 +59,7 @@ func (p *OpenAIProvider) HandleMessage(source requester.MessageSource, messageTy
 
 	// 处理错误事件
 	if event.IsError() {
+		logger.SysError("event error: " + event.Error())
 		return false, nil, nil, &event
 	}
 

relay/realtime.go

@@ -64,7 +64,7 @@ func ChatRealtime(c *gin.Context) {
 
 	relay.usage = &types.UsageEvent{}
 
-	wsProxy := requester.NewWSProxy(relay.userConn, relay.providerConn, time.Minute*1, relay.messageHandler, relay.usageHandler)
+	wsProxy := requester.NewWSProxy(relay.userConn, relay.providerConn, time.Minute*2, relay.messageHandler, relay.usageHandler)
 
 	wsProxy.Start()