fix mypy

sjrl · sjrl · commit 72e7fe1bcae6 · 2026-04-28T12:37:38.000+02:00
diff --git a/haystack/components/agents/agent.py b/haystack/components/agents/agent.py
@@ -824,7 +824,7 @@ async def _run_step_async(self, exe_context: _ExecutionContext, agent_span: Any)
                 **exe_context.chat_generator_inputs,
             }
             step_span.set_content_tag("haystack.agent.step.input", chat_generator_inputs["messages"])
-            result = await self.chat_generator.run_async(**chat_generator_inputs)  # type: ignore[union-attr]
+            result = await self.chat_generator.run_async(**chat_generator_inputs)  # type: ignore[attr-defined]
             llm_messages = result["replies"]
             step_span.set_content_tag("haystack.agent.step.llm_output", llm_messages)
             exe_context.state.set("messages", llm_messages)
@@ -840,7 +840,7 @@ async def _run_step_async(self, exe_context: _ExecutionContext, agent_span: Any)
             )
             exe_context.state.set(key="messages", value=new_chat_history, handler_override=replace_values)
 
-            tool_invoker_result = await self._tool_invoker.run_async(  # type: ignore[union-attr]
+            tool_invoker_result = await self._tool_invoker.run_async(
                 messages=modified_tool_call_messages, state=exe_context.state, **exe_context.tool_invoker_inputs
             )
             tool_messages = tool_invoker_result["tool_messages"]

Original file line number	Diff line number	Diff line change
`@@ -824,7 +824,7 @@ async def _run_step_async(self, exe_context: _ExecutionContext, agent_span: Any)`
`824`	`824`	`**exe_context.chat_generator_inputs,`
`825`	`825`	`}`
`826`	`826`	`step_span.set_content_tag("haystack.agent.step.input", chat_generator_inputs["messages"])`
`827`		`- result = await self.chat_generator.run_async(**chat_generator_inputs) # type: ignore[union-attr]`
	`827`	`+ result = await self.chat_generator.run_async(**chat_generator_inputs) # type: ignore[attr-defined]`
`828`	`828`	`llm_messages = result["replies"]`
`829`	`829`	`step_span.set_content_tag("haystack.agent.step.llm_output", llm_messages)`
`830`	`830`	`exe_context.state.set("messages", llm_messages)`
`@@ -840,7 +840,7 @@ async def _run_step_async(self, exe_context: _ExecutionContext, agent_span: Any)`
`840`	`840`	`)`
`841`	`841`	`exe_context.state.set(key="messages", value=new_chat_history, handler_override=replace_values)`
`842`	`842`
`843`		`- tool_invoker_result = await self._tool_invoker.run_async( # type: ignore[union-attr]`
	`843`	`+ tool_invoker_result = await self._tool_invoker.run_async(`
`844`	`844`	`messages=modified_tool_call_messages, state=exe_context.state, **exe_context.tool_invoker_inputs`
`845`	`845`	`)`
`846`	`846`	`tool_messages = tool_invoker_result["tool_messages"]`