Fix linter warnings and errors

Author: Ryzhtus

integrations/dspy/pyproject.toml

@@ -22,7 +22,7 @@ classifiers = [
     "Programming Language :: Python :: Implementation :: CPython",
     "Programming Language :: Python :: Implementation :: PyPy",
 ]
-dependencies = ["haystack-ai>=2.0.0", "dspy>=3.0.0"]
+dependencies = ["haystack-ai>=2.22.0", "dspy>=3.0.0"]
 
 [project.urls]
 Documentation = "https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/dspy#readme"
@@ -73,6 +73,10 @@ non_interactive = true
 check_untyped_defs = true
 disallow_incomplete_defs = true
 
+[[tool.mypy.overrides]]
+module = "dspy.*"
+ignore_missing_imports = true
+
 [tool.hatch.metadata]
 allow-direct-references = true
 
@@ -129,7 +133,7 @@ ban-relative-imports = "parents"
 
 [tool.ruff.lint.per-file-ignores]
 # Tests can use magic values, assertions, and relative imports
-"tests/**/*" = ["PLR2004", "S101", "TID252"]
+"tests/**/*" = ["PLR2004", "S101", "TID252", "ARG002"]
 # Examples can print their output
 "examples/**" = ["T201"]
 

integrations/dspy/src/haystack_integrations/components/generators/dspy/chat/chat_generator.py

@@ -25,7 +25,7 @@ def _configure_dspy_lm(model: str, api_base: str | None = None, **kwargs: Any) -
     return lm
 
 
-def _get_dspy_module_class(module_type: str):
+def _get_dspy_module_class(module_type: str) -> type:
     """
     Map a module type string to the corresponding DSPy module class.
 
@@ -148,7 +148,7 @@ def __init__(
         module_class = _get_dspy_module_class(self.module_type)
         self._module = module_class(self.signature, **self.module_kwargs)
 
-    def _build_dspy_inputs(self, prompt: str, **kwargs) -> dict[str, Any]:
+    def _build_dspy_inputs(self, prompt: str, **kwargs: Any) -> dict[str, Any]:
         """Build the input dict for the DSPy module call."""
         if self.input_mapping:
             dspy_inputs = {}
@@ -180,8 +180,8 @@ def _extract_last_user_message(messages: list[ChatMessage]) -> str:
         """Extract the text of the last user message from a list of chat messages."""
         for msg in reversed(messages):
             if msg.role == ChatRole.USER:
-                return msg.text
-        return messages[-1].text
+                return msg.text or ""
+        return messages[-1].text or ""
 
     def _signature_to_string(self) -> str:
         """
@@ -227,7 +227,7 @@ def run(
         self,
         messages: list[ChatMessage],
         generation_kwargs: dict[str, Any] | None = None,
-        **kwargs,
+        **kwargs: Any,
     ) -> dict[str, Any]:
         """
         Run the DSPy module on the given messages.
@@ -256,7 +256,7 @@ async def run_async(
         self,
         messages: list[ChatMessage],
         generation_kwargs: dict[str, Any] | None = None,
-        **kwargs,
+        **kwargs: Any,
     ) -> dict[str, Any]:
         """
         Asynchronously run the DSPy module on the given messages.

integrations/dspy/tests/test_chat_generator.py

@@ -405,7 +405,7 @@ def test_run_with_input_mapping(self, mock_dspy_module):
             input_mapping={"context": "context", "question": "question"},
         )
         messages = [ChatMessage.from_user("What is ML?")]
-        response = component.run(messages=messages, context="Machine learning is a subset of AI.")
+        component.run(messages=messages, context="Machine learning is a subset of AI.")
 
         call_kwargs = mock_dspy_module.call_args.kwargs
         assert call_kwargs.get("context") == "Machine learning is a subset of AI."