Update tests.

Author: umaannamalai

tests/agent_features/test_llm_custom_attributes.py

@@ -25,7 +25,6 @@ def test_llm_custom_attributes():
     with WithLlmCustomAttributes({"test": "attr", "test1": "attr1"}):
         assert transaction._llm_context_attrs == {"llm.test": "attr", "llm.test1": "attr1"}
 
-    assert transaction._llm_context_attrs is None
 
 
 @pytest.mark.parametrize("context_attrs", (None, "not-a-dict"))
@@ -44,5 +43,3 @@ 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 transaction._llm_context_attrs is None