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

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
parent dc847ba145
commit 4a026fa352
15 changed files with 913 additions and 0 deletions

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={}
)