Fix : short_term_memory with bedrock - using user defined model(when passed as attribute) rather than default#1959