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

Enhancement: add model provider - Amazon Sagemaker (#6255)

Browse source 
Co-authored-by: Yuanbo Li <ybalbert@amazon.com>
Co-authored-by: crazywoola <427733928@qq.com>
This commit is contained in:
ybalbert001 2024-07-18 19:32:31 +08:00 • committed by GitHub
commit 4a026fa352
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
15 changed files with 913 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

19
api/tests/integration_tests/model_runtime/sagemaker/test_provider.py Normal file
View file

@ -0,0 +1,19 @@
import os
import pytest
from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.model_runtime.model_providers.sagemaker.sagemaker import SageMakerProvider
def test_validate_provider_credentials():
provider = SageMakerProvider()
with pytest.raises(CredentialsValidateFailedError):
provider.validate_provider_credentials(
credentials={}
)
provider.validate_provider_credentials(
credentials={}
)