From 3c70b619fdc3c0e8cf81b02288d54c3f727f64da Mon Sep 17 00:00:00 2001 From: crazywoola <427733928@qq.com> Date: Thu, 19 Jun 2025 13:58:23 +0800 Subject: [PATCH] fix: mypy --- .../ops/arize_phoenix_trace/arize_phoenix_trace.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 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 d3807f290b..28ccd271e2 100644 --- a/api/core/ops/arize_phoenix_trace/arize_phoenix_trace.py +++ b/api/core/ops/arize_phoenix_trace/arize_phoenix_trace.py @@ -84,13 +84,17 @@ def setup_tracer(arize_phoenix_config: ArizeConfig | PhoenixConfig) -> tuple[Tra raise -def datetime_to_nanos(dt: datetime) -> int: - """Convert datetime to nanoseconds since epoch.""" +def datetime_to_nanos(dt: Optional[datetime]) -> int: + """Convert datetime to nanoseconds since epoch. If None, use current time.""" + if dt is None: + dt = datetime.now() return int(dt.timestamp() * 1_000_000_000) -def uuid_to_trace_id(string: str) -> int: +def uuid_to_trace_id(string: Optional[str]) -> int: """Convert UUID string to a valid trace ID (16-byte integer).""" + if string is None: + string = "" hash_object = hashlib.sha256(string.encode()) # Take the first 16 bytes (128 bits) of the hash @@ -156,6 +160,7 @@ class ArizePhoenixDataTrace(BaseTraceInstance): span_id = RandomIdGenerator().generate_span_id() context = SpanContext( trace_id=trace_id, + span_id=span_id, is_remote=False, trace_flags=TraceFlags(TraceFlags.SAMPLED), trace_state=TraceState(),