AI アシスタントを適用し、Python の関数に型アノテーションを追加します。