Address Douwe's feedback

Author: dmontagu

pydantic_ai_slim/pydantic_ai/agent/__init__.py

@@ -687,46 +687,46 @@ async def get_instructions(run_context: RunContext[AgentDepsT]) -> str | None:
                 if instrumentation_settings and run_span.is_recording():
                     run_span.set_attributes(
                         self._run_span_end_attributes(
-                            state, usage, instrumentation_settings, graph_deps.new_message_index
+                            instrumentation_settings, usage, state.message_history, graph_deps.new_message_index
                         )
                     )
             finally:
                 run_span.end()
 
     def _run_span_end_attributes(
         self,
-        state: _agent_graph.GraphAgentState,
-        usage: _usage.RunUsage,
         settings: InstrumentationSettings,
+        usage: _usage.RunUsage,
+        message_history: list[_messages.ModelMessage],
         new_message_index: int,
     ):
         if settings.version == 1:
             attrs = {
                 'all_messages_events': json.dumps(
-                    [
-                        InstrumentedModel.event_to_dict(e)
-                        for e in settings.messages_to_otel_events(state.message_history)
-                    ]
+                    [InstrumentedModel.event_to_dict(e) for e in settings.messages_to_otel_events(message_history)]
                 )
             }
         else:
             # Store the last instructions here for convenience
-            last_instructions = InstrumentedModel._get_instructions(state.message_history)  # pyright: ignore[reportPrivateUsage]
+            last_instructions = InstrumentedModel._get_instructions(message_history)  # pyright: ignore[reportPrivateUsage]
             attrs: dict[str, Any] = {
-                'pydantic_ai.all_messages': json.dumps(settings.messages_to_otel_messages(list(state.message_history))),
+                'pydantic_ai.all_messages': json.dumps(settings.messages_to_otel_messages(list(message_history))),
+                'pydantic_ai.new_message_index': new_message_index,
                 **settings.system_instructions_attributes(last_instructions),
             }
 
-            # Store an attribute that indicates that the instructions from this agent run were not always the same
+            # If the instructions for this agent run were not always the same, store an attribute that indicates that.
             # This can signal to an observability UI that different steps in the agent run had different instructions
             # Note: We purposely only look at "new" messages because they are the only ones produced by this agent run.
-            for m in state.message_history[new_message_index:]:
-                if (
+            if any(
+                (
                     isinstance(m, _messages.ModelRequest)
                     and m.instructions is not None
                     and m.instructions != last_instructions
-                ):
-                    attrs['pydantic_ai.variable_instructions'] = True
+                )
+                for m in message_history[new_message_index:]
+            ):
+                attrs['pydantic_ai.variable_instructions'] = True
 
         return {
             **usage.opentelemetry_attributes(),