Incompatibility with cohere >=5.20

[DarthJuri]

Initial Checks

• I'm using the latest version of Pydantic AI
• I've searched for my issue in the issue tracker before opening this issue

Description

The AssistantMessageV2ContentItem was probably replaced by the AssistantMessageV2ContentOneItem in the cohere library. Could be replaced by TextAssistantMessageV2ContentItem | ThinkingAssistantMessageV2ContentItem in the type annotation (I have prepared PR to fix this bug).

Minimal, Reproducible Example

from pydantic_ai.models.cohere import CohereModel

Logfire Trace

No response

Python, Pydantic AI & LLM client version

• Python: 3.13.11
• Pydantic AI: 1.62.0
• LLM provider SDK: cohere 5.20.6

[dsfaccini]

hey @DarthJuri , was this causing issues for you in any way?

[DarthJuri]

hey @dsfaccini, it's not critical for us, but we are preparing "bring your own token" solution for our customers and we wanted to support all model providers Pydantic AI supports. You might find my solution here: #4384