chore: linting fixes

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

@ -65,8 +65,11 @@ 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": [dict(msg, **{"usage_metadata": usage_metadata, "file_list": file_list}) for msg in "messages": [
v] if isinstance(v, list) else v, 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 = {

Loading…
Cancel
Save