phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-02 00:34:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(telemetry): get rid of annoying sqlite span export error

Browse source
This commit is contained in:
Ashwin Bharambe 2025-05-24 14:30:31 -07:00
parent 3faf1e4a79
commit 83d02df028
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/providers/inline/telemetry/meta_reference/sqlite_span_processor.py
View file

@ -9,6 +9,7 @@ import os
import sqlite3 import sqlite3
import threading import threading
from datetime import datetime, timezone from datetime import datetime, timezone
from enum import Enum
from opentelemetry.sdk.trace import SpanProcessor from opentelemetry.sdk.trace import SpanProcessor
from opentelemetry.trace import Span from opentelemetry.trace import Span
@ -153,6 +154,9 @@ class SQLiteSpanProcessor(SpanProcessor):
) )
for event in span.events: for event in span.events:
name = event.name
if isinstance(name, Enum):
name = name.value
cursor.execute( cursor.execute(
""" """
INSERT INTO span_events ( INSERT INTO span_events (
@ -161,7 +165,7 @@ class SQLiteSpanProcessor(SpanProcessor):
""", """,
( (
span_id, span_id,
event.name, name,
datetime.fromtimestamp(event.timestamp / 1e9, timezone.utc).isoformat(), datetime.fromtimestamp(event.timestamp / 1e9, timezone.utc).isoformat(),
json.dumps(dict(event.attributes)), json.dumps(dict(event.attributes)),
), ),