Update tests.

newrelic · Oct 9, 2024 · 46d5521 · 46d5521
1 parent 4f7c555
commit 46d5521
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/tests/agent_features/test_llm_custom_attributes.py b/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