From 684e20ff847aacb91252f5d7426071e975101e46 Mon Sep 17 00:00:00 2001
From: Ruben Fischer <ruben.fischer@onyva.de>
Date: Thu, 19 Feb 2026 21:10:23 +0100
Subject: [PATCH] fixed token tracking

---
 src/services/telegram_service.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/services/telegram_service.py b/src/services/telegram_service.py
index 5109fc5..5d98dd6 100644
--- a/src/services/telegram_service.py
+++ b/src/services/telegram_service.py
@@ -410,6 +410,11 @@ class TelegramService:
             # Single writer pass — no critic loop
             from src.agents.writer import WriterAgent
             writer = WriterAgent()
+            writer.set_tracking_context(
+                operation="post_creation",
+                user_id=user_id,
+                company_id=str(profile.company_id) if profile and profile.company_id else None,
+            )
             post_content = await writer.process(
                 topic={"title": topic_text, "fact": topic_text, "relevance": "User-specified topic"},
                 profile_analysis=profile_analysis.full_analysis,