Add state agent

[rayrayraykk]

---

name: Add state agent

Description

Add state agent

Checklist

Please check the following items before code is ready to be reviewed.

• [X] Code has passed all tests
• [X] Docstrings have been added/updated in Google Style
• [X] Documentation has been updated
• [X] Code is ready for review