Merge branch 'main' into feat/kreuzberg-converter

v-tan · web-flow · commit c6c008d652c2 · 2026-03-16T17:17:51.000+05:30
diff --git a/integrations/anthropic/CHANGELOG.md b/integrations/anthropic/CHANGELOG.md
@@ -1,5 +1,11 @@
 # Changelog
 
+## [integrations/anthropic-v5.7.0] - 2026-03-13
+
+### 🌀 Miscellaneous
+
+- Feat: List supported models for `AnthropicChatGenerator` (#2958)
+
 ## [integrations/anthropic-v5.6.1] - 2026-03-12
 
 ### 🧹 Chores
diff --git a/integrations/anthropic/src/haystack_integrations/components/generators/anthropic/chat/chat_generator.py b/integrations/anthropic/src/haystack_integrations/components/generators/anthropic/chat/chat_generator.py
@@ -110,6 +110,20 @@ class AnthropicChatGenerator:
         "output_config",
     ]
 
+    SUPPORTED_MODELS: ClassVar[list[str]] = [
+        "claude-opus-4-6",
+        "claude-sonnet-4-6",
+        "claude-haiku-4-5-20251001",
+        "claude-sonnet-4-5-20250929",
+        "claude-opus-4-5-20251101",
+        "claude-opus-4-1-20250805",
+        "claude-sonnet-4-20250514",
+        "claude-opus-4-20250514",
+        "claude-3-haiku-20240307",
+    ]
+    """A non-exhaustive list of chat models supported by this component. See
+     https://platform.claude.com/docs/en/about-claude/models/overview for the full list."""
+
     def __init__(
         self,
         api_key: Secret = Secret.from_env_var("ANTHROPIC_API_KEY"),  # noqa: B008
diff --git a/integrations/anthropic/tests/test_chat_generator.py b/integrations/anthropic/tests/test_chat_generator.py
@@ -90,6 +90,13 @@ def mock_anthropic_completion():
 
 
 class TestAnthropicChatGenerator:
+    def test_supported_models(self) -> None:
+        """SUPPORTED_MODELS is a non-empty list of strings."""
+        models = AnthropicChatGenerator.SUPPORTED_MODELS
+        assert isinstance(models, list)
+        assert len(models) > 0
+        assert all(isinstance(m, str) for m in models)
+
     def test_init_default(self, monkeypatch):
         """
         Test the default initialization of the AnthropicChatGenerator component.
diff --git a/integrations/astra/tests/test_document_store.py b/integrations/astra/tests/test_document_store.py
@@ -157,7 +157,8 @@ def run_before_tests(self, document_store: AstraDocumentStore):
         document_store.delete_all_documents()
         assert document_store.count_documents() == 0
 
-    def assert_documents_are_equal(self, received: list[Document], expected: list[Document]):
+    @staticmethod
+    def assert_documents_are_equal(received: list[Document], expected: list[Document]):
         """
         Assert that two lists of Documents are equal.
         This is used in every test, if a Document Store implementation has a different behaviour
@@ -174,7 +175,7 @@ def test_comparison_equal_with_none(self, document_store, filterable_docs):
         document_store.write_documents(filterable_docs)
         result = document_store.filter_documents(filters={"field": "meta.number", "operator": "==", "value": None})
         # Astra does not support filtering on None, it returns empty list
-        self.assert_documents_are_equal(result, [])
+        TestDocumentStore.assert_documents_are_equal(result, [])
 
     def test_write_documents(self, document_store: AstraDocumentStore):
         """
@@ -185,9 +186,9 @@ def test_write_documents(self, document_store: AstraDocumentStore):
         doc2 = Document(id="1", content="test doc 2")
 
         assert document_store.write_documents([doc2], policy=DuplicatePolicy.OVERWRITE) == 1
-        self.assert_documents_are_equal(document_store.filter_documents(), [doc2])
+        TestDocumentStore.assert_documents_are_equal(document_store.filter_documents(), [doc2])
         assert document_store.write_documents(documents=[doc1], policy=DuplicatePolicy.OVERWRITE) == 1
-        self.assert_documents_are_equal(document_store.filter_documents(), [doc1])
+        TestDocumentStore.assert_documents_are_equal(document_store.filter_documents(), [doc1])
 
     def test_write_documents_skip_duplicates(self, document_store: AstraDocumentStore):
         docs = [
@@ -264,7 +265,7 @@ def test_filter_documents_nested_filters(self, document_store, filterable_docs):
         document_store.write_documents(filterable_docs)
         result = document_store.filter_documents(filters=filter_criteria)
 
-        self.assert_documents_are_equal(
+        TestDocumentStore.assert_documents_are_equal(
             result,
             [
                 d
@@ -278,7 +279,7 @@ def test_filter_documents_by_id(self, document_store):
         docs = [Document(id="1", content="test doc 1"), Document(id="2", content="test doc 2")]
         document_store.write_documents(docs)
         result = document_store.filter_documents(filters={"field": "id", "operator": "==", "value": "1"})
-        self.assert_documents_are_equal(result, [docs[0]])
+        TestDocumentStore.assert_documents_are_equal(result, [docs[0]])
 
     def test_filter_documents_by_in_operator(self, document_store):
         docs = [Document(id="3", content="test doc 3"), Document(id="4", content="test doc 4")]
@@ -288,8 +289,8 @@ def test_filter_documents_by_in_operator(self, document_store):
         # Sort the result in place by the id field
         result.sort(key=lambda x: x.id)
 
-        self.assert_documents_are_equal([result[0]], [docs[0]])
-        self.assert_documents_are_equal([result[1]], [docs[1]])
+        TestDocumentStore.assert_documents_are_equal([result[0]], [docs[0]])
+        TestDocumentStore.assert_documents_are_equal([result[1]], [docs[1]])
 
     def test_count_documents_by_filter(self, document_store: AstraDocumentStore):
         docs = [
diff --git a/integrations/meta_llama/CHANGELOG.md b/integrations/meta_llama/CHANGELOG.md
@@ -1,5 +1,22 @@
 # Changelog
 
+## [integrations/meta_llama-v2.2.0] - 2026-03-13
+
+### 🚀 Features
+
+- Add list of supported models for MetaLlamaChatGenerator (#2959)
+
+## [integrations/meta_llama-v2.1.0] - 2026-02-26
+
+### 🚀 Features
+
+- *(meta-llama)* Add timeout and max_retries to chat generator (#2872)
+
+### 📚 Documentation
+
+- Simplify pydoc configs (#2855)
+
+
 ## [integrations/meta_llama-v2.0.0] - 2026-01-12
 
 ### ⚙️ CI
diff --git a/integrations/meta_llama/src/haystack_integrations/components/generators/meta_llama/chat/chat_generator.py b/integrations/meta_llama/src/haystack_integrations/components/generators/meta_llama/chat/chat_generator.py
@@ -3,7 +3,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from typing import Any
+from typing import Any, ClassVar
 
 from haystack import component, default_to_dict, logging
 from haystack.components.generators.chat import OpenAIChatGenerator
@@ -54,6 +54,15 @@ class MetaLlamaChatGenerator(OpenAIChatGenerator):
     ```
     """
 
+    SUPPORTED_MODELS: ClassVar[list[str]] = [
+        "Llama-4-Maverick-17B-128E-Instruct-FP8",
+        "Llama-4-Scout-17B-16E-Instruct-FP8",
+        "Llama-3.3-70B-Instruct",
+        "Llama-3.3-8B-Instruct",
+    ]
+    """A non-exhaustive list of chat models supported by this component.
+    See https://llama.developer.meta.com/docs/models for the full list."""
+
     def __init__(
         self,
         *,
diff --git a/integrations/meta_llama/tests/test_llama_chat_generator.py b/integrations/meta_llama/tests/test_llama_chat_generator.py
@@ -103,6 +103,13 @@ def mock_chat_completion():
 
 
 class TestLlamaChatGenerator:
+    def test_supported_models(self) -> None:
+        """SUPPORTED_MODELS is a non-empty list of strings."""
+        models = MetaLlamaChatGenerator.SUPPORTED_MODELS
+        assert isinstance(models, list)
+        assert len(models) > 0
+        assert all(isinstance(m, str) for m in models)
+
     def test_init_default(self, monkeypatch):
         monkeypatch.setenv("LLAMA_API_KEY", "test-api-key")
         component = MetaLlamaChatGenerator()
diff --git a/integrations/mistral/CHANGELOG.md b/integrations/mistral/CHANGELOG.md
@@ -1,5 +1,12 @@
 # Changelog
 
+## [integrations/mistral-v1.2.0] - 2026-03-13
+
+### 🧹 Chores
+
+- Update mistral integration to work with mistralai>=2.0.0 (#2957)
+
+
 ## [integrations/mistral-v1.1.0] - 2026-03-12
 
 ### 🚀 Features
diff --git a/integrations/mistral/pyproject.toml b/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"
diff --git a/integrations/mistral/src/haystack_integrations/components/converters/mistral/ocr_document_converter.py b/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__)
diff --git a/integrations/mistral/tests/test_ocr_document_converter.py b/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 (
diff --git a/integrations/stackit/CHANGELOG.md b/integrations/stackit/CHANGELOG.md
@@ -1,5 +1,20 @@
 # Changelog
 
+## [integrations/stackit-v2.1.0] - 2026-03-16
+
+### 🚀 Features
+
+- Add list of supported models for StackIT components (#2962)
+
+### 📚 Documentation
+
+- Simplify pydoc configs (#2855)
+
+### 🧪 Testing
+
+- Stackit - make integration tests more robust (#2882)
+
+
 ## [integrations/stackit-v2.0.0] - 2026-01-13
 
 
diff --git a/integrations/stackit/src/haystack_integrations/components/embedders/stackit/document_embedder.py b/integrations/stackit/src/haystack_integrations/components/embedders/stackit/document_embedder.py
@@ -1,7 +1,7 @@
 # SPDX-FileCopyrightText: 2023-present deepset GmbH <info@deepset.ai>
 #
 # SPDX-License-Identifier: Apache-2.0
-from typing import Any
+from typing import Any, ClassVar
 
 from haystack import component, default_to_dict
 from haystack.components.embedders import OpenAIDocumentEmbedder
@@ -30,6 +30,14 @@ class STACKITDocumentEmbedder(OpenAIDocumentEmbedder):
     ```
     """
 
+    SUPPORTED_MODELS: ClassVar[list[str]] = [
+        "intfloat/e5-mistral-7b-instruct",
+        "Qwen/Qwen3-VL-Embedding-8B",
+    ]
+    """A non-exhaustive list of embedding models supported by this component.
+    See https://docs.stackit.cloud/products/data-and-ai/ai-model-serving/basics/available-shared-models
+    for the full list."""
+
     def __init__(
         self,
         model: str,
diff --git a/integrations/stackit/src/haystack_integrations/components/embedders/stackit/text_embedder.py b/integrations/stackit/src/haystack_integrations/components/embedders/stackit/text_embedder.py
@@ -1,7 +1,7 @@
 # SPDX-FileCopyrightText: 2023-present deepset GmbH <info@deepset.ai>
 #
 # SPDX-License-Identifier: Apache-2.0
-from typing import Any
+from typing import Any, ClassVar
 
 from haystack import component, default_to_dict
 from haystack.components.embedders import OpenAITextEmbedder
@@ -23,6 +23,14 @@ class STACKITTextEmbedder(OpenAITextEmbedder):
     ```
     """
 
+    SUPPORTED_MODELS: ClassVar[list[str]] = [
+        "intfloat/e5-mistral-7b-instruct",
+        "Qwen/Qwen3-VL-Embedding-8B",
+    ]
+    """A non-exhaustive list of embedding models supported by this component.
+    See https://docs.stackit.cloud/products/data-and-ai/ai-model-serving/basics/available-shared-models
+    for the full list."""
+
     def __init__(
         self,
         model: str,
diff --git a/integrations/stackit/src/haystack_integrations/components/generators/stackit/chat/chat_generator.py b/integrations/stackit/src/haystack_integrations/components/generators/stackit/chat/chat_generator.py
@@ -1,7 +1,7 @@
 # SPDX-FileCopyrightText: 2025-present deepset GmbH <info@deepset.ai>
 #
 # SPDX-License-Identifier: Apache-2.0
-from typing import Any
+from typing import Any, ClassVar
 
 from haystack import component, default_to_dict
 from haystack.components.generators.chat import OpenAIChatGenerator
@@ -38,6 +38,19 @@ class STACKITChatGenerator(OpenAIChatGenerator):
     ```
     """
 
+    SUPPORTED_MODELS: ClassVar[list[str]] = [
+        "Qwen/Qwen3-VL-235B-A22B-Instruct-FP8",
+        "cortecs/Llama-3.3-70B-Instruct-FP8-Dynamic",
+        "openai/gpt-oss-120b",
+        "google/gemma-3-27b-it",
+        "openai/gpt-oss-20b",
+        "neuralmagic/Mistral-Nemo-Instruct-2407-FP8",
+        "neuralmagic/Meta-Llama-3.1-8B-Instruct-FP8",
+    ]
+    """A non-exhaustive list of chat models supported by this component.
+    See https://docs.stackit.cloud/products/data-and-ai/ai-model-serving/basics/available-shared-models
+    for the full list."""
+
     def __init__(
         self,
         model: str,
diff --git a/integrations/stackit/tests/test_stackit_chat_generator.py b/integrations/stackit/tests/test_stackit_chat_generator.py
@@ -63,6 +63,13 @@ def mock_chat_completion():
 
 
 class TestSTACKITChatGenerator:
+    def test_supported_models(self):
+        """SUPPORTED_MODELS is a non-empty list of strings."""
+        models = STACKITChatGenerator.SUPPORTED_MODELS
+        assert isinstance(models, list)
+        assert len(models) > 0
+        assert all(isinstance(m, str) for m in models)
+
     def test_init_default(self, monkeypatch):
         monkeypatch.setenv("STACKIT_API_KEY", "test-api-key")
         component = STACKITChatGenerator(model="neuralmagic/Meta-Llama-3.1-8B-Instruct-FP8")
diff --git a/integrations/stackit/tests/test_stackit_document_embedder.py b/integrations/stackit/tests/test_stackit_document_embedder.py
@@ -11,6 +11,13 @@
 
 
 class TestSTACKITDocumentEmbedder:
+    def test_supported_models(self):
+        """SUPPORTED_MODELS is a non-empty list of strings."""
+        models = STACKITDocumentEmbedder.SUPPORTED_MODELS
+        assert isinstance(models, list)
+        assert len(models) > 0
+        assert all(isinstance(m, str) for m in models)
+
     def test_init_default(self, monkeypatch):
         monkeypatch.setenv("STACKIT_API_KEY", "test-api-key")
 
diff --git a/integrations/stackit/tests/test_stackit_text_embedder.py b/integrations/stackit/tests/test_stackit_text_embedder.py
@@ -10,6 +10,13 @@
 
 
 class TestSTACKITTextEmbedder:
+    def test_supported_models(self):
+        """SUPPORTED_MODELS is a non-empty list of strings."""
+        models = STACKITTextEmbedder.SUPPORTED_MODELS
+        assert isinstance(models, list)
+        assert len(models) > 0
+        assert all(isinstance(m, str) for m in models)
+
     def test_init_default(self, monkeypatch):
         monkeypatch.setenv("STACKIT_API_KEY", "test-api-key")
 
diff --git a/integrations/watsonx/CHANGELOG.md b/integrations/watsonx/CHANGELOG.md
@@ -1,5 +1,12 @@
 # Changelog
 
+## [integrations/watsonx-v1.4.0] - 2026-03-13
+
+### 🚀 Features
+
+- List supported models in `WatsonxChatGenerator` (#2956)
+
+
 ## [integrations/watsonx-v1.3.0] - 2026-03-12
 
 ### 🚀 Features
diff --git a/integrations/watsonx/src/haystack_integrations/components/generators/watsonx/chat/chat_generator.py b/integrations/watsonx/src/haystack_integrations/components/generators/watsonx/chat/chat_generator.py
diff --git a/integrations/watsonx/src/haystack_integrations/components/generators/watsonx/generator.py b/integrations/watsonx/src/haystack_integrations/components/generators/watsonx/generator.py
diff --git a/integrations/watsonx/tests/test_chat_generator.py b/integrations/watsonx/tests/test_chat_generator.py
diff --git a/integrations/watsonx/tests/test_generator.py b/integrations/watsonx/tests/test_generator.py

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ classifiers = [`
`22`	`22`	`"Programming Language :: Python :: Implementation :: CPython",`
`23`	`23`	`"Programming Language :: Python :: Implementation :: PyPy",`
`24`	`24`	`]`
`25`		`-dependencies = ["haystack-ai>=2.22.0", "mistralai>=1.9.11,<2.0.0"]`
	`25`	`+dependencies = ["haystack-ai>=2.22.0", "mistralai>=2.0.0"]`
`26`	`26`
`27`	`27`	`[project.urls]`
`28`	`28`	`Documentation = "https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/mistral#readme"`