Jelajahi Sumber

feat: add conditional setting for HTTP headers in OpenRouter channel type

CaIon 1 bulan lalu
induk
melakukan
da8a52f50a
1 mengubah file dengan 6 tambahan dan 2 penghapusan
  1. 6 2
      relay/channel/openai/adaptor.go

+ 6 - 2
relay/channel/openai/adaptor.go

@@ -225,8 +225,12 @@ func (a *Adaptor) SetupRequestHeader(c *gin.Context, header *http.Header, info *
 		}
 	}
 	if info.ChannelType == constant.ChannelTypeOpenRouter {
-		header.Set("HTTP-Referer", "https://www.newapi.ai")
-		header.Set("X-OpenRouter-Title", "New API")
+		if header.Get("HTTP-Referer") == "" {
+			header.Set("HTTP-Referer", "https://www.newapi.ai")
+		}
+		if header.Get("X-OpenRouter-Title") == "" {
+			header.Set("X-OpenRouter-Title", "New API")
+		}
 	}
 	return nil
 }