mirror of
https://gh-proxy.org/https://github.com/klinsmaya/new-api.git
synced 2026-11-08 01:17:09 +08:00
RedwindA
19d435e7fd
* fix(relay): mask only the upstream address in error messages MaskSensitiveInfo masked every URL, domain and IPv4 in upstream error messages, which garbled documentation links, email domains and dotted identifiers that clients need to read. It now takes the upstream addresses of the request: the channel base URL, the origin the gateway actually dialed (recorded in doRequest, DoWssRequest, the AWS SDK client and the Xunfei and Volcengine TTS dials, which covers absolute Advanced Custom routes), the channel proxy and the environment proxy. URLs and bare hostnames on those hosts and all IPv4 addresses are masked; other URLs and domains stay readable. With no upstream address no address is masked. Credentials are masked on every URL regardless of host: api_key values, query values and userinfo. The URL matcher also keeps quotes, colons and semicolons that follow a URL, keeps parentheses and apostrophes inside it, masks an IPv4 URL host whole and keeps a country-code TLD when the host has a port. * fix(relay): keep preset official endpoints readable in masked errors A channel on the base URL new-api presets for its type reaches a public official endpoint, so its address is not masked. Only addresses the admin configured are: a custom base URL, the origin of an Advanced Custom route, and the channel and environment proxies. Preset hosts are compared with their port so a configured localhost upstream stays apart from the Ollama preset. The origins recorded for the AWS SDK, Xunfei and Volcengine TTS dials and the plugin allowedHosts masking are dropped: they are all official endpoints. * fix(relay): mask IPv4 addresses and plugin allowed hosts in every error IPv4 addresses in network errors include the gateway's own address, so they are masked even when the channel uses a preset base URL. Task plugin build failures mask the plugin's allowedHosts again: an installed plugin may point them at private hosts. * fix(relay): mask bracketed IPv6 addresses in errors Go network errors write IPv6 addresses in brackets, as in "dial tcp [2001:db8::1]:443". Like IPv4, they are the resolved upstream or the gateway's own address, so they are masked on every error once the bracketed content parses as an IP.
89 lines
4.4 KiB
Go
89 lines
4.4 KiB
Go
package constant
|
|
|
|
type ContextKey string
|
|
|
|
const (
|
|
ContextKeyTokenCountMeta ContextKey = "token_count_meta"
|
|
ContextKeyPromptTokens ContextKey = "prompt_tokens"
|
|
ContextKeyEstimatedTokens ContextKey = "estimated_tokens"
|
|
|
|
ContextKeyOriginalModel ContextKey = "original_model"
|
|
ContextKeyRequestStartTime ContextKey = "request_start_time"
|
|
|
|
/* token related keys */
|
|
ContextKeyTokenUnlimited ContextKey = "token_unlimited_quota"
|
|
ContextKeyTokenKey ContextKey = "token_key"
|
|
ContextKeyTokenId ContextKey = "token_id"
|
|
ContextKeyTokenGroup ContextKey = "token_group"
|
|
ContextKeyOriginTasks ContextKey = "origin_tasks"
|
|
ContextKeyChannelConstraints ContextKey = "channel_constraints"
|
|
ContextKeyTokenModelLimitEnabled ContextKey = "token_model_limit_enabled"
|
|
ContextKeyTokenModelLimit ContextKey = "token_model_limit"
|
|
ContextKeyTokenCrossGroupRetry ContextKey = "token_cross_group_retry"
|
|
ContextKeyTokenAutoGroups ContextKey = "token_auto_groups"
|
|
|
|
/* channel related keys */
|
|
ContextKeyChannelId ContextKey = "channel_id"
|
|
ContextKeyChannelName ContextKey = "channel_name"
|
|
ContextKeyChannelCreateTime ContextKey = "channel_create_time"
|
|
ContextKeyChannelBaseUrl ContextKey = "base_url"
|
|
ContextKeyChannelType ContextKey = "channel_type"
|
|
ContextKeyChannelSetting ContextKey = "channel_setting"
|
|
ContextKeyChannelOtherSetting ContextKey = "channel_other_setting"
|
|
ContextKeyChannelParamOverride ContextKey = "param_override"
|
|
ContextKeyChannelHeaderOverride ContextKey = "header_override"
|
|
ContextKeyChannelOrganization ContextKey = "channel_organization"
|
|
ContextKeyChannelAutoBan ContextKey = "auto_ban"
|
|
ContextKeyChannelModelMapping ContextKey = "model_mapping"
|
|
ContextKeyChannelStatusCodeMapping ContextKey = "status_code_mapping"
|
|
ContextKeyChannelIsMultiKey ContextKey = "channel_is_multi_key"
|
|
ContextKeyChannelMultiKeyIndex ContextKey = "channel_multi_key_index"
|
|
ContextKeyChannelKey ContextKey = "channel_key"
|
|
|
|
// ContextKeyUpstreamRequestOrigin is the scheme and host of the last upstream
|
|
// request, which differs from the base URL for absolute Advanced Custom routes
|
|
// and for adaptors that build official endpoints themselves.
|
|
ContextKeyUpstreamRequestOrigin ContextKey = "upstream_request_origin"
|
|
|
|
ContextKeyAutoGroup ContextKey = "auto_group"
|
|
ContextKeyAutoGroupIndex ContextKey = "auto_group_index"
|
|
ContextKeyAutoGroupRetryIndex ContextKey = "auto_group_retry_index"
|
|
|
|
/* user related keys */
|
|
ContextKeyUserId ContextKey = "id"
|
|
ContextKeyUserSetting ContextKey = "user_setting"
|
|
ContextKeyUserQuota ContextKey = "user_quota"
|
|
ContextKeyUserStatus ContextKey = "user_status"
|
|
ContextKeyUserEmail ContextKey = "user_email"
|
|
ContextKeyUserGroup ContextKey = "user_group"
|
|
ContextKeyUsingGroup ContextKey = "group"
|
|
ContextKeyUserName ContextKey = "username"
|
|
|
|
ContextKeyLocalCountTokens ContextKey = "local_count_tokens"
|
|
|
|
ContextKeySystemPromptOverride ContextKey = "system_prompt_override"
|
|
|
|
// ContextKeyFileSourcesToCleanup stores file sources that need cleanup when request ends
|
|
ContextKeyFileSourcesToCleanup ContextKey = "file_sources_to_cleanup"
|
|
|
|
// ContextKeyAdminRejectReason stores an admin-only reject/block reason extracted from upstream responses.
|
|
// It is not returned to end users, but can be persisted into consume/error logs for debugging.
|
|
ContextKeyAdminRejectReason ContextKey = "admin_reject_reason"
|
|
|
|
// ContextKeyLanguage stores the user's language preference for i18n
|
|
ContextKeyLanguage ContextKey = "language"
|
|
ContextKeyIsStream ContextKey = "is_stream"
|
|
ContextKeyResponseStreamStatus ContextKey = "response_stream_status"
|
|
|
|
// ContextKeyAuditLogged marks that the current request has already recorded
|
|
// a manage/operation audit log inside the handler. When set, the admin-audit
|
|
// fallback in authHelper (finishAdminAudit) skips its record to avoid
|
|
// duplicate entries.
|
|
ContextKeyAuditLogged ContextKey = "audit_logged"
|
|
|
|
// ContextKeyTokenAuditParams contains only the API token operation's safe metadata.
|
|
ContextKeyTokenAuditParams ContextKey = "token_audit_params"
|
|
// ContextKeyTokenAuditSucceeded disambiguates token responses that exceed the audit buffer.
|
|
ContextKeyTokenAuditSucceeded ContextKey = "token_audit_succeeded"
|
|
)
|