From 31217429b0331909f6387194cb2e85f31f1fed31 Mon Sep 17 00:00:00 2001 From: Junxiang Huang Date: Tue, 2 Jun 2026 17:38:08 +0800 Subject: [PATCH] fix: ocr ai statictis --- dtable_events/tasks/ai_stats_worker.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/dtable_events/tasks/ai_stats_worker.py b/dtable_events/tasks/ai_stats_worker.py index e77d04e4..d01c95d4 100644 --- a/dtable_events/tasks/ai_stats_worker.py +++ b/dtable_events/tasks/ai_stats_worker.py @@ -85,9 +85,9 @@ def save_to_memory(self, usage_info, session): self.owner_stats[owner_info['owner_id']][model]['output_tokens'] += usage.get('output_tokens') or 0 else: # for non-assistant call, set stat obj to common user - if usage_info['org_id'] != -1: - self.org_stats[usage_info['org_id']][model]['input_tokens'] += usage.get('input_tokens') or 0 - self.org_stats[usage_info['org_id']][model]['output_tokens'] += usage.get('output_tokens') or 0 + if org_id != -1: + self.org_stats[org_id][model]['input_tokens'] += usage.get('input_tokens') or 0 + self.org_stats[org_id][model]['output_tokens'] += usage.get('output_tokens') or 0 else: self.owner_stats[usage_info['username']][model]['input_tokens'] += usage.get('input_tokens') or 0 self.owner_stats[usage_info['username']][model]['output_tokens'] += usage.get('output_tokens') or 0