Magistral - handling of mistralai.models.thinkchunk.ThinkChunk

[Yach0]

Initial Checks

• I confirm that I'm using the latest version of Pydantic AI
• I confirm that I searched for my issue in https://github.com/pydantic/pydantic-ai/issues before opening this issue

Description

When using thinking mistral models, like "magistral-small-latest", pydantic crashes with:

AssertionError: Other data types like (Image, Reference) are not yet supported,  got <class 'mistralai.models.thinkchunk.ThinkChunk'>

Example Code

Python, Pydantic AI & LLM client version

Python 3.12.11
Pydantic 0.8.0

[Yach0]

An update, seems to happen only with "magistral-small-latest", the medium one works without any issues