chore: Update mistral integration to work with mistralai>=2.0.0 (#2957)

Author: maxdswain

integrations/mistral/pyproject.toml

@@ -22,7 +22,7 @@ classifiers = [
   "Programming Language :: Python :: Implementation :: CPython",
   "Programming Language :: Python :: Implementation :: PyPy",
 ]
-dependencies = ["haystack-ai>=2.22.0", "mistralai>=1.9.11,<2.0.0"]
+dependencies = ["haystack-ai>=2.22.0", "mistralai>=2.0.0"]
 
 [project.urls]
 Documentation = "https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/mistral#readme"

integrations/mistral/src/haystack_integrations/components/converters/mistral/ocr_document_converter.py

@@ -10,14 +10,14 @@
 )
 from haystack.dataclasses import ByteStream
 from haystack.utils import Secret, deserialize_secrets_inplace
-from mistralai import Mistral
-from mistralai.extra import response_format_from_pydantic_model
-from mistralai.models import (
+from mistralai.client import Mistral
+from mistralai.client.models import (
     DocumentURLChunk,
     FileChunk,
     ImageURLChunk,
     OCRResponse,
 )
+from mistralai.extra import response_format_from_pydantic_model  # type: ignore[import-untyped]
 from pydantic import BaseModel
 
 logger = logging.getLogger(__name__)

integrations/mistral/tests/test_ocr_document_converter.py

@@ -8,7 +8,7 @@
 from haystack import Document
 from haystack.dataclasses import ByteStream
 from haystack.utils import Secret
-from mistralai.models import DocumentURLChunk, FileChunk, ImageURLChunk
+from mistralai.client.models import DocumentURLChunk, FileChunk, ImageURLChunk
 from pydantic import BaseModel, Field
 
 from haystack_integrations.components.converters.mistral import (