fix tests

This commit is contained in:
Ishaan Jaff 2025-03-29 17:38:24 -07:00
parent 4e106ce217
commit a3df0269bb
2 changed files with 2 additions and 1 deletions

View file

@ -555,7 +555,7 @@ class TestSpendLogsPayload:
"model": "claude-3-7-sonnet-20250219", "model": "claude-3-7-sonnet-20250219",
"user": "", "user": "",
"team_id": "", "team_id": "",
"metadata": '{"applied_guardrails": [], "batch_models": null, "additional_usage_values": {"completion_tokens_details": null, "prompt_tokens_details": {"audio_tokens": null, "cached_tokens": 0, "text_tokens": null, "image_tokens": null}, "cache_creation_input_tokens": 0, "cache_read_input_tokens": 0}}', "metadata": '{"applied_guardrails": [], "batch_models": null, "mcp_tool_call_metadata": null, "additional_usage_values": {"completion_tokens_details": null, "prompt_tokens_details": {"audio_tokens": null, "cached_tokens": 0, "text_tokens": null, "image_tokens": null}, "cache_creation_input_tokens": 0, "cache_read_input_tokens": 0}}',
"cache_key": "Cache OFF", "cache_key": "Cache OFF",
"spend": 0.01383, "spend": 0.01383,
"total_tokens": 2598, "total_tokens": 2598,

View file

@ -274,6 +274,7 @@ def validate_redacted_message_span_attributes(span):
"metadata.user_api_key_end_user_id", "metadata.user_api_key_end_user_id",
"metadata.user_api_key_user_email", "metadata.user_api_key_user_email",
"metadata.applied_guardrails", "metadata.applied_guardrails",
"metadata.mcp_tool_call_metadata",
] ]
_all_attributes = set( _all_attributes = set(