estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat(api): support wenxin bge-large and tao embedding model. (#7393)

Browse Source
This commit is contained in:
Chengyu Yan
2024-08-19 22:25:09 +08:00
committed by GitHub
parent 31f9977411
commit 1f944c6eeb
5 changed files with 89 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/core/model_runtime/model_providers/wenxin/_common.py
View File

@@ -118,6 +118,9 @@ class _CommonWenxin:
'ernie-4.0-turbo-8k-preview': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-4.0-turbo-8k-preview',
'yi_34b_chat': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/yi_34b_chat',
'embedding-v1': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/embeddings/embedding-v1',
'bge-large-en': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/embeddings/bge_large_en',
'bge-large-zh': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/embeddings/bge_large_zh',
'tao-8k': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/embeddings/tao_8k',
}
function_calling_supports = [

9
api/core/model_runtime/model_providers/wenxin/text_embedding/bge-large-en.yaml Normal file
View File

@@ -0,0 +1,9 @@
model: bge-large-en
model_type: text-embedding
model_properties:
context_size: 512
max_chunks: 16
pricing:
input: '0.0005'
unit: '0.001'
currency: RMB

9
api/core/model_runtime/model_providers/wenxin/text_embedding/bge-large-zh.yaml Normal file
View File

@@ -0,0 +1,9 @@
model: bge-large-zh
model_type: text-embedding
model_properties:
context_size: 512
max_chunks: 16
pricing:
input: '0.0005'
unit: '0.001'
currency: RMB

9
api/core/model_runtime/model_providers/wenxin/text_embedding/tao-8k.yaml Normal file
View File

@@ -0,0 +1,9 @@
model: tao-8k
model_type: text-embedding
model_properties:
context_size: 8192
max_chunks: 1
pricing:
input: '0.0005'
unit: '0.001'
currency: RMB