コードブロックでの行折り返しで横スクロールバーを回避

コードブロックで水平スクロールバーを避けるために、文字折り返しを許可することは可能でしょうか?

さらに、特定の言語のコードブロックにのみ影響するように設定できれば、より良いです。

.lang-jai.hljs {
  /* 垂直スクロールバーを避けるために max-height を無効化。コードブロックは常に完全に表示されます */
  max-height: none;

  /* 水平スクロールバーを避けるために jai コードで単語折り返しを有効化 */
  overflow-x: auto; 
  white-space: pre-wrap;
}