Assert attribute deletion was successful.

Author: umaannamalai

tests/agent_features/test_llm_custom_attributes.py

@@ -25,6 +25,8 @@ def test_llm_custom_attributes():
     with WithLlmCustomAttributes({"test": "attr", "test1": "attr1"}):
         assert transaction._llm_context_attrs == {"llm.test": "attr", "llm.test1": "attr1"}
 
+    assert not hasattr(transaction, "_llm_context_attrs")
+
 
 @pytest.mark.parametrize("context_attrs", (None, "not-a-dict"))
 @background_task()
@@ -33,7 +35,9 @@ def test_llm_custom_attributes_no_attrs(context_attrs):
 
     with pytest.raises(TypeError):
         with WithLlmCustomAttributes(context_attrs):
-            assert transaction._llm_context_attrs is None
+            pass
+
+    assert not hasattr(transaction, "_llm_context_attrs")
 
 
 @background_task()
@@ -42,3 +46,5 @@ def test_llm_custom_attributes_prefixed_attrs():
     with WithLlmCustomAttributes({"llm.test": "attr", "test1": "attr1"}):
         # Validate API does not prefix attributes that already begin with "llm."
         assert transaction._llm_context_attrs == {"llm.test": "attr", "llm.test1": "attr1"}
+
+    assert not hasattr(transaction, "_llm_context_attrs")