refactor: advanced custom channel route editor (#6865)
* refactor: advanced custom channel route editor * fix(channels): show raw balance response from balance cell
This commit is contained in:
@@ -304,6 +304,34 @@ func sanitizeFetchModelsError(err error, key string) error {
|
||||
return errors.New(message)
|
||||
}
|
||||
|
||||
func sanitizeAdvancedCustomRequestError(err error, key string, requestURL string) error {
|
||||
err = sanitizeFetchModelsError(err, key)
|
||||
if err == nil {
|
||||
return nil
|
||||
}
|
||||
parsedURL, parseErr := url.Parse(requestURL)
|
||||
if parseErr != nil {
|
||||
return err
|
||||
}
|
||||
message := err.Error()
|
||||
for _, value := range parsedURL.Query() {
|
||||
for _, secret := range value {
|
||||
if secret == "" {
|
||||
continue
|
||||
}
|
||||
message = strings.ReplaceAll(message, secret, "[REDACTED]")
|
||||
message = strings.ReplaceAll(message, url.QueryEscape(secret), "[REDACTED]")
|
||||
message = strings.ReplaceAll(message, url.PathEscape(secret), "[REDACTED]")
|
||||
}
|
||||
}
|
||||
if key != "" {
|
||||
message = strings.ReplaceAll(message, key, "[REDACTED]")
|
||||
message = strings.ReplaceAll(message, url.QueryEscape(key), "[REDACTED]")
|
||||
message = strings.ReplaceAll(message, url.PathEscape(key), "[REDACTED]")
|
||||
}
|
||||
return errors.New(message)
|
||||
}
|
||||
|
||||
func getFetchModelsResponseBody(method string, requestURL string, channel *model.Channel, headers http.Header) ([]byte, error) {
|
||||
request, err := http.NewRequest(method, requestURL, nil)
|
||||
if err != nil {
|
||||
@@ -409,7 +437,7 @@ func fetchChannelUpstreamModelIDs(channel *model.Channel) ([]string, error) {
|
||||
|
||||
body, err := getFetchModelsResponseBody(http.MethodGet, url, channel, headers)
|
||||
if err != nil {
|
||||
return nil, sanitizeFetchModelsError(err, key)
|
||||
return nil, sanitizeAdvancedCustomRequestError(err, key, url)
|
||||
}
|
||||
|
||||
var result OpenAIModelsResponse
|
||||
|
||||
Reference in New Issue
Block a user