fix: span json decoder for event lists (#3867)

Arize-ai · Jul 9, 2024 · 066895b · 066895b
1 parent ccc8cf0
commit 066895b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/phoenix/trace/span_json_decoder.py b/src/phoenix/trace/span_json_decoder.py
@@ -80,7 +80,9 @@ def json_to_span(data: Dict[str, Any]) -> Any:
                 attributes=event.get("attributes") or {},
                 timestamp=datetime.fromisoformat(event["timestamp"]),
             )
-            for event in data["events"]
+            for event in (
+                data["events"] if isinstance(data["events"], list) else json.loads(data["events"])
+            )
         ]
         data["conversation"] = (
             SpanConversationAttributes(**data["conversation"])

diff --git a/src/phoenix/trace/trace_dataset.py b/src/phoenix/trace/trace_dataset.py
@@ -5,6 +5,7 @@
 from uuid import UUID, uuid4
 from warnings import warn
 
+import numpy as np
 import pandas as pd
 from openinference.semconv.trace import (
     DocumentAttributes,
@@ -60,6 +61,7 @@ def normalize_dataframe(dataframe: DataFrame) -> "DataFrame":
     # Convert the start and end times to datetime
     dataframe["start_time"] = normalize_timestamps(dataframe["start_time"])
     dataframe["end_time"] = normalize_timestamps(dataframe["end_time"])
+    dataframe = dataframe.replace({np.nan: None})
     return dataframe