diff --git a/src/agents/orchestrator.py b/src/agents/orchestrator.py index b382db1..452eb32 100644 --- a/src/agents/orchestrator.py +++ b/src/agents/orchestrator.py @@ -1321,10 +1321,10 @@ class AgentOrchestrator: year_month = datetime.now(TIMEZONE).strftime('%Y-%m') await db.execute(""" INSERT INTO token_usage_monthly - (organization_id, year_month, input_tokens, output_tokens, + (organization_id, year_month, source, input_tokens, output_tokens, cache_creation_tokens, cache_read_tokens, total_cost_usd, api_calls, refresh_count) - VALUES (?, ?, ?, ?, ?, ?, ?, ?, 1) - ON CONFLICT(organization_id, year_month) DO UPDATE SET + VALUES (?, ?, 'monitor', ?, ?, ?, ?, ?, ?, 1) + ON CONFLICT(organization_id, year_month, source) DO UPDATE SET input_tokens = input_tokens + excluded.input_tokens, output_tokens = output_tokens + excluded.output_tokens, cache_creation_tokens = cache_creation_tokens + excluded.cache_creation_tokens,