mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-31 16:01:46 +00:00
better fix
This commit is contained in:
parent
83d02df028
commit
45a7f9a145
2 changed files with 2 additions and 6 deletions
|
@ -9,7 +9,6 @@ import os
|
|||
import sqlite3
|
||||
import threading
|
||||
from datetime import datetime, timezone
|
||||
from enum import Enum
|
||||
|
||||
from opentelemetry.sdk.trace import SpanProcessor
|
||||
from opentelemetry.trace import Span
|
||||
|
@ -154,9 +153,6 @@ class SQLiteSpanProcessor(SpanProcessor):
|
|||
)
|
||||
|
||||
for event in span.events:
|
||||
name = event.name
|
||||
if isinstance(name, Enum):
|
||||
name = name.value
|
||||
cursor.execute(
|
||||
"""
|
||||
INSERT INTO span_events (
|
||||
|
@ -165,7 +161,7 @@ class SQLiteSpanProcessor(SpanProcessor):
|
|||
""",
|
||||
(
|
||||
span_id,
|
||||
name,
|
||||
event.name,
|
||||
datetime.fromtimestamp(event.timestamp / 1e9, timezone.utc).isoformat(),
|
||||
json.dumps(dict(event.attributes)),
|
||||
),
|
||||
|
|
|
@ -148,7 +148,7 @@ class TelemetryAdapter(TelemetryDatasetMixin, Telemetry):
|
|||
if span:
|
||||
timestamp_ns = int(event.timestamp.timestamp() * 1e9)
|
||||
span.add_event(
|
||||
name=event.type,
|
||||
name=event.type.value,
|
||||
attributes={
|
||||
"message": event.message,
|
||||
"severity": event.severity.value,
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue