I may be late but I tried different approaches but the best I could do is solve it manuelly simply using useState:
const [isSubMenuOpen, setIsSubMenuOpen] = useState(true);
<SubMenu
open={isSubMenuOpen}
onOpenChange={() => setIsSubMenuOpen(true)}