Update pydantic_ai_slim/pydantic_ai/models/cohere.py

Co-authored-by: David Montague <35119617+dmontagu@users.noreply.github.com>
pydantic · Feb 24, 2025 · a226969 · a226969
1 parent 23aa147
commit a226969
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pydantic_ai_slim/pydantic_ai/models/cohere.py b/pydantic_ai_slim/pydantic_ai/models/cohere.py
@@ -245,7 +245,7 @@ def _map_user_message(cls, message: ModelRequest) -> Iterable[ChatMessageV2]:
                 if isinstance(part.content, str):
                     yield UserChatMessageV2(role='user', content=part.content)
                 else:
-                    raise RuntimeError('Only string as user prompt is supported for Cohere.')
+                    raise RuntimeError('Cohere does not yet support multi-modal inputs.')
             elif isinstance(part, ToolReturnPart):
                 yield ToolChatMessageV2(
                     role='tool',