Add a bit more typing around json schemas for extension points

- Adds helper to go from type -> basic schema
- Uses schema -> type in one more place