mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-08-01 16:24:44 +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 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
|
||||||
|
@ -154,9 +153,6 @@ 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 (
|
||||||
|
@ -165,7 +161,7 @@ class SQLiteSpanProcessor(SpanProcessor):
|
||||||
""",
|
""",
|
||||||
(
|
(
|
||||||
span_id,
|
span_id,
|
||||||
name,
|
event.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)),
|
||||||
),
|
),
|
||||||
|
|
|
@ -148,7 +148,7 @@ class TelemetryAdapter(TelemetryDatasetMixin, Telemetry):
|
||||||
if span:
|
if span:
|
||||||
timestamp_ns = int(event.timestamp.timestamp() * 1e9)
|
timestamp_ns = int(event.timestamp.timestamp() * 1e9)
|
||||||
span.add_event(
|
span.add_event(
|
||||||
name=event.type,
|
name=event.type.value,
|
||||||
attributes={
|
attributes={
|
||||||
"message": event.message,
|
"message": event.message,
|
||||||
"severity": event.severity.value,
|
"severity": event.severity.value,
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue