SOLVED

Hi all,

I have been trying for 1 day and am out of options, the documentation for the AWS Bedrock API is quite poor to be honest. I am invoking text-to-image Stability AI model from a python lambda function. I have tried my prompt and all the parameters from the AWS CLI and it works fine. but I keep getting the following response using the API: "HTTP Status Code: 200", but then when I see the contents of the botocore.response.StreamingBody object I get : {'Output': {'__type': 'com.amazon.coral.service#UnknownOperationException'}, 'Version': '1.0'}. At first I thought I was decoding the output Base64 incorrectly and tried different things to manipulate the object, but in the end I realized that this is the actual output that the model is giving me. What puzzles me is that I am getting an HTTP Status Code of 200 but then not getting the Base64 object as it should. Anyone has an idea?

I have tried with all the parameters for the model, without the parameters (they are all optional), with different text prompts, etc. Always the same response.

To give more context, here is my Bedrock Request:

bedrock_body = {'text_prompts': [{'text': 'Sri lanka tea plantation', 'weight': 1}]}        
response = invoke_bedrock(
            provider="stability",
            model_id="stable-diffusion-xl-v1",
            payload=json.dumps(bedrock_body),
            embeddings=false
        )

And this is the response:

{'ResponseMetadata': {'RequestId': '65578504-6360-496d-9786-adb135ae866c', 'HTTPStatusCode': 200, 'HTTPHeaders': {'date': 'Tue, 21 May 2024 18:54:15 GMT', 'content-type': 'application/json', 'content-length': '90', 'connection': 'keep-alive', 'x-amzn-requestid': '65578504-6360-496d-9786-adb135ae866c'}, 'RetryAttempts': 0}, 'contentType': 'application/json', 'body': <botocore.response.StreamingBody object at 0x7fe524a19cf0>}

After json_output = json.loads(response['body'].read())

I get:

json_output:  {'Output': {'__type': 'com.amazon.coral.service#UnknownOperationException'}, 'Version': '1.0'}