@DouweM can you take a first look before I proceed with the other modules? Pulled type into top-level module, updated imports, updated examples. Excluded: 2 deprecated types, these are still only importable from pydantic_ai.messages.

@moritzwilksch Looks great! Can you replace from pydantic_ai.messages import with from pydantic_ai import inside the docs directory as well, and then run make update-examples?

will do! do you prefer separate PRs for the other modules or should I continue here?

@moritzwilksch I'm fine with you continuing in this PR

@moritzwilksch Thanks Moritz! Can your revert the changes for .direct, though, please? The docs state The direct module provides low-level methods which wouldn't be correct anymore, and that's explicitly a non-standard more-limited way of using Pydantic AI, so I'd prefer to keep it in a separate module.

Done!