Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion consts/consts.go
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ const (

func ParseModelType(s string) ModelType {
switch s {
case "llm", "chat", "analysis":
case "llm", "chat", "analysis", "analysis-vl":
return ModelTypeChat
case "coder", "code":
return ModelTypeCoder
Expand Down
3 changes: 1 addition & 2 deletions test/ui_example/src/App.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -112,8 +112,7 @@ function App() {
<MenuItem value={'audio'}>音频模型</MenuItem>
<MenuItem value={'reranker'}>重排序模型</MenuItem>
<MenuItem value={'analysis'}>分析模型</MenuItem>
<MenuItem value={'monkeycode-chat'}>对话模型</MenuItem>
<MenuItem value={'monkeycode-code'}>代码补全模型</MenuItem>
<MenuItem value={'analysis-vl'}>图像分析模型</MenuItem>
</Select>
</FormControl>
</Box>
Expand Down
7 changes: 5 additions & 2 deletions ui/ModelModal/src/ModelModal.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@ const titleMap: Record<string, string> = {
['rerank']: '重排序模型',
['reranker']: '重排序模型',
['analysis']: '分析模型',
['analysis-vl']: '图像分析模型',
};

export const ModelModal: React.FC<ModelModalProps> = ({
Expand Down Expand Up @@ -477,6 +478,8 @@ export const ModelModal: React.FC<ModelModalProps> = ({
return it.rerank;
case 'analysis':
return it.analysis;
case 'analysis-vl':
return it.analysis_vl;
default:
return (
it.label === 'BaiZhiCloud' || it.label === 'Other'
Expand Down Expand Up @@ -1035,9 +1038,9 @@ export const ModelModal: React.FC<ModelModalProps> = ({
)}
</>
)}
{/* 高级设置部分 - 在选择了模型或者是其它供应商时显示,但不包括embedding、rerank、reranker、analysis类型 */}
{/* 高级设置部分 - 在选择了模型或者是其它供应商时显示,但不包括embedding、rerank、reranker、analysis-vl 类型 */}
{(modelUserList.length !== 0 || providerBrand === 'Other') &&
!['embedding', 'rerank', 'reranker'].includes(model_type) && (
!['embedding', 'rerank', 'reranker', 'analysis-vl'].includes(model_type) && (
<Box sx={{ mt: 2 }}>
<Accordion
sx={{
Expand Down
39 changes: 39 additions & 0 deletions ui/ModelModal/src/constants/providers.ts
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: true,
rerank: true,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://model-square.app.baizhi.cloud/token',
defaultBaseUrl: 'https://model-square.app.baizhi.cloud/v1',
},
Expand All @@ -29,6 +30,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://bigmodel.cn/usercenter/proj-mgmt/apikeys',
defaultBaseUrl: 'https://open.bigmodel.cn/api/paas/v4',
},
Expand All @@ -44,6 +46,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://platform.deepseek.com/api_keys',
defaultBaseUrl: 'https://api.deepseek.com/v1',
},
Expand All @@ -59,6 +62,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://console.cloud.tencent.com/hunyuan/api-key',
defaultBaseUrl: 'https://api.hunyuan.cloud.tencent.com/v1',
},
Expand All @@ -74,6 +78,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://bailian.console.aliyun.com/?tab=model#/api-key',
defaultBaseUrl: 'https://dashscope.aliyuncs.com/compatible-mode/v1',
},
Expand All @@ -89,6 +94,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://console.volcengine.com/ark/region:ark+cn-beijing/apiKey',
defaultBaseUrl: 'https://ark.cn-beijing.volces.com/api/v3',
},
Expand All @@ -104,6 +110,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://platform.openai.com/api-keys',
defaultBaseUrl: 'https://api.openai.com/v1',
},
Expand All @@ -119,6 +126,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://github.com/ollama/ollama/tree/main/docs',
defaultBaseUrl: 'http://172.17.0.1:11434',
},
Expand All @@ -134,6 +142,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://cloud.siliconflow.cn/me/account/ak',
defaultBaseUrl: 'https://api.siliconflow.cn/v1',
},
Expand All @@ -149,6 +158,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://platform.moonshot.cn/console/api-keys',
defaultBaseUrl: 'https://api.moonshot.cn/v1',
},
Expand All @@ -164,6 +174,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://portal.azure.com/#view/Microsoft_Azure_ProjectOxford/CognitiveServicesHub/~/OpenAI',
defaultBaseUrl: 'https://<resource_name>.openai.azure.com',
},
Expand All @@ -179,6 +190,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://aistudio.google.com/app/apikey',
defaultBaseUrl: 'https://generativelanguage.googleapis.com',
},
Expand All @@ -194,6 +206,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://portal.qiniu.com/ai-inference/api-key',
defaultBaseUrl: 'https://api.qnaigc.com/v1',
},
Expand Down Expand Up @@ -249,6 +262,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: true,
rerank: true,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://inference.readthedocs.io/zh-cn/v1.2.0/getting_started/installation.html#installation',
defaultBaseUrl: 'http://172.17.0.1:9997',
},
Expand All @@ -264,6 +278,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: true,
rerank: true,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://docs.gpustack.ai/latest/quickstart/',
defaultBaseUrl: 'http://172.17.0.1',
},
Expand All @@ -279,6 +294,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://platform.lingyiwanwu.com/apikeys',
defaultBaseUrl: 'https://api.lingyiwanwu.com/v1',
},
Expand Down Expand Up @@ -355,6 +371,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://huiju.ctyun.cn/service/serviceGroup',
defaultBaseUrl: 'https://wishub-x1.ctyun.cn/v1',
},
Expand All @@ -370,6 +387,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://console.cloud.tencent.com/lkeap/api',
defaultBaseUrl: 'https://api.lkeap.cloud.tencent.com/v1',
},
Expand All @@ -385,6 +403,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://console.bce.baidu.com/iam/#/iam/apikey/list',
defaultBaseUrl: 'https://qianfan.baidubce.com/v2',
},
Expand All @@ -400,6 +419,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: false,
analysis_vl: false,
modelDocumentUrl: 'https://modelscope.cn/my/myaccesstoken',
defaultBaseUrl: 'https://api-inference.modelscope.cn/v1',
},
Expand All @@ -415,6 +435,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://cloud.infini-ai.com/iam/secret/key',
defaultBaseUrl: 'https://cloud.infini-ai.com/maas/v1',
},
Expand All @@ -430,6 +451,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: false,
analysis_vl: false,
modelDocumentUrl: 'https://platform.stepfun.com/interface-key',
defaultBaseUrl: 'https://api.stepfun.com/v1',
},
Expand All @@ -445,6 +467,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: false,
analysis_vl: false,
modelDocumentUrl: 'https://maas.lanyun.net/#/system/apiKey',
defaultBaseUrl: 'https://maas-api.lanyun.net/v1',
},
Expand All @@ -460,6 +483,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: false,
analysis_vl: false,
modelDocumentUrl: ' https://www.alayanew.com/backend/register',
defaultBaseUrl: 'https://deepseek.alayanew.com/v1',
},
Expand All @@ -475,6 +499,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: false,
analysis_vl: false,
modelDocumentUrl: 'https://ppio.com/settings/key-management',
defaultBaseUrl: 'https://api.ppinfra.com/v3/openai',
},
Expand All @@ -490,6 +515,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: false,
analysis_vl: false,
modelDocumentUrl: 'https://aihubmix.com',
defaultBaseUrl: 'https://aihubmix.com/v1',
},
Expand All @@ -505,6 +531,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://one.ocoolai.com/token',
defaultBaseUrl: 'https://api.ocoolai.com/v1',
},
Expand All @@ -520,6 +547,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: false,
analysis_vl: false,
modelDocumentUrl: 'https://www.dmxapi.cn/register',
defaultBaseUrl: 'https://www.dmxapi.cn/v1',
},
Expand All @@ -535,6 +563,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://ai.burncloud.com/console/token',
defaultBaseUrl: 'https://ai.burncloud.com/v1',
},
Expand All @@ -560,6 +589,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://build.nvidia.com/?integrate_nim=true&hosted_api=true&modal=integrate-nim',
defaultBaseUrl: 'https://integrate.api.nvidia.com/v1',
},
Expand All @@ -575,6 +605,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: false,
analysis_vl: false,
modelDocumentUrl: 'https://tokenflux.ai/docs',
defaultBaseUrl: 'https://tokenflux.ai/v1',
},
Expand All @@ -590,6 +621,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://dash.302.ai/apis/list',
defaultBaseUrl: 'https://api.302.ai/v1',
},
Expand All @@ -605,6 +637,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: false,
analysis_vl: false,
modelDocumentUrl: 'https://cephalon.cloud/api',
defaultBaseUrl: 'https://cephalon.cloud/user-center/v1/model',
},
Expand All @@ -620,6 +653,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: false,
analysis_vl: false,
modelDocumentUrl: 'https://openrouter.ai/settings/keys',
defaultBaseUrl: 'https://openrouter.ai/api/v1',
},
Expand All @@ -635,6 +669,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: true,
analysis_vl: true,
modelDocumentUrl: 'https://app.fireworks.ai/settings/users/api-keys',
defaultBaseUrl: 'https://api.fireworks.ai/inference/v1',
},
Expand All @@ -650,6 +685,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: false,
analysis_vl: false,
modelDocumentUrl: 'https://console.mistral.ai/api-keys/',
defaultBaseUrl: 'https://api.mistral.ai/v1',
},
Expand All @@ -665,6 +701,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: false,
analysis_vl: false,
modelDocumentUrl: 'https://www.perplexity.ai/settings/api',
defaultBaseUrl: 'https://api.perplexity.ai',
},
Expand All @@ -680,6 +717,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: false,
rerank: false,
analysis: false,
analysis_vl: false,
modelDocumentUrl: 'https://app.hyperbolic.xyz/settings',
defaultBaseUrl: 'https://api.hyperbolic.xyz/v1',
},
Expand Down Expand Up @@ -715,6 +753,7 @@ export const DEFAULT_MODEL_PROVIDERS: ModelProviderMap = {
embedding: true,
rerank: true,
analysis: true,
analysis_vl: true,
modelDocumentUrl: '',
defaultBaseUrl: '',
},
Expand Down
1 change: 1 addition & 0 deletions ui/ModelModal/src/types/types.ts
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,7 @@ export interface ModelProviderConfig {
embedding: boolean;
rerank: boolean;
analysis: boolean;
analysis_vl: boolean;
modelDocumentUrl?: string;
defaultBaseUrl: string;
}
Expand Down