[WIP] Properly handle model fragments in codemod

[richgt]

If this PR updates API docs, preview them by:

• install bun (if needed)
• install volta and configure it for pnpm (if needed)
• run pnpm install in the root (if needed)
• run pnpm preview in the root

---

• Read the full contributing documentation
• If you do not have permission to add labels or run the test-suite in CI, a team member will do this for you.