AmazonBedrockChatGenerator - set max_tokens

[moshere-cirm]

Hi,
I'm using AmazonBedrockChatGenerator and I would like to set max_tokens (to enable large output).
I tried this code

chat_generator = AmazonBedrockChatGenerator(
    model="anthropic.claude-3-5-sonnet-20240620-v1:0",
    boto3_config=boto3_config_dict,  # ✅ Pass the dictionary, not the Config object,
    generation_kwargs={"max_tokens": 512000}
)

However seems this doesn't help.
Please suggest how I can set the max_tokens for output.
Thank you!

[moshere-cirm]

Resolved using

chat_generator = AmazonBedrockChatGenerator(
    model="anthropic.claude-3-5-sonnet-20240620-v1:0",
    boto3_config=boto3_config_dict,  # ✅ Pass the dictionary, not the Config object,
    generation_kwargs={"maxTokens": 8192}  # ✅ Add generation parameters
)