jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add supported_model_types field and filter in provider list (#1581)

Browse source
This commit is contained in:
takatost 2023-11-21 15:06:47 +08:00 • committed by GitHub
commit c9368925a3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
18 changed files with 78 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

3
api/core/model_providers/rules/anthropic.json
View file

@ -12,6 +12,9 @@
"quota_limit": 0
},
"model_flexibility": "fixed",
"supported_model_types": [
"text-generation"
],
"price_config": {
"claude-instant-1": {
"prompt": "1.63",

4
api/core/model_providers/rules/azure_openai.json
View file

@ -4,6 +4,10 @@
],
"system_config": null,
"model_flexibility": "configurable",
"supported_model_types": [
"text-generation",
"embeddings"
],
"price_config":{
"gpt-4": {
"prompt": "0.03",

3
api/core/model_providers/rules/baichuan.json
View file

@ -4,6 +4,9 @@
],
"system_config": null,
"model_flexibility": "fixed",
"supported_model_types": [
"text-generation"
],
"price_config": {
"baichuan2-53b": {
"prompt": "0.01",

5
api/core/model_providers/rules/chatglm.json
View file

@ -3,5 +3,8 @@
"custom"
],
"system_config": null,
"model_flexibility": "fixed"
"model_flexibility": "fixed",
"supported_model_types": [
"text-generation"
]
}

5
api/core/model_providers/rules/cohere.json
View file

@ -3,5 +3,8 @@
"custom"
],
"system_config": null,
"model_flexibility": "fixed"
"model_flexibility": "fixed",
"supported_model_types": [
"reranking"
]
}

6
api/core/model_providers/rules/huggingface_hub.json
View file

@ -3,5 +3,9 @@
"custom"
],
"system_config": null,
"model_flexibility": "configurable"
"model_flexibility": "configurable",
"supported_model_types": [
"text-generation",
"embeddings"
]
}

6
api/core/model_providers/rules/localai.json
View file

@ -3,5 +3,9 @@
"custom"
],
"system_config": null,
"model_flexibility": "configurable"
"model_flexibility": "configurable",
"supported_model_types": [
"text-generation",
"embeddings"
]
}

4
api/core/model_providers/rules/minimax.json
View file

@ -10,6 +10,10 @@
"quota_unit": "tokens"
},
"model_flexibility": "fixed",
"supported_model_types": [
"text-generation",
"embeddings"
],
"price_config": {
"abab5.5-chat": {
"prompt": "0.015",

6
api/core/model_providers/rules/openai.json
View file

@ -11,6 +11,12 @@
"quota_limit": 200
},
"model_flexibility": "fixed",
"supported_model_types": [
"text-generation",
"embeddings",
"speech2text",
"moderation"
],
"price_config": {
"gpt-4": {
"prompt": "0.03",

6
api/core/model_providers/rules/openllm.json
View file

@ -3,5 +3,9 @@
"custom"
],
"system_config": null,
"model_flexibility": "configurable"
"model_flexibility": "configurable",
"supported_model_types": [
"text-generation",
"embeddings"
]
}

6
api/core/model_providers/rules/replicate.json
View file

@ -3,5 +3,9 @@
"custom"
],
"system_config": null,
"model_flexibility": "configurable"
"model_flexibility": "configurable",
"supported_model_types": [
"text-generation",
"embeddings"
]
}

3
api/core/model_providers/rules/spark.json
View file

@ -10,6 +10,9 @@
"quota_unit": "tokens"
},
"model_flexibility": "fixed",
"supported_model_types": [
"text-generation"
],
"price_config": {
"spark": {
"prompt": "0.18",

3
api/core/model_providers/rules/tongyi.json
View file

@ -4,6 +4,9 @@
],
"system_config": null,
"model_flexibility": "fixed",
"supported_model_types": [
"text-generation"
],
"price_config": {
"qwen-turbo": {
"prompt": "0.012",

3
api/core/model_providers/rules/wenxin.json
View file

@ -4,6 +4,9 @@
],
"system_config": null,
"model_flexibility": "fixed",
"supported_model_types": [
"text-generation"
],
"price_config": {
"ernie-bot-4": {
"prompt": "0",

6
api/core/model_providers/rules/xinference.json
View file

@ -3,5 +3,9 @@
"custom"
],
"system_config": null,
"model_flexibility": "configurable"
"model_flexibility": "configurable",
"supported_model_types": [
"text-generation",
"embeddings"
]
}

4
api/core/model_providers/rules/zhipuai.json
View file

@ -10,6 +10,10 @@
"quota_unit": "tokens"
},
"model_flexibility": "fixed",
"supported_model_types": [
"text-generation",
"embeddings"
],
"price_config": {
"chatglm_turbo": {
"prompt": "0.005",