type check

Author: gary-huang

ddtrace/llmobs/_experiment.py

@@ -182,15 +182,15 @@ def as_dataframe(self) -> None:
         column_tuples = set()
         data_rows = []
         for record in self._records:
-            flat_record = {}
+            flat_record = {}  # type: Dict[Any, Any]
 
             input_data = record.get("input_data", {})
             if isinstance(input_data, dict):
                 for k, v in input_data.items():
                     flat_record[("input_data", k)] = v
                     column_tuples.add(("input_data", k))
             else:
-                flat_record[("input_data", "")] = input_data  # Use empty string for single input
+                flat_record[("input_data", "")] = input_data
                 column_tuples.add(("input_data", ""))
 
             expected_output = record.get("expected_output", {})
@@ -199,7 +199,7 @@ def as_dataframe(self) -> None:
                     flat_record[("expected_output", k)] = v
                     column_tuples.add(("expected_output", k))
             else:
-                flat_record[("expected_output", "")] = expected_output  # Use empty string for single output
+                flat_record[("expected_output", "")] = expected_output
                 column_tuples.add(("expected_output", ""))
 
             for k, v in record.get("metadata", {}).items():