From edc30c7de6f88fdd86d68d41c02fbb893b40fc0f Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 29 Sep 2026 22:51:06 -0700 Subject: [PATCH] style(agent-tracing): apply ruff lint + format to clickhouse schema --- litellm/integrations/clickhouse/schema.py | 14 +++----------- 1 file changed, 3 insertions(+), 11 deletions(-) diff --git a/litellm/integrations/clickhouse/schema.py b/litellm/integrations/clickhouse/schema.py index 6756dfa175b..bd1708d9045 100644 --- a/litellm/integrations/clickhouse/schema.py +++ b/litellm/integrations/clickhouse/schema.py @@ -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)