添加邀请返利开关；为所有模型路由添加claude-前缀

common/constants.go

@@ -124,6 +124,7 @@ var TelegramBotName = ""
 var QuotaForNewUser = 0
 var QuotaForInviter = 0
 var QuotaForInvitee = 0
+var AffiliateRewardsEnabled = true
 var ChannelDisableThreshold = 5.0
 var AutomaticDisableChannelEnabled = false
 var AutomaticEnableChannelEnabled = false

controller/misc.go

@@ -92,6 +92,7 @@ func GetStatus(c *gin.Context) {
 		"register_enabled":              common.RegisterEnabled,
 		"password_login_enabled":        common.PasswordLoginEnabled,
 		"password_register_enabled":     common.PasswordRegisterEnabled,
+		"affiliate_rewards_enabled":     common.AffiliateRewardsEnabled,
 		"default_use_auto_group":        setting.DefaultUseAutoGroup,
 
 		"usd_exchange_rate": operation_setting.USDExchangeRate,

controller/model.go

@@ -265,11 +265,15 @@ func ListModels(c *gin.Context, modelType int) {
 	switch modelType {
 	case constant.ChannelTypeAnthropic:
 		useranthropicModels := make([]dto.AnthropicModel, len(userOpenAiModels))
-		for i, model := range userOpenAiModels {
+		for i, m := range userOpenAiModels {
+			id := m.Id
+			if !strings.HasPrefix(id, "claude-") {
+				id = "claude-" + id
+			}
 			useranthropicModels[i] = dto.AnthropicModel{
-				ID:          model.Id,
-				CreatedAt:   time.Unix(int64(model.Created), 0).UTC().Format(time.RFC3339),
-				DisplayName: model.Id,
+				ID:          id,
+				CreatedAt:   time.Unix(int64(m.Created), 0).UTC().Format(time.RFC3339),
+				DisplayName: m.Id,
 				Type:        "model",
 			}
 		}
@@ -329,11 +333,19 @@ func EnabledListModels(c *gin.Context) {
 
 func RetrieveModel(c *gin.Context, modelType int) {
 	modelId := c.Param("model")
-	if aiModel, ok := openAIModelsMap[modelId]; ok {
+	lookupId := modelId
+	if modelType == constant.ChannelTypeAnthropic {
+		lookupId = resolveAnthropicModelID(modelId)
+	}
+	if aiModel, ok := openAIModelsMap[lookupId]; ok {
 		switch modelType {
 		case constant.ChannelTypeAnthropic:
+			id := aiModel.Id
+			if !strings.HasPrefix(id, "claude-") {
+				id = "claude-" + id
+			}
 			c.JSON(200, dto.AnthropicModel{
-				ID:          aiModel.Id,
+				ID:          id,
 				CreatedAt:   time.Unix(int64(aiModel.Created), 0).UTC().Format(time.RFC3339),
 				DisplayName: aiModel.Id,
 				Type:        "model",
@@ -353,3 +365,17 @@ func RetrieveModel(c *gin.Context, modelType int) {
 		})
 	}
 }
+
+func resolveAnthropicModelID(id string) string {
+	if !strings.HasPrefix(id, "claude-") {
+		return id
+	}
+	if _, ok := openAIModelsMap[id]; ok {
+		return id
+	}
+	stripped := id[len("claude-"):]
+	if stripped == "" {
+		return id
+	}
+	return stripped
+}

controller/status_affiliate_rewards_test.go

@@ -0,0 +1,74 @@
+package controller
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/QuantumNous/new-api/common"
+	"github.com/gin-gonic/gin"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestGetStatusIncludesAffiliateRewardsEnabled(t *testing.T) {
+	originalEnabled := common.AffiliateRewardsEnabled
+	originalOptionMap := common.OptionMap
+	t.Cleanup(func() {
+		common.AffiliateRewardsEnabled = originalEnabled
+		common.OptionMap = originalOptionMap
+	})
+
+	common.AffiliateRewardsEnabled = false
+	common.OptionMap = map[string]string{
+		"HeaderNavModules":    "[]",
+		"SidebarModulesAdmin": "[]",
+	}
+
+	gin.SetMode(gin.TestMode)
+	recorder := httptest.NewRecorder()
+	context, _ := gin.CreateTestContext(recorder)
+	context.Request = httptest.NewRequest(http.MethodGet, "/api/status", nil)
+
+	GetStatus(context)
+
+	require.Equal(t, http.StatusOK, recorder.Code)
+
+	var payload struct {
+		Success bool `json:"success"`
+		Data    struct {
+			AffiliateRewardsEnabled bool `json:"affiliate_rewards_enabled"`
+		} `json:"data"`
+	}
+	require.NoError(t, common.Unmarshal(recorder.Body.Bytes(), &payload))
+	require.True(t, payload.Success)
+	assert.False(t, payload.Data.AffiliateRewardsEnabled)
+}
+
+func TestGetTopUpInfoIncludesAffiliateRewardsEnabled(t *testing.T) {
+	originalEnabled := common.AffiliateRewardsEnabled
+	t.Cleanup(func() {
+		common.AffiliateRewardsEnabled = originalEnabled
+	})
+
+	common.AffiliateRewardsEnabled = false
+
+	gin.SetMode(gin.TestMode)
+	recorder := httptest.NewRecorder()
+	context, _ := gin.CreateTestContext(recorder)
+	context.Request = httptest.NewRequest(http.MethodGet, "/api/user/topup/info", nil)
+
+	GetTopUpInfo(context)
+
+	require.Equal(t, http.StatusOK, recorder.Code)
+
+	var payload struct {
+		Success bool `json:"success"`
+		Data    struct {
+			AffiliateRewardsEnabled bool `json:"affiliate_rewards_enabled"`
+		} `json:"data"`
+	}
+	require.NoError(t, common.Unmarshal(recorder.Body.Bytes(), &payload))
+	require.True(t, payload.Success)
+	assert.False(t, payload.Data.AffiliateRewardsEnabled)
+}

controller/topup.go

@@ -102,6 +102,7 @@ func GetTopUpInfo(c *gin.Context) {
 		"enable_waffo_topup":               enableWaffo,
 		"enable_waffo_pancake_topup":       enableWaffoPancake,
 		"enable_redemption":                complianceConfirmed,
+		"affiliate_rewards_enabled":        common.AffiliateRewardsEnabled,
 		"payment_compliance_confirmed":     complianceConfirmed,
 		"payment_compliance_terms_version": operation_setting.CurrentComplianceTermsVersion,
 		"waffo_pay_methods": func() interface{} {

middleware/distributor.go

@@ -39,6 +39,9 @@ func Distribute() func(c *gin.Context) {
 			abortWithOpenAiMessage(c, http.StatusBadRequest, i18n.T(c, i18n.MsgDistributorInvalidRequest, map[string]any{"Error": err.Error()}))
 			return
 		}
+		if shouldSelectChannel && strings.HasPrefix(c.Request.URL.Path, "/v1/messages") {
+			modelRequest.Model = resolveAnthropicModelForRouting(modelRequest.Model)
+		}
 		if ok {
 			id, err := strconv.Atoi(channelId.(string))
 			if err != nil {
@@ -170,6 +173,24 @@ func Distribute() func(c *gin.Context) {
 	}
 }
 
+// resolveAnthropicModelForRouting strips the `claude-` prefix from model names
+// that were disguised for Claude Code Desktop compatibility. Real Claude models
+// (e.g. claude-sonnet-4-6) are returned unchanged because they exist in the
+// ability system; disguised models (e.g. claude-gpt-4o) have their prefix stripped.
+func resolveAnthropicModelForRouting(modelName string) string {
+	if !strings.HasPrefix(modelName, "claude-") {
+		return modelName
+	}
+	if model.IsModelInAnyAbility(modelName) {
+		return modelName
+	}
+	stripped := modelName[len("claude-"):]
+	if stripped == "" {
+		return modelName
+	}
+	return stripped
+}
+
 // channelSupportsRequestPath reports whether a channel can serve the request path.
 // Only Advanced Custom (type 58) channels are path-checked; all other channel types
 // always pass. A type-58 channel is usable only when one of its routes matches.

model/ability.go

@@ -54,6 +54,22 @@ func GetEnabledModels() []string {
 	return models
 }
 
+func IsModelInAnyAbility(modelName string) bool {
+	if !common.MemoryCacheEnabled {
+		var count int64
+		DB.Model(&Ability{}).Where("model = ? and enabled = ?", modelName, true).Limit(1).Count(&count)
+		return count > 0
+	}
+	channelSyncLock.RLock()
+	defer channelSyncLock.RUnlock()
+	for _, modelMap := range group2model2channels {
+		if channels, ok := modelMap[modelName]; ok && len(channels) > 0 {
+			return true
+		}
+	}
+	return false
+}
+
 func GetAllEnableAbilities() []Ability {
 	var abilities []Ability
 	DB.Find(&abilities, "enabled = ?", true)

model/option.go

@@ -134,6 +134,7 @@ func InitOptionMap() {
 	common.OptionMap["QuotaForNewUser"] = strconv.Itoa(common.QuotaForNewUser)
 	common.OptionMap["QuotaForInviter"] = strconv.Itoa(common.QuotaForInviter)
 	common.OptionMap["QuotaForInvitee"] = strconv.Itoa(common.QuotaForInvitee)
+	common.OptionMap["AffiliateRewardsEnabled"] = strconv.FormatBool(common.AffiliateRewardsEnabled)
 	common.OptionMap["QuotaRemindThreshold"] = strconv.Itoa(common.QuotaRemindThreshold)
 	common.OptionMap["PreConsumedQuota"] = strconv.Itoa(common.PreConsumedQuota)
 	common.OptionMap["ModelRequestRateLimitCount"] = strconv.Itoa(setting.ModelRequestRateLimitCount)
@@ -351,6 +352,8 @@ func updateOptionMap(key string, value string) (err error) {
 			common.TaskEnabled = boolValue
 		case "DataExportEnabled":
 			common.DataExportEnabled = boolValue
+		case "AffiliateRewardsEnabled":
+			common.AffiliateRewardsEnabled = boolValue
 		case "DefaultCollapseSidebar":
 			common.DefaultCollapseSidebar = boolValue
 		case "MjNotifyEnabled":

model/option_affiliate_rewards_test.go

@@ -0,0 +1,26 @@
+package model
+
+import (
+	"testing"
+
+	"github.com/QuantumNous/new-api/common"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestUpdateOptionMapUpdatesAffiliateRewardsEnabled(t *testing.T) {
+	originalEnabled := common.AffiliateRewardsEnabled
+	originalOptionMap := common.OptionMap
+	t.Cleanup(func() {
+		common.AffiliateRewardsEnabled = originalEnabled
+		common.OptionMap = originalOptionMap
+	})
+
+	common.OptionMap = map[string]string{}
+	common.AffiliateRewardsEnabled = true
+
+	require.NoError(t, updateOptionMap("AffiliateRewardsEnabled", "false"))
+
+	assert.False(t, common.AffiliateRewardsEnabled)
+	assert.Equal(t, "false", common.OptionMap["AffiliateRewardsEnabled"])
+}

web/src/features/auth/types.ts

@@ -128,6 +128,7 @@ export interface SystemStatus {
     custom_currency_symbol?: string
     custom_currency_exchange_rate?: number
     demo_site_enabled?: boolean
+    affiliate_rewards_enabled?: boolean
     user_agreement_enabled?: boolean
     privacy_policy_enabled?: boolean
     oauth_register_enabled?: boolean
@@ -173,6 +174,7 @@ export interface SystemStatus {
   custom_currency_symbol?: string
   custom_currency_exchange_rate?: number
   demo_site_enabled?: boolean
+  affiliate_rewards_enabled?: boolean
   user_agreement_enabled?: boolean
   privacy_policy_enabled?: boolean
   oauth_register_enabled?: boolean

web/src/features/system-settings/billing/index.tsx

@@ -29,6 +29,7 @@ const defaultBillingSettings: BillingSettings = {
   PreConsumedQuota: 0,
   QuotaForInviter: 0,
   QuotaForInvitee: 0,
+  AffiliateRewardsEnabled: true,
   TopUpLink: '',
   'general_setting.docs_link': '',
   'quota_setting.enable_free_model_pre_consume': true,

web/src/features/system-settings/billing/section-registry.tsx

@@ -63,6 +63,7 @@ const BILLING_SECTIONS = [
           PreConsumedQuota: settings.PreConsumedQuota,
           QuotaForInviter: settings.QuotaForInviter,
           QuotaForInvitee: settings.QuotaForInvitee,
+          AffiliateRewardsEnabled: settings.AffiliateRewardsEnabled,
           TopUpLink: settings.TopUpLink,
           general_setting: {
             docs_link: settings['general_setting.docs_link'],

web/src/features/system-settings/general/quota-settings-section.tsx

@@ -55,6 +55,7 @@ const quotaSchema = z.object({
   PreConsumedQuota: z.coerce.number().min(0),
   QuotaForInviter: z.coerce.number().min(0),
   QuotaForInvitee: z.coerce.number().min(0),
+  AffiliateRewardsEnabled: z.boolean(),
   TopUpLink: z.string(),
   general_setting: z.object({
     docs_link: z.string(),
@@ -237,6 +238,32 @@ export function QuotaSettingsSection({
               )}
             />
 
+            <SettingsFormGridItem span='full'>
+              <FormField
+                control={form.control}
+                name='AffiliateRewardsEnabled'
+                render={({ field }) => (
+                  <SettingsSwitchItem>
+                    <SettingsSwitchContent>
+                      <FormLabel>{t('Enable Invitation Rewards')}</FormLabel>
+                      <FormDescription>
+                        {t(
+                          'When disabled, users will not see the invitation rewards panel.'
+                        )}
+                      </FormDescription>
+                    </SettingsSwitchContent>
+                    <FormControl>
+                      <Switch
+                        checked={field.value}
+                        onCheckedChange={field.onChange}
+                        disabled={updateOption.isPending}
+                      />
+                    </FormControl>
+                  </SettingsSwitchItem>
+                )}
+              />
+            </SettingsFormGridItem>
+
             <SettingsFormGridItem span='full'>
               <FormField
                 control={form.control}

web/src/features/system-settings/types.ts

@@ -251,6 +251,7 @@ export type BillingSettings = {
   PreConsumedQuota: number
   QuotaForInviter: number
   QuotaForInvitee: number
+  AffiliateRewardsEnabled: boolean
   TopUpLink: string
   'general_setting.docs_link': string
   'quota_setting.enable_free_model_pre_consume': boolean

web/src/features/wallet/hooks/use-affiliate.ts

@@ -30,15 +30,27 @@ import { generateAffiliateLink } from '../lib'
 // Affiliate Hook
 // ============================================================================
 
-export function useAffiliate() {
+interface UseAffiliateOptions {
+  enabled?: boolean
+}
+
+export function useAffiliate(options: UseAffiliateOptions = {}) {
+  const { enabled = true } = options
   const [affiliateCode, setAffiliateCode] = useState<string>('')
   const [affiliateLink, setAffiliateLink] = useState<string>('')
-  const [loading, setLoading] = useState(true)
+  const [loading, setLoading] = useState(enabled)
   const [transferring, setTransferring] = useState(false)
   const { copyToClipboard } = useCopyToClipboard()
 
   // Fetch affiliate code
   const fetchAffiliateCode = useCallback(async () => {
+    if (!enabled) {
+      setAffiliateCode('')
+      setAffiliateLink('')
+      setLoading(false)
+      return
+    }
+
     try {
       setLoading(true)
       const response = await getAffiliateCode()
@@ -54,7 +66,7 @@ export function useAffiliate() {
     } finally {
       setLoading(false)
     }
-  }, [])
+  }, [enabled])
 
   // Copy affiliate link
   const copyAffiliateLink = useCallback(() => {
@@ -84,8 +96,15 @@ export function useAffiliate() {
   }, [])
 
   useEffect(() => {
-    fetchAffiliateCode()
-  }, [fetchAffiliateCode])
+    if (enabled) {
+      fetchAffiliateCode()
+      return
+    }
+
+    setAffiliateCode('')
+    setAffiliateLink('')
+    setLoading(false)
+  }, [enabled, fetchAffiliateCode])
 
   return {
     affiliateCode,