apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-04-28 10:26:33 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

chore: ran linting

Browse source
This commit is contained in:
Anish Sarkar 2026-04-14 21:52:26 +05:30
parent a74ed014cc
commit 9fc0976d5e
5 changed files with 97 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

23
surfsense_backend/app/services/token_tracking_service.py
View file

@ -117,12 +117,16 @@ class TokenTrackingCallback(CustomLogger):
) -> None:
acc = _turn_accumulator.get()
if acc is None:
logger.debug("[TokenTracking] async_log_success_event fired but no accumulator in context")
logger.debug(
"[TokenTracking] async_log_success_event fired but no accumulator in context"
)
return
usage = getattr(response_obj, "usage", None)
if not usage:
logger.debug("[TokenTracking] async_log_success_event fired but response has no usage data")
logger.debug(
"[TokenTracking] async_log_success_event fired but response has no usage data"
)
return
prompt_tokens = getattr(usage, "prompt_tokens", 0) or 0
@ -139,7 +143,11 @@ class TokenTrackingCallback(CustomLogger):
)
logger.info(
"[TokenTracking] Captured: model=%s prompt=%d completion=%d total=%d (accumulator now has %d calls)",
model, prompt_tokens, completion_tokens, total_tokens, len(acc.calls),
model,
prompt_tokens,
completion_tokens,
total_tokens,
len(acc.calls),
)
@ -187,11 +195,16 @@ async def record_token_usage(
session.add(record)
logger.debug(
"[TokenTracking] recorded %s usage: prompt=%d completion=%d total=%d",
usage_type, prompt_tokens, completion_tokens, total_tokens,
usage_type,
prompt_tokens,
completion_tokens,
total_tokens,
)
return record
except Exception:
logger.warning(
"[TokenTracking] failed to record %s token usage", usage_type, exc_info=True,
"[TokenTracking] failed to record %s token usage",
usage_type,
exc_info=True,
)
return None