feat(provider): add baichuan provider
This commit is contained in:
parent
8bff4df722
commit
810c44f7fc
BIN
src/renderer/src/assets/images/models/baichuan.png
Normal file
BIN
src/renderer/src/assets/images/models/baichuan.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 3.1 KiB |
BIN
src/renderer/src/assets/images/providers/baichuan.png
Normal file
BIN
src/renderer/src/assets/images/providers/baichuan.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 2.6 KiB |
@ -340,5 +340,28 @@ export const SYSTEM_MODELS: Record<string, SystemModel[]> = {
|
|||||||
group: 'Gemma',
|
group: 'Gemma',
|
||||||
defaultEnabled: false
|
defaultEnabled: false
|
||||||
}
|
}
|
||||||
|
],
|
||||||
|
baichuan: [
|
||||||
|
{
|
||||||
|
id: 'Baichuan4',
|
||||||
|
provider: 'baichuan',
|
||||||
|
name: 'Baichuan4',
|
||||||
|
group: 'Baichuan4',
|
||||||
|
defaultEnabled: true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
id: 'Baichuan3-Turbo',
|
||||||
|
provider: 'baichuan',
|
||||||
|
name: 'Baichuan3 Turbo',
|
||||||
|
group: 'Baichuan3',
|
||||||
|
defaultEnabled: true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
id: 'Baichuan3-Turbo-128k',
|
||||||
|
provider: 'baichuan',
|
||||||
|
name: 'Baichuan3 Turbo 128k',
|
||||||
|
group: 'Baichuan3',
|
||||||
|
defaultEnabled: true
|
||||||
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
|
|||||||
@ -69,5 +69,12 @@ export const PROVIDER_CONFIG = {
|
|||||||
docs: 'https://github.com/ollama/ollama/tree/main/docs',
|
docs: 'https://github.com/ollama/ollama/tree/main/docs',
|
||||||
models: 'https://ollama.com/library'
|
models: 'https://ollama.com/library'
|
||||||
}
|
}
|
||||||
|
},
|
||||||
|
baichuan: {
|
||||||
|
websites: {
|
||||||
|
official: 'https://www.baichuan-ai.com/',
|
||||||
|
docs: 'https://platform.baichuan-ai.com/docs',
|
||||||
|
models: 'https://platform.baichuan-ai.com/price'
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@ -77,7 +77,8 @@ const resources = {
|
|||||||
yi: 'Yi',
|
yi: 'Yi',
|
||||||
zhipu: 'ZHIPU AI',
|
zhipu: 'ZHIPU AI',
|
||||||
groq: 'Groq',
|
groq: 'Groq',
|
||||||
ollama: 'Ollama'
|
ollama: 'Ollama',
|
||||||
|
baichuan: 'Baichuan'
|
||||||
},
|
},
|
||||||
settings: {
|
settings: {
|
||||||
title: 'Settings',
|
title: 'Settings',
|
||||||
@ -191,7 +192,8 @@ const resources = {
|
|||||||
yi: '零一万物',
|
yi: '零一万物',
|
||||||
zhipu: '智谱AI',
|
zhipu: '智谱AI',
|
||||||
groq: 'Groq',
|
groq: 'Groq',
|
||||||
ollama: 'Ollama'
|
ollama: 'Ollama',
|
||||||
|
baichuan: '百川'
|
||||||
},
|
},
|
||||||
settings: {
|
settings: {
|
||||||
title: '设置',
|
title: '设置',
|
||||||
|
|||||||
@ -7,6 +7,7 @@ import ZhipuProviderLogo from '@renderer/assets/images/providers/zhipu.png'
|
|||||||
import OllamaProviderLogo from '@renderer/assets/images/providers/ollama.png'
|
import OllamaProviderLogo from '@renderer/assets/images/providers/ollama.png'
|
||||||
import MoonshotProviderLogo from '@renderer/assets/images/providers/moonshot.jpeg'
|
import MoonshotProviderLogo from '@renderer/assets/images/providers/moonshot.jpeg'
|
||||||
import OpenRouterProviderLogo from '@renderer/assets/images/providers/openrouter.png'
|
import OpenRouterProviderLogo from '@renderer/assets/images/providers/openrouter.png'
|
||||||
|
import BaichuanProviderLogo from '@renderer/assets/images/providers/baichuan.png'
|
||||||
import ChatGPTModelLogo from '@renderer/assets/images/models/chatgpt.jpeg'
|
import ChatGPTModelLogo from '@renderer/assets/images/models/chatgpt.jpeg'
|
||||||
import ChatGLMModelLogo from '@renderer/assets/images/models/chatglm.jpeg'
|
import ChatGLMModelLogo from '@renderer/assets/images/models/chatglm.jpeg'
|
||||||
import DeepSeekModelLogo from '@renderer/assets/images/models/deepseek.png'
|
import DeepSeekModelLogo from '@renderer/assets/images/models/deepseek.png'
|
||||||
@ -17,6 +18,7 @@ import LlamaModelLogo from '@renderer/assets/images/models/llama.jpeg'
|
|||||||
import MixtralModelLogo from '@renderer/assets/images/models/mixtral.jpeg'
|
import MixtralModelLogo from '@renderer/assets/images/models/mixtral.jpeg'
|
||||||
import MoonshotModelLogo from '@renderer/assets/images/providers/moonshot.jpeg'
|
import MoonshotModelLogo from '@renderer/assets/images/providers/moonshot.jpeg'
|
||||||
import MicrosoftModelLogo from '@renderer/assets/images/models/microsoft.png'
|
import MicrosoftModelLogo from '@renderer/assets/images/models/microsoft.png'
|
||||||
|
import BaichuanModelLogo from '@renderer/assets/images/models/baichuan.png'
|
||||||
|
|
||||||
export function getProviderLogo(providerId: string) {
|
export function getProviderLogo(providerId: string) {
|
||||||
switch (providerId) {
|
switch (providerId) {
|
||||||
@ -38,6 +40,8 @@ export function getProviderLogo(providerId: string) {
|
|||||||
return MoonshotProviderLogo
|
return MoonshotProviderLogo
|
||||||
case 'openrouter':
|
case 'openrouter':
|
||||||
return OpenRouterProviderLogo
|
return OpenRouterProviderLogo
|
||||||
|
case 'baichuan':
|
||||||
|
return BaichuanProviderLogo
|
||||||
default:
|
default:
|
||||||
return undefined
|
return undefined
|
||||||
}
|
}
|
||||||
@ -55,7 +59,8 @@ export function getModelLogo(modelId: string) {
|
|||||||
mixtral: MixtralModelLogo,
|
mixtral: MixtralModelLogo,
|
||||||
mistral: MixtralModelLogo,
|
mistral: MixtralModelLogo,
|
||||||
moonshot: MoonshotModelLogo,
|
moonshot: MoonshotModelLogo,
|
||||||
phi: MicrosoftModelLogo
|
phi: MicrosoftModelLogo,
|
||||||
|
baichuan: BaichuanModelLogo
|
||||||
}
|
}
|
||||||
|
|
||||||
for (const key in logoMap) {
|
for (const key in logoMap) {
|
||||||
|
|||||||
@ -19,7 +19,7 @@ const persistedReducer = persistReducer(
|
|||||||
{
|
{
|
||||||
key: 'cherry-studio',
|
key: 'cherry-studio',
|
||||||
storage,
|
storage,
|
||||||
version: 9,
|
version: 10,
|
||||||
blacklist: ['runtime'],
|
blacklist: ['runtime'],
|
||||||
migrate
|
migrate
|
||||||
},
|
},
|
||||||
|
|||||||
@ -93,6 +93,15 @@ const initialState: LlmState = {
|
|||||||
models: [],
|
models: [],
|
||||||
isSystem: true,
|
isSystem: true,
|
||||||
enabled: false
|
enabled: false
|
||||||
|
},
|
||||||
|
{
|
||||||
|
id: 'baichuan',
|
||||||
|
name: 'BAICHUAN AI',
|
||||||
|
apiKey: '',
|
||||||
|
apiHost: 'https://api.baichuan-ai.com',
|
||||||
|
models: SYSTEM_MODELS.baichuan.filter((m) => m.defaultEnabled),
|
||||||
|
isSystem: true,
|
||||||
|
enabled: false
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
|
|||||||
@ -156,6 +156,27 @@ const migrate = createMigrate({
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
},
|
||||||
|
// @ts-ignore store type is unknown
|
||||||
|
'10': (state: RootState) => {
|
||||||
|
return {
|
||||||
|
...state,
|
||||||
|
llm: {
|
||||||
|
...state.llm,
|
||||||
|
providers: [
|
||||||
|
...state.llm.providers,
|
||||||
|
{
|
||||||
|
id: 'baichuan',
|
||||||
|
name: 'BAICHUAN AI',
|
||||||
|
apiKey: '',
|
||||||
|
apiHost: 'https://api.baichuan-ai.com',
|
||||||
|
models: SYSTEM_MODELS.baichuan.filter((m) => m.defaultEnabled),
|
||||||
|
isSystem: true,
|
||||||
|
enabled: false
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user