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

feat: support doubao llm and embeding models (#4431)

Browse source
This commit is contained in:
sino 2024-05-16 11:41:24 +08:00 • committed by GitHub
commit 6e9066ebf4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 156 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api/tests/integration_tests/model_runtime/volcengine_maas/test_embedding.py
View file

@ -21,6 +21,7 @@ def test_validate_credentials():
'volc_access_key_id': 'INVALID',
'volc_secret_access_key': 'INVALID',
'endpoint_id': 'INVALID',
'base_model_name': 'Doubao-embedding',
}
)
@ -32,6 +33,7 @@ def test_validate_credentials():
'volc_access_key_id': os.environ.get('VOLC_API_KEY'),
'volc_secret_access_key': os.environ.get('VOLC_SECRET_KEY'),
'endpoint_id': os.environ.get('VOLC_EMBEDDING_ENDPOINT_ID'),
'base_model_name': 'Doubao-embedding',
},
)
@ -47,6 +49,7 @@ def test_invoke_model():
'volc_access_key_id': os.environ.get('VOLC_API_KEY'),
'volc_secret_access_key': os.environ.get('VOLC_SECRET_KEY'),
'endpoint_id': os.environ.get('VOLC_EMBEDDING_ENDPOINT_ID'),
'base_model_name': 'Doubao-embedding',
},
texts=[
"hello",
@ -71,6 +74,7 @@ def test_get_num_tokens():
'volc_access_key_id': os.environ.get('VOLC_API_KEY'),
'volc_secret_access_key': os.environ.get('VOLC_SECRET_KEY'),
'endpoint_id': os.environ.get('VOLC_EMBEDDING_ENDPOINT_ID'),
'base_model_name': 'Doubao-embedding',
},
texts=[
"hello",