fix: lint

pull/19840/head
crazywoola 11 months ago
parent 7a8406c3d9
commit 2ce66479f9

@ -407,8 +407,8 @@ class ArizePhoenixDataTrace(BaseTraceInstance):
"message_id": trace_info.message_id, "message_id": trace_info.message_id,
"tool_name": "moderation", "tool_name": "moderation",
"status": trace_info.message_data.status, "status": trace_info.message_data.status,
"status_message": trace_info.error or "", "status_message": trace_info.message_data.error or "",
"level": "ERROR" if trace_info.error else "DEFAULT", "level": "ERROR" if trace_info.message_data.error else "DEFAULT",
} }
metadata.update(trace_info.metadata) metadata.update(trace_info.metadata)
@ -443,13 +443,13 @@ class ArizePhoenixDataTrace(BaseTraceInstance):
) )
try: try:
if trace_info.error: if trace_info.message_data.error:
span.add_event( span.add_event(
"exception", "exception",
attributes={ attributes={
"exception.message": trace_info.error, "exception.message": trace_info.message_data.error,
"exception.type": "Error", "exception.type": "Error",
"exception.stacktrace": trace_info.error, "exception.stacktrace": trace_info.message_data.error,
}, },
) )
finally: finally:
@ -520,8 +520,8 @@ class ArizePhoenixDataTrace(BaseTraceInstance):
"message_id": trace_info.message_id, "message_id": trace_info.message_id,
"tool_name": "dataset_retrieval", "tool_name": "dataset_retrieval",
"status": trace_info.message_data.status, "status": trace_info.message_data.status,
"status_message": trace_info.error or "", "status_message": trace_info.message_data.error or "",
"level": "ERROR" if trace_info.error else "DEFAULT", "level": "ERROR" if trace_info.message_data.error else "DEFAULT",
"ls_provider": trace_info.message_data.model_provider, "ls_provider": trace_info.message_data.model_provider,
"ls_model_name": trace_info.message_data.model_id, "ls_model_name": trace_info.message_data.model_id,
} }
@ -552,13 +552,13 @@ class ArizePhoenixDataTrace(BaseTraceInstance):
) )
try: try:
if trace_info.error: if trace_info.message_data.error:
span.add_event( span.add_event(
"exception", "exception",
attributes={ attributes={
"exception.message": trace_info.error, "exception.message": trace_info.message_data.error,
"exception.type": "Error", "exception.type": "Error",
"exception.stacktrace": trace_info.error, "exception.stacktrace": trace_info.message_data.error,
}, },
) )
finally: finally:
@ -624,8 +624,8 @@ class ArizePhoenixDataTrace(BaseTraceInstance):
"project_name": self.project, "project_name": self.project,
"message_id": trace_info.message_id, "message_id": trace_info.message_id,
"status": trace_info.message_data.status, "status": trace_info.message_data.status,
"status_message": trace_info.error or "", "status_message": trace_info.message_data.error or "",
"level": "ERROR" if trace_info.error else "DEFAULT", "level": "ERROR" if trace_info.message_data.error else "DEFAULT",
} }
metadata.update(trace_info.metadata) metadata.update(trace_info.metadata)
@ -655,13 +655,13 @@ class ArizePhoenixDataTrace(BaseTraceInstance):
) )
try: try:
if trace_info.error: if trace_info.message_data.error:
span.add_event( span.add_event(
"exception", "exception",
attributes={ attributes={
"exception.message": trace_info.error, "exception.message": trace_info.message_data.error,
"exception.type": "Error", "exception.type": "Error",
"exception.stacktrace": trace_info.error, "exception.stacktrace": trace_info.message_data.error,
}, },
) )
finally: finally:

Loading…
Cancel
Save