Skip to content

Commit ae84064

Browse files
committed
Fix A001 builtin-variable-shadowing
1 parent 729f738 commit ae84064

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

newrelic/hooks/external_botocore.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -894,7 +894,7 @@ def handle_embedding_event(transaction, bedrock_attrs):
894894
trace_id = bedrock_attrs.get("trace_id", None)
895895
request_id = bedrock_attrs.get("request_id", None)
896896
model = bedrock_attrs.get("model", None)
897-
input = bedrock_attrs.get("input")
897+
input_ = bedrock_attrs.get("input")
898898

899899
embedding_dict = {
900900
"vendor": "bedrock",
@@ -903,7 +903,7 @@ def handle_embedding_event(transaction, bedrock_attrs):
903903
"span_id": span_id,
904904
"trace_id": trace_id,
905905
"token_count": (
906-
settings.ai_monitoring.llm_token_count_callback(model, input)
906+
settings.ai_monitoring.llm_token_count_callback(model, input_)
907907
if settings.ai_monitoring.llm_token_count_callback
908908
else None
909909
),
@@ -916,7 +916,7 @@ def handle_embedding_event(transaction, bedrock_attrs):
916916
embedding_dict.update(llm_metadata_dict)
917917

918918
if settings.ai_monitoring.record_content.enabled:
919-
embedding_dict["input"] = input
919+
embedding_dict["input"] = input_
920920

921921
embedding_dict = {k: v for k, v in embedding_dict.items() if v is not None}
922922
transaction.record_custom_event("LlmEmbedding", embedding_dict)

0 commit comments

Comments
 (0)