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

def check_connection(connection_config)
  connection_config = connection_config.with_indifferent_access
  create_connection(connection_config)
  model = connection_config[:inference_profile] || connection_config[:model_id]
  payload = format_request(model, connection_config[:request_format])
  @client_runtime.invoke_model(
    model_id: model,
    content_type: "application/json",
    accept: "application/json",
    body: payload
  )
  success_status
rescue StandardError => e
  ConnectionStatus.new(status: ConnectionStatusType["failed"], message: e.message).to_multiwoven_message
end