update readme
Browse files
README.md
CHANGED
@@ -6855,22 +6855,25 @@ import requests
|
|
6855 |
from aws_requests_auth.boto_utils import BotoAWSRequestsAuth
|
6856 |
|
6857 |
region = "us-east-1"
|
6858 |
-
base_uri = f"
|
6859 |
batch_job_uri = f"https://{base_uri}/model-invocation-job/"
|
6860 |
|
6861 |
-
|
|
|
|
|
|
|
6862 |
payload = {
|
6863 |
"inputDataConfig": {
|
6864 |
"s3InputDataConfig": {
|
6865 |
-
"s3Uri": "s3://
|
6866 |
"s3InputFormat": "JSONL"
|
6867 |
}
|
6868 |
},
|
6869 |
-
"jobName": "embeddings-v2-
|
6870 |
"modelId": "amazon.titan-embed-text-v2",
|
6871 |
"outputDataConfig": {
|
6872 |
"s3OutputDataConfig": {
|
6873 |
-
"s3Uri": "s3://
|
6874 |
}
|
6875 |
},
|
6876 |
"roleArn": role_arn
|
|
|
6855 |
from aws_requests_auth.boto_utils import BotoAWSRequestsAuth
|
6856 |
|
6857 |
region = "us-east-1"
|
6858 |
+
base_uri = f"bedrock.{region}.amazonaws.com"
|
6859 |
batch_job_uri = f"https://{base_uri}/model-invocation-job/"
|
6860 |
|
6861 |
+
# For details on how to set up an IAM role for batch inference, see
|
6862 |
+
# https://docs.aws.amazon.com/bedrock/latest/userguide/batch-inference-permissions.html
|
6863 |
+
role_arn = "arn:aws:iam::111122223333:role/my-batch-inference-role"
|
6864 |
+
|
6865 |
payload = {
|
6866 |
"inputDataConfig": {
|
6867 |
"s3InputDataConfig": {
|
6868 |
+
"s3Uri": "s3://my-input-bucket/batch-input/",
|
6869 |
"s3InputFormat": "JSONL"
|
6870 |
}
|
6871 |
},
|
6872 |
+
"jobName": "embeddings-v2-batch-job",
|
6873 |
"modelId": "amazon.titan-embed-text-v2",
|
6874 |
"outputDataConfig": {
|
6875 |
"s3OutputDataConfig": {
|
6876 |
+
"s3Uri": "s3://my-output-bucket/batch-output/"
|
6877 |
}
|
6878 |
},
|
6879 |
"roleArn": role_arn
|