Skip to content

Commit

Permalink
Cleanup files.
Browse files Browse the repository at this point in the history
  • Loading branch information
umaannamalai committed Sep 20, 2024
1 parent 37e5b3b commit 977f1b7
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 5 deletions.
4 changes: 0 additions & 4 deletions newrelic/hooks/external_botocore.py
Original file line number Diff line number Diff line change
Expand Up @@ -790,10 +790,6 @@ def handle_chat_completion_event(transaction, bedrock_attrs):
if llm_context_attrs:
llm_metadata_dict.update(llm_context_attrs)

llm_context_attrs = getattr(transaction, "_llm_context_attrs", None)
if llm_context_attrs:
llm_metadata_dict.update(llm_context_attrs)

span_id = bedrock_attrs.get("span_id", None)
trace_id = bedrock_attrs.get("trace_id", None)
request_id = bedrock_attrs.get("request_id", None)
Expand Down
2 changes: 1 addition & 1 deletion tests/agent_features/test_llm_custom_attributes.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,4 +46,4 @@ def test_llm_custom_attributes_prefixed_attrs():
# 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
assert transaction._llm_context_attrs is None

0 comments on commit 977f1b7

Please sign in to comment.