chore: mistral - improve type checking + add py.typed (#2009)

* chore: mistral - improve type checking + add py.typed

* fixes

Author: anakin87

.github/workflows/mistral.yml

@@ -50,11 +50,9 @@ jobs:
       - name: Install Hatch
         run: pip install --upgrade hatch
 
-    # TODO: Once this integration is properly typed, use hatch run test:types
-    # https://github.com/deepset-ai/haystack-core-integrations/issues/1771
       - name: Lint
         if: matrix.python-version == '3.9' && runner.os == 'Linux'
-        run: hatch run fmt-check && hatch run lint:typing
+        run: hatch run fmt-check && hatch run test:types
 
       - name: Generate docs
         if: matrix.python-version == '3.9' && runner.os == 'Linux'

integrations/mistral/pyproject.toml

@@ -2,7 +2,7 @@
 requires = ["hatchling", "hatch-vcs"]
 build-backend = "hatchling.build"
 
-[project]
+[project] 
 name = "mistral-haystack"
 dynamic = ["version"]
 description = ''
@@ -67,19 +67,14 @@ integration = 'pytest -m "integration" {args:tests}'
 all = 'pytest {args:tests}'
 cov-retry = 'all --cov=haystack_integrations --reruns 3 --reruns-delay 30 -x'
 
-types = "mypy --install-types --non-interactive --explicit-package-bases {args:src/ tests}"
-
-# TODO: remove lint environment once this integration is properly typed
-# test environment should be used instead
-# https://github.com/deepset-ai/haystack-core-integrations/issues/1771
-[tool.hatch.envs.lint]
-installer = "uv"
-detached = true
-dependencies = ["pip", "mypy>=1.0.0", "ruff>=0.0.243"]
-
-[tool.hatch.envs.lint.scripts]
-typing = "mypy --install-types --non-interactive --explicit-package-bases {args:src/ tests}"
+types = """mypy -p haystack_integrations.components.embedders.mistral \
+-p haystack_integrations.components.generators.mistral {args}"""
 
+[tool.mypy]
+install_types = true
+non_interactive = true
+check_untyped_defs = true
+disallow_incomplete_defs = true
 
 [tool.ruff]
 target-version = "py38"
@@ -156,23 +151,9 @@ show_missing = true
 exclude_lines = ["no cov", "if __name__ == .__main__.:", "if TYPE_CHECKING:"]
 
 
-[[tool.mypy.overrides]]
-module = [
-  "mistral.*",
-  "haystack.*",
-  "haystack_integrations.*",
-  "openai.*",
-  "pytest.*",
-  "numpy.*",
-]
-ignore_missing_imports = true
-
 [tool.pytest.ini_options]
 addopts = "--strict-markers"
 markers = [
   "integration: integration tests",
-  "unit: unit tests",
-  "embedders: embedders tests",
-  "chat_generators: chat_generators tests",
 ]
 log_cli = true

integrations/mistral/src/haystack_integrations/components/embedders/py.typed

@@ -9,8 +9,6 @@
 
 from haystack_integrations.components.embedders.mistral.document_embedder import MistralDocumentEmbedder
 
-pytestmark = pytest.mark.embedders
-
 
 class TestMistralDocumentEmbedder:
     def test_init_default(self, monkeypatch):

integrations/mistral/src/haystack_integrations/components/generators/py.typed

@@ -8,8 +8,6 @@
 
 from haystack_integrations.components.embedders.mistral.text_embedder import MistralTextEmbedder
 
-pytestmark = pytest.mark.embedders
-
 
 class TestMistralTextEmbedder:
     def test_init_default(self, monkeypatch):