class Multiwoven::Integrations::Source::AwsBedrockModel::Client

def format_request(model, payload)
  case model
  when *MISTRAL_AI_MODEL
    payload_request = JSON.parse(payload)
    prompt = payload_request["prompt"]
    payload_request["prompt"] = "<s>[INST] #{prompt} [/INST]" unless prompt.start_with?("<s>[INST]") && prompt.end_with?("[/INST]")
    payload_request.to_json
  else
    payload
  end
end