phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-25 21:57:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge 0360e0d941 into cbe89d2bdd

Browse source
This commit is contained in:
Charlie Doern 2025-07-24 21:06:24 +02:00 • committed by GitHub
commit 57860b6c4c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 18 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

27
llama_stack/providers/utils/telemetry/sqlite_trace_store.py
View file

@ -11,6 +11,9 @@ from typing import Protocol
import aiosqlite
from llama_stack.apis.telemetry import QueryCondition, Span, SpanWithStatus, Trace
from llama_stack.log import get_logger
logger = get_logger(__name__, category="core")
class TraceStore(Protocol):
@ -100,15 +103,21 @@ class SQLiteTraceStore(TraceStore):
conn.row_factory = aiosqlite.Row
async with conn.execute(query, params) as cursor:
rows = await cursor.fetchall()
return [
Trace(
trace_id=row["trace_id"],
root_span_id=row["root_span_id"],
start_time=datetime.fromisoformat(row["start_time"]),
end_time=datetime.fromisoformat(row["end_time"]),
)
for row in rows
]
results = []
for row in rows:
try:
results.append(
Trace(
trace_id=row["trace_id"],
root_span_id=row["root_span_id"],
start_time=datetime.fromisoformat(row["start_time"]),
end_time=datetime.fromisoformat(row["end_time"]),
)
)
except (KeyError, TypeError, ValueError) as e:
logger.warning(f"Could not construct Trace due to missing information in row: {e}")
continue
return results
async def get_span_tree(
self,