File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -894,7 +894,7 @@ def handle_embedding_event(transaction, bedrock_attrs):
894
894
trace_id = bedrock_attrs .get ("trace_id" , None )
895
895
request_id = bedrock_attrs .get ("request_id" , None )
896
896
model = bedrock_attrs .get ("model" , None )
897
- input = bedrock_attrs .get ("input" )
897
+ input_ = bedrock_attrs .get ("input" )
898
898
899
899
embedding_dict = {
900
900
"vendor" : "bedrock" ,
@@ -903,7 +903,7 @@ def handle_embedding_event(transaction, bedrock_attrs):
903
903
"span_id" : span_id ,
904
904
"trace_id" : trace_id ,
905
905
"token_count" : (
906
- settings .ai_monitoring .llm_token_count_callback (model , input )
906
+ settings .ai_monitoring .llm_token_count_callback (model , input_ )
907
907
if settings .ai_monitoring .llm_token_count_callback
908
908
else None
909
909
),
@@ -916,7 +916,7 @@ def handle_embedding_event(transaction, bedrock_attrs):
916
916
embedding_dict .update (llm_metadata_dict )
917
917
918
918
if settings .ai_monitoring .record_content .enabled :
919
- embedding_dict ["input" ] = input
919
+ embedding_dict ["input" ] = input_
920
920
921
921
embedding_dict = {k : v for k , v in embedding_dict .items () if v is not None }
922
922
transaction .record_custom_event ("LlmEmbedding" , embedding_dict )
You can’t perform that action at this time.
0 commit comments