fork-it/gpt-load
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: retrycount log

Browse Source
This commit is contained in:
tbphp
2025-06-10 14:39:46 +08:00
parent ed84d53769
commit 57b6d26a87
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
internal/proxy/server.go
View File

@@ -147,20 +147,17 @@ func (ps *ProxyServer) isStreamRequest(bodyBytes []byte, c *gin.Context) bool {
func (ps *ProxyServer) executeRequestWithRetry(c *gin.Context, startTime time.Time, bodyBytes []byte, isStreamRequest bool, retryCount int, retryErrors []types.RetryError) { func (ps *ProxyServer) executeRequestWithRetry(c *gin.Context, startTime time.Time, bodyBytes []byte, isStreamRequest bool, retryCount int, retryErrors []types.RetryError) {
keysConfig := ps.configManager.GetKeysConfig() keysConfig := ps.configManager.GetKeysConfig()
// Check retry limit if retryCount > keysConfig.MaxRetries {
if retryCount >= keysConfig.MaxRetries { logrus.Debugf("Max retries exceeded (%d)", retryCount-1)
logrus.Debugf("Max retries exceeded (%d)", retryCount)
// Return detailed error information
errorResponse := gin.H{ errorResponse := gin.H{
"error": "Max retries exceeded", "error": "Max retries exceeded",
"code": errors.ErrProxyRetryExhausted, "code": errors.ErrProxyRetryExhausted,
"retry_count": retryCount, "retry_count": retryCount - 1,
"retry_errors": retryErrors, "retry_errors": retryErrors,
"timestamp": time.Now().UTC().Format(time.RFC3339), "timestamp": time.Now().UTC().Format(time.RFC3339),
} }
// Use the last error's status code if available
statusCode := http.StatusBadGateway statusCode := http.StatusBadGateway
if len(retryErrors) > 0 && retryErrors[len(retryErrors)-1].StatusCode > 0 { if len(retryErrors) > 0 && retryErrors[len(retryErrors)-1].StatusCode > 0 {
statusCode = retryErrors[len(retryErrors)-1].StatusCode statusCode = retryErrors[len(retryErrors)-1].StatusCode