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

This commit is contained in:
sino
2024-05-16 11:41:24 +08:00
committed by GitHub
parent dd94931116
commit 6e9066ebf4
5 changed files with 156 additions and 8 deletions

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",