[codex] Add Anthropic early-close regression coverage (#2080)

Co-authored-by: Alex Hall <alex.mojaki@gmail.com>

Author: chocoHacks33

tests/otel_integrations/test_anthropic.py

@@ -1,6 +1,8 @@
 from __future__ import annotations as _annotations
 
+import gc
 import json
+import logging
 from collections.abc import AsyncIterator, Iterator
 from typing import Any, cast
 
@@ -618,6 +620,43 @@ def test_sync_messages_stream(instrumented_client: anthropic.Anthropic, exporter
     )
 
 
+def test_sync_messages_stream_close_early(
+    instrumented_client: anthropic.Anthropic, exporter: TestExporter, caplog: pytest.LogCaptureFixture
+) -> None:
+    response = instrumented_client.messages.create(
+        max_tokens=1000,
+        model='claude-3-haiku-20240307',
+        system='You are a helpful assistant.',
+        messages=[{'role': 'user', 'content': 'What is four plus five?'}],
+        stream=True,
+    )
+    chunks_seen = 0
+    chunk = None
+
+    with caplog.at_level(logging.ERROR, logger='opentelemetry.context'):
+        with response as stream:
+            for chunk in stream:
+                if hasattr(chunk, 'delta') and isinstance(chunk.delta, TextDelta):  # type: ignore
+                    chunks_seen += 1
+                    break
+        del chunk
+        del stream
+        del response
+        gc.collect()
+
+    assert chunks_seen == 1
+    assert not [
+        record
+        for record in caplog.records
+        if record.name == 'opentelemetry.context' and 'Failed to detach context' in record.getMessage()
+    ]
+    span_names = [span['name'] for span in exporter.exported_spans_as_dict(parse_json_attributes=True)]
+    assert span_names == [
+        'Message with {request_data[model]!r}',
+        'streaming response from {request_data[model]!r} took {duration:.2f}s',
+    ]
+
+
 def test_messages_stream_text_block_none() -> None:
     from logfire._internal.integrations.llm_providers.anthropic import (
         AnthropicMessageStreamState,