fix: resolve review comments from @aiswaryasankarwork

pull/14262/head
Bharat Ramanathan 1 year ago
parent a93ee24fbe
commit 019045d4bd

@ -65,7 +65,8 @@ class WeaveTraceModel(WeaveTokenUsage, WeaveMultiModel):
v = replace_text_with_content(data=v) v = replace_text_with_content(data=v)
if field_name == "inputs": if field_name == "inputs":
data = { data = {
"messages": v, "messages": [dict(msg, **{"usage_metadata": usage_metadata, "file_list": file_list}) for msg in
v] if isinstance(v, list) else v,
} }
elif field_name == "outputs": elif field_name == "outputs":
data = { data = {

@ -279,8 +279,8 @@ class WeaveDataTrace(BaseTraceInstance):
attributes = trace_info.metadata attributes = trace_info.metadata
attributes["tags"] = ["moderation"] attributes["tags"] = ["moderation"]
attributes["message_id"] = trace_info.message_id attributes["message_id"] = trace_info.message_id
attributes["start_time"] = (trace_info.start_time or trace_info.message_data.created_at,) attributes["start_time"] = trace_info.start_time or trace_info.message_data.created_at
attributes["end_time"] = (trace_info.end_time or trace_info.message_data.updated_at,) attributes["end_time"] = trace_info.end_time or trace_info.message_data.updated_at
moderation_run = WeaveTraceModel( moderation_run = WeaveTraceModel(
id=str(uuid.uuid4()), id=str(uuid.uuid4()),

Loading…
Cancel
Save