Can you guys append type annotation to the key classes & functions, it makes IDE stuck when chasing the source code

Like the example snip for the official doc site below
<img width="685" alt="Image" src="https://github.com/user-attachments/assets/d4e1ee1f-517c-43e0-b8ff-3322df92a56f" />