scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

feat(clickhouse): add agent names to the per-trace rollup

Browse source
This commit is contained in:
Ishaan Jaff 2026-09-30 00:40:10 -07:00
parent eaa97090f4
commit f512c161f0
No known key found for this signature in database
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/integrations/clickhouse/schema.py
View file

@ -91,6 +91,7 @@ CREATE TABLE IF NOT EXISTS {db}.{AGENT_TRACES_TABLE}
InputTokens SimpleAggregateFunction(sum, UInt64),
OutputTokens SimpleAggregateFunction(sum, UInt64),
Models SimpleAggregateFunction(groupUniqArrayArray, Array(String)),
AgentNames SimpleAggregateFunction(groupUniqArrayArray, Array(String)),
RequestIds SimpleAggregateFunction(groupArrayArray, Array(String))
)
ENGINE = AggregatingMergeTree
@ -116,6 +117,7 @@ SELECT
sum(InputTokens) AS InputTokens,
sum(OutputTokens) AS OutputTokens,
groupUniqArrayIf(toString(Model), Model != '') AS Models,
groupUniqArrayIf(SpanName, ObservationType = 'agent') AS AgentNames,
groupArrayIf(LiteLLMRequestId, LiteLLMRequestId != '') AS RequestIds
FROM {db}.{OTEL_TRACES_TABLE}
GROUP BY TeamId, TraceId