From 2ce66479f9fd49b6857214f61c7b38a014448562 Mon Sep 17 00:00:00 2001
From: crazywoola <427733928@qq.com>
Date: Thu, 19 Jun 2025 13:49:10 +0800
Subject: [PATCH] fix: lint

---
 .../arize_phoenix_trace.py                    | 30 +++++++++----------
 1 file changed, 15 insertions(+), 15 deletions(-)

diff --git a/api/core/ops/arize_phoenix_trace/arize_phoenix_trace.py b/api/core/ops/arize_phoenix_trace/arize_phoenix_trace.py
index 5e5cf86388..d3807f290b 100644
--- a/api/core/ops/arize_phoenix_trace/arize_phoenix_trace.py
+++ b/api/core/ops/arize_phoenix_trace/arize_phoenix_trace.py
@@ -407,8 +407,8 @@ class ArizePhoenixDataTrace(BaseTraceInstance):
             "message_id": trace_info.message_id,
             "tool_name": "moderation",
             "status": trace_info.message_data.status,
-            "status_message": trace_info.error or "",
-            "level": "ERROR" if trace_info.error else "DEFAULT",
+            "status_message": trace_info.message_data.error or "",
+            "level": "ERROR" if trace_info.message_data.error else "DEFAULT",
         }
         metadata.update(trace_info.metadata)
 
@@ -443,13 +443,13 @@ class ArizePhoenixDataTrace(BaseTraceInstance):
         )
 
         try:
-            if trace_info.error:
+            if trace_info.message_data.error:
                 span.add_event(
                     "exception",
                     attributes={
-                        "exception.message": trace_info.error,
+                        "exception.message": trace_info.message_data.error,
                         "exception.type": "Error",
-                        "exception.stacktrace": trace_info.error,
+                        "exception.stacktrace": trace_info.message_data.error,
                     },
                 )
         finally:
@@ -520,8 +520,8 @@ class ArizePhoenixDataTrace(BaseTraceInstance):
             "message_id": trace_info.message_id,
             "tool_name": "dataset_retrieval",
             "status": trace_info.message_data.status,
-            "status_message": trace_info.error or "",
-            "level": "ERROR" if trace_info.error else "DEFAULT",
+            "status_message": trace_info.message_data.error or "",
+            "level": "ERROR" if trace_info.message_data.error else "DEFAULT",
             "ls_provider": trace_info.message_data.model_provider,
             "ls_model_name": trace_info.message_data.model_id,
         }
@@ -552,13 +552,13 @@ class ArizePhoenixDataTrace(BaseTraceInstance):
         )
 
         try:
-            if trace_info.error:
+            if trace_info.message_data.error:
                 span.add_event(
                     "exception",
                     attributes={
-                        "exception.message": trace_info.error,
+                        "exception.message": trace_info.message_data.error,
                         "exception.type": "Error",
-                        "exception.stacktrace": trace_info.error,
+                        "exception.stacktrace": trace_info.message_data.error,
                     },
                 )
         finally:
@@ -624,8 +624,8 @@ class ArizePhoenixDataTrace(BaseTraceInstance):
             "project_name": self.project,
             "message_id": trace_info.message_id,
             "status": trace_info.message_data.status,
-            "status_message": trace_info.error or "",
-            "level": "ERROR" if trace_info.error else "DEFAULT",
+            "status_message": trace_info.message_data.error or "",
+            "level": "ERROR" if trace_info.message_data.error else "DEFAULT",
         }
         metadata.update(trace_info.metadata)
 
@@ -655,13 +655,13 @@ class ArizePhoenixDataTrace(BaseTraceInstance):
         )
 
         try:
-            if trace_info.error:
+            if trace_info.message_data.error:
                 span.add_event(
                     "exception",
                     attributes={
-                        "exception.message": trace_info.error,
+                        "exception.message": trace_info.message_data.error,
                         "exception.type": "Error",
-                        "exception.stacktrace": trace_info.error,
+                        "exception.stacktrace": trace_info.message_data.error,
                     },
                 )
         finally: