From 46d552112a9b53d0a30c16251ab7fa0cdb998178 Mon Sep 17 00:00:00 2001 From: Uma Annamalai Date: Tue, 8 Oct 2024 17:32:43 -0700 Subject: [PATCH] Update tests. --- tests/agent_features/test_llm_custom_attributes.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/tests/agent_features/test_llm_custom_attributes.py b/tests/agent_features/test_llm_custom_attributes.py index 2ed8adfeb..6abba23f4 100644 --- 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