Spaces:
Build error
Build error
File size: 1,591 Bytes
d660b02 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
from loguru import logger
try:
import boto3
from botocore.exceptions import ClientError
except ModuleNotFoundError:
logger.warning("Couldn't load AWS or SageMaker imports. Run 'poetry install --with aws' to support AWS.")
from llm_engineering.settings import settings
class ResourceManager:
def __init__(self) -> None:
self.sagemaker_client = boto3.client(
"sagemaker",
region_name=settings.AWS_REGION,
aws_access_key_id=settings.AWS_ACCESS_KEY,
aws_secret_access_key=settings.AWS_SECRET_KEY,
)
def endpoint_config_exists(self, endpoint_config_name: str) -> bool:
"""Check if the SageMaker endpoint configuration exists."""
try:
self.sagemaker_client.describe_endpoint_config(EndpointConfigName=endpoint_config_name)
logger.info(f"Endpoint configuration '{endpoint_config_name}' exists.")
return True
except ClientError:
logger.info(f"Endpoint configuration '{endpoint_config_name}' does not exist.")
return False
def endpoint_exists(self, endpoint_name: str) -> bool:
"""Check if the SageMaker endpoint exists."""
try:
self.sagemaker_client.describe_endpoint(EndpointName=endpoint_name)
logger.info(f"Endpoint '{endpoint_name}' exists.")
return True
except self.sagemaker_client.exceptions.ResourceNotFoundException:
logger.info(f"Endpoint '{endpoint_name}' does not exist.")
return False
|