feat: add baichuan llm support (#1294)

Co-authored-by: zxhlyh <jasonapring2015@outlook.com>
2023-10-10 12:09:26 +08:00
parent 677aacc8e3
commit 1d4f019de4
9 changed files with 745 additions and 1 deletions
--- a/api/core/model_providers/rules/_providers.json
+++ b/api/core/model_providers/rules/_providers.json
@@ -7,10 +7,11 @@
  "spark",
  "wenxin",
  "zhipuai",
+  "baichuan",
  "chatglm",
  "replicate",
  "huggingface_hub",
  "xinference",
  "openllm",
  "localai"
-]
+]
--- a/api/core/model_providers/rules/baichuan.json
+++ b/api/core/model_providers/rules/baichuan.json
@@ -0,0 +1,15 @@
+{
+    "support_provider_types": [
+        "custom"
+    ],
+    "system_config": null,
+    "model_flexibility": "fixed",
+    "price_config": {
+        "baichuan2-53b": {
+            "prompt": "0.01",
+            "completion": "0.01",
+            "unit": "0.001",
+            "currency": "RMB"
+        }
+    }
+}