diff --git a/utils/bedrock.py b/utils/bedrock.py index bf16c7a5..1069e146 100644 --- a/utils/bedrock.py +++ b/utils/bedrock.py @@ -2,6 +2,7 @@ # SPDX-License-Identifier: MIT-0 """Helper utilities for working with Amazon Bedrock from Python notebooks""" # Python Built-Ins: +from enum import Enum import json import os from time import sleep @@ -9,9 +10,8 @@ # External Dependencies: import boto3 -from pydantic import root_validator -from enum import Enum from botocore.config import Config +from pydantic import root_validator def get_bedrock_client(