fix: add type assertions for internal zod types

We export a lot of zod types in `AgentConnectionTypes` and `AgentInputContextTypes` that mirror Typescript types. We should either:
1. Make all the Typescript types just be inferred from the zod types
2. Add types assertions that they match so we get a build error if one is not updated