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

style(agent-tracing): apply ruff lint + format to clickhouse schema

Browse source
This commit is contained in:
Ishaan Jaff 2026-09-29 22:51:06 -07:00
parent b4743ced3d
commit edc30c7de6
No known key found for this signature in database
1 changed files with 3 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -9,8 +9,6 @@ ClickHouse DDL for agent tracing.
Join: otel_traces.LiteLLMRequestId = spend_logs.response_id
"""
from typing import List
from litellm.integrations.clickhouse.clickhouse_client import ClickHouseClient
OTEL_TRACES_TABLE = "otel_traces"
@ -20,9 +18,7 @@ SPEND_LOGS_TABLE = "spend_logs"
_ATTR_MAP = "Map(LowCardinality(String), String)"
def schema_statements(
database: str, trace_retention_days: int, spend_log_retention_days: int
) -> List[str]:
def schema_statements(database: str, trace_retention_days: int, spend_log_retention_days: int) -> list[str]:
db = database
return [
f"CREATE DATABASE IF NOT EXISTS {db}",
@ -172,10 +168,6 @@ TTL toDateTime(start_time) + INTERVAL {spend_log_retention_days} DAY
]
async def ensure_schema(
client: ClickHouseClient, trace_retention_days: int, spend_log_retention_days: int
) -> None:
for statement in schema_statements(
client.database, trace_retention_days, spend_log_retention_days
):
async def ensure_schema(client: ClickHouseClient, trace_retention_days: int, spend_log_retention_days: int) -> None:
for statement in schema_statements(client.database, trace_retention_days, spend_log_retention_days):
await client.execute(statement)