feat: Add support for AIHubMix topup and charge functionality

src/main/services/WindowService.ts

@@ -163,7 +163,11 @@ export class WindowService {
     mainWindow.webContents.setWindowOpenHandler((details) => {
       const { url } = details
 
-      const oauthProviderUrls = ['https://account.siliconflow.cn/oauth', 'https://aihubmix.com/oauth']
+      const oauthProviderUrls = [
+        'https://account.siliconflow.cn/oauth',
+        'https://aihubmix.com/oauth',
+        'https://aihubmix.com/topup'
+      ]
 
       if (oauthProviderUrls.some((link) => url.startsWith(link))) {
         return {

src/renderer/src/pages/settings/ProviderSettings/ProviderSetting.tsx

@@ -7,6 +7,7 @@ import {
   PlusOutlined,
   SettingOutlined
 } from '@ant-design/icons'
+import { HStack } from '@renderer/components/Layout'
 import ModelTags from '@renderer/components/ModelTags'
 import OAuthButton from '@renderer/components/OAuth/OAuthButton'
 import { EMBEDDING_REGEX, getModelLogo, REASONING_REGEX, VISION_REGEX } from '@renderer/config/models'
@@ -17,10 +18,11 @@ import { useProvider } from '@renderer/hooks/useProvider'
 import i18n from '@renderer/i18n'
 import { isOpenAIProvider } from '@renderer/providers/ProviderFactory'
 import { checkApi } from '@renderer/services/ApiService'
-import { isProviderSupportAuth } from '@renderer/services/ProviderService'
+import { isProviderSupportAuth, isProviderSupportCharge } from '@renderer/services/ProviderService'
 import { useAppDispatch } from '@renderer/store'
 import { setModel } from '@renderer/store/assistants'
 import { Model, ModelType, Provider } from '@renderer/types'
+import { aihubmixCharge } from '@renderer/utils/oauth'
 import { Avatar, Button, Card, Checkbox, Divider, Flex, Input, Popover, Space, Switch } from 'antd'
 import Link from 'antd/es/typography/Link'
 import { groupBy, isEmpty } from 'lodash'
@@ -270,9 +272,14 @@ const ProviderSetting: FC<Props> = ({ provider: _provider }) => {
       </Space.Compact>
       {apiKeyWebsite && (
         <SettingHelpTextRow style={{ justifyContent: 'space-between' }}>
-          <SettingHelpLink target="_blank" href={apiKeyWebsite}>
-            {t('settings.provider.get_api_key')}
-          </SettingHelpLink>
+          <HStack gap={10}>
+            <SettingHelpLink target="_blank" href={apiKeyWebsite}>
+              {t('settings.provider.get_api_key')}
+            </SettingHelpLink>
+            {isProviderSupportCharge(provider) && (
+              <SettingHelpLink onClick={aihubmixCharge}>{t('settings.provider.charge')}</SettingHelpLink>
+            )}
+          </HStack>
           <SettingHelpText>{t('settings.provider.api_key.tip')}</SettingHelpText>
         </SettingHelpTextRow>
       )}

src/renderer/src/providers/OpenAIProvider.ts

@@ -164,8 +164,8 @@ export default class OpenAIProvider extends BaseProvider {
     if (['o1', 'o1-2024-12-17'].includes(model.id) || model.id.startsWith('o3')) {
       systemMessage = {
         role: 'developer',
-        content: `Formatting re-enabled${systemMessage ? "\n" + systemMessage.content : ""}`
-      };
+        content: `Formatting re-enabled${systemMessage ? '\n' + systemMessage.content : ''}`
+      }
     }
 
     const userMessages: ChatCompletionMessageParam[] = []
@@ -199,9 +199,7 @@ export default class OpenAIProvider extends BaseProvider {
     // @ts-ignore key is not typed
     const stream = await this.sdk.chat.completions.create({
       model: model.id,
-      messages: [systemMessage, ...userMessages].filter(
-        Boolean
-      ) as ChatCompletionMessageParam[],
+      messages: [systemMessage, ...userMessages].filter(Boolean) as ChatCompletionMessageParam[],
       temperature: this.getTemperature(assistant, model),
       top_p: this.getTopP(assistant, model),
       max_tokens: maxTokens,

src/renderer/src/services/ProviderService.ts

@@ -19,3 +19,8 @@ export function isProviderSupportAuth(provider: Provider) {
   const supportProviders = ['silicon', 'aihubmix']
   return supportProviders.includes(provider.id)
 }
+
+export function isProviderSupportCharge(provider: Provider) {
+  const supportProviders = ['aihubmix']
+  return supportProviders.includes(provider.id)
+}

src/renderer/src/utils/oauth.ts

@@ -57,3 +57,12 @@ export const oauthWithAihubmix = async (setKey) => {
   window.removeEventListener('message', messageHandler)
   window.addEventListener('message', messageHandler)
 }
+
+export const aihubmixCharge = async () => {
+  const chargeUrl = `https://aihubmix.com/topup?client_id=cherry_studio_oauth&lang=${getLanguageCode()}&aff=SJyh`
+  window.open(
+    chargeUrl,
+    'oauth',
+    'width=720,height=900,toolbar=no,location=no,status=no,menubar=no,scrollbars=yes,resizable=yes,alwaysOnTop=yes,alwaysRaised=yes'
+  )
+}