chore: update maas model provider description (langgenius#5056)

dengpeng · Jun 16, 2024 · 4ad48b0 · 4ad48b0
1 parent 5b0f723
commit 4ad48b0
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 2 deletions.
diff --git a/api/core/model_runtime/model_providers/volcengine_maas/llm/models.py b/api/core/model_runtime/model_providers/volcengine_maas/llm/models.py
@@ -69,4 +69,24 @@
             'mode': 'chat',
         }
     },
+    'Llama3-8B': {
+         'req_params': {
+            'max_prompt_tokens': 8192,
+            'max_new_tokens': 8192,
+        },
+        'model_properties': {
+            'context_size': 8192,
+            'mode': 'chat',
+        }
+    },
+    'Llama3-70B': {
+         'req_params': {
+            'max_prompt_tokens': 8192,
+            'max_new_tokens': 8192,
+        },
+        'model_properties': {
+            'context_size': 8192,
+            'mode': 'chat',
+        }
+    }
 }
diff --git a/api/core/model_runtime/model_providers/volcengine_maas/volcengine_maas.yaml b/api/core/model_runtime/model_providers/volcengine_maas/volcengine_maas.yaml
@@ -2,7 +2,8 @@ provider: volcengine_maas
 label:
   en_US: Volcengine
 description:
-  en_US: Volcengine MaaS models.
+  en_US: Volcengine Ark models.
+  zh_Hans: 火山方舟提供的模型，例如 Doubao-pro-4k、Doubao-pro-32k 和 Doubao-pro-128k。
 icon_small:
   en_US: icon_s_en.svg
 icon_large:
@@ -12,6 +13,7 @@ background: "#F9FAFB"
 help:
   title:
     en_US: Get your Access Key and Secret Access Key from Volcengine Console
+    zh_Hans: 从火山引擎控制台获取您的 Access Key 和 Secret Access Key
   url:
     en_US: https://console.volcengine.com/iam/keymanage/
 supported_model_types:
@@ -50,7 +52,7 @@ model_credential_schema:
       required: true
       label:
         en_US: Volcengine Region
-        zh_Hans: 火山引擎地区
+        zh_Hans: 火山引擎地域
       type: text-input
       default: cn-beijing
       placeholder:
@@ -124,6 +126,18 @@ model_credential_schema:
           show_on:
             - variable: __model_type
               value: llm
+        - label:
+            en_US: Llama3-8B
+          value: Llama3-8B
+          show_on:
+            - variable: __model_type
+              value: llm
+        - label:
+            en_US: Llama3-70B
+          value: Llama3-70B
+          show_on:
+            - variable: __model_type
+              value: llm
         - label:
             en_US: Doubao-embedding
           value: Doubao-embedding