Bedrock optimizer model?

[austinmw]

Can I use a Bedrock LLM for the optimizer?

[austinmw]

Update:

I was almost able to use the library as-is by setting optimizer like this:

  "optimizer": {
    "model": {
      "model_provider": "bedrock_converse",
      "model": "anthropic.claude-3-sonnet-20240229-v1:0",
      "max_tokens": 4096
    }

However, the _ensure_stricty seems to break Bedrock functionality by attempting to pass strict parameter to Bedrock, where this parameter doesn't exist? I'm using a StructuredPrompt. Commenting out this function allowed Bedrock models to work successfully for me.