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

2023-11-21 15:06:47 +08:00
parent 0d9ce1bab0
commit c9368925a3
18 changed files with 78 additions and 10 deletions
--- a/api/core/model_providers/rules/anthropic.json
+++ b/api/core/model_providers/rules/anthropic.json
@@ -12,6 +12,9 @@
        "quota_limit": 0
    },
    "model_flexibility": "fixed",
+    "supported_model_types": [
+        "text-generation"
+    ],
    "price_config": {
        "claude-instant-1": {
            "prompt": "1.63",
--- a/api/core/model_providers/rules/azure_openai.json
+++ b/api/core/model_providers/rules/azure_openai.json
@@ -4,6 +4,10 @@
    ],
    "system_config": null,
    "model_flexibility": "configurable",
+    "supported_model_types": [
+        "text-generation",
+        "embeddings"
+    ],
    "price_config":{
        "gpt-4": {
            "prompt": "0.03",
--- a/api/core/model_providers/rules/baichuan.json
+++ b/api/core/model_providers/rules/baichuan.json
@@ -4,6 +4,9 @@
    ],
    "system_config": null,
    "model_flexibility": "fixed",
+    "supported_model_types": [
+        "text-generation"
+    ],
    "price_config": {
        "baichuan2-53b": {
            "prompt": "0.01",
--- a/api/core/model_providers/rules/chatglm.json
+++ b/api/core/model_providers/rules/chatglm.json
@@ -3,5 +3,8 @@
        "custom"
    ],
    "system_config": null,
-    "model_flexibility": "fixed"
+    "model_flexibility": "fixed",
+    "supported_model_types": [
+        "text-generation"
+    ]
 }
--- a/api/core/model_providers/rules/cohere.json
+++ b/api/core/model_providers/rules/cohere.json
@@ -3,5 +3,8 @@
        "custom"
    ],
    "system_config": null,
-    "model_flexibility": "fixed"
+    "model_flexibility": "fixed",
+    "supported_model_types": [
+        "reranking"
+    ]
 }
--- a/api/core/model_providers/rules/huggingface_hub.json
+++ b/api/core/model_providers/rules/huggingface_hub.json
@@ -3,5 +3,9 @@
        "custom"
    ],
    "system_config": null,
-    "model_flexibility": "configurable"
+    "model_flexibility": "configurable",
+    "supported_model_types": [
+        "text-generation",
+        "embeddings"
+    ]
 }
--- a/api/core/model_providers/rules/localai.json
+++ b/api/core/model_providers/rules/localai.json
@@ -3,5 +3,9 @@
        "custom"
    ],
    "system_config": null,
-    "model_flexibility": "configurable"
+    "model_flexibility": "configurable",
+    "supported_model_types": [
+        "text-generation",
+        "embeddings"
+    ]
 }
--- a/api/core/model_providers/rules/minimax.json
+++ b/api/core/model_providers/rules/minimax.json
@@ -10,6 +10,10 @@
        "quota_unit": "tokens"
    },
    "model_flexibility": "fixed",
+    "supported_model_types": [
+        "text-generation",
+        "embeddings"
+    ],
    "price_config": {
        "abab5.5-chat": {
            "prompt": "0.015",
--- a/api/core/model_providers/rules/openai.json
+++ b/api/core/model_providers/rules/openai.json
@@ -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",
--- a/api/core/model_providers/rules/openllm.json
+++ b/api/core/model_providers/rules/openllm.json
@@ -3,5 +3,9 @@
        "custom"
    ],
    "system_config": null,
-    "model_flexibility": "configurable"
+    "model_flexibility": "configurable",
+    "supported_model_types": [
+        "text-generation",
+        "embeddings"
+    ]
 }
--- a/api/core/model_providers/rules/replicate.json
+++ b/api/core/model_providers/rules/replicate.json
@@ -3,5 +3,9 @@
        "custom"
    ],
    "system_config": null,
-    "model_flexibility": "configurable"
+    "model_flexibility": "configurable",
+    "supported_model_types": [
+        "text-generation",
+        "embeddings"
+    ]
 }
--- a/api/core/model_providers/rules/spark.json
+++ b/api/core/model_providers/rules/spark.json
@@ -10,6 +10,9 @@
        "quota_unit": "tokens"
    },
    "model_flexibility": "fixed",
+    "supported_model_types": [
+        "text-generation"
+    ],
    "price_config": {
        "spark": {
            "prompt": "0.18",
--- a/api/core/model_providers/rules/tongyi.json
+++ b/api/core/model_providers/rules/tongyi.json
@@ -4,6 +4,9 @@
    ],
    "system_config": null,
    "model_flexibility": "fixed",
+    "supported_model_types": [
+        "text-generation"
+    ],
    "price_config": {
        "qwen-turbo": {
            "prompt": "0.012",
--- a/api/core/model_providers/rules/wenxin.json
+++ b/api/core/model_providers/rules/wenxin.json
@@ -4,6 +4,9 @@
    ],
    "system_config": null,
    "model_flexibility": "fixed",
+    "supported_model_types": [
+        "text-generation"
+    ],
    "price_config": {
        "ernie-bot-4": {
            "prompt": "0",
--- a/api/core/model_providers/rules/xinference.json
+++ b/api/core/model_providers/rules/xinference.json
@@ -3,5 +3,9 @@
        "custom"
    ],
    "system_config": null,
-    "model_flexibility": "configurable"
+    "model_flexibility": "configurable",
+    "supported_model_types": [
+        "text-generation",
+        "embeddings"
+    ]
 }
--- a/api/core/model_providers/rules/zhipuai.json
+++ b/api/core/model_providers/rules/zhipuai.json
@@ -10,6 +10,10 @@
        "quota_unit": "tokens"
    },
    "model_flexibility": "fixed",
+    "supported_model_types": [
+        "text-generation",
+        "embeddings"
+    ],
    "price_config": {
        "chatglm_turbo": {
            "prompt": "0.005",