refactor: Use Pydantic models instead of custom dict representations

refactor: Use Pydantic models instead of custom dict representations #921