fix codex review in openai#2242 (review)

Author: habema

src/agents/run.py

@@ -1397,6 +1397,7 @@ async def _run_single_turn_streamed(
         model = cls._get_model(agent, run_config)
         model_settings = agent.model_settings.resolve(run_config.model_settings)
         model_settings = RunImpl.maybe_reset_tool_choice(agent, tool_use_tracker, model_settings)
+        function_map = {tool.name: tool for tool in all_tools if isinstance(tool, FunctionTool)}
 
         final_response: ModelResponse | None = None
 
@@ -1489,20 +1490,13 @@ async def _run_single_turn_streamed(
                         emitted_tool_call_ids.add(call_id)
 
                         # Try to get origin info if this is a FunctionTool call
+                        # Use same lookup logic as _run_impl (function_map with last-wins semantics)
                         tool_origin = None
                         if isinstance(output_item, ResponseFunctionToolCall):
                             tool_name = getattr(output_item, "name", None)
-                            if tool_name:
-                                function_tool = next(
-                                    (
-                                        tool
-                                        for tool in all_tools
-                                        if isinstance(tool, FunctionTool) and tool.name == tool_name
-                                    ),
-                                    None,
-                                )
-                                if function_tool:
-                                    tool_origin = _get_tool_origin_info(function_tool)
+                            if tool_name and tool_name in function_map:
+                                function_tool = function_map[tool_name]
+                                tool_origin = _get_tool_origin_info(function_tool)
 
                         tool_item = ToolCallItem(
                             raw_item=cast(ToolCallItemTypes, output_item),