yumoqing/llmage
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(llmage): 修复备份SQL条件 use_date < today（非yesterday）+ 提高备份频率到每5分钟

Browse Source
This commit is contained in:
yumoqing 2026-05-24 17:05:58 +08:00
parent 96317c1512
commit 8755ab5d6d
3 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
llmage/__pycache__/init.cpython-311.pyc
View File

Binary file not shown.

BIN
llmage/__pycache__/utils.cpython-311.pyc
View File

Binary file not shown.

16
llmage/accounting.py
View File

@ -240,19 +240,19 @@ async def llm_accoung_failed(luid, reason=None):
async def backup_accounted_llmusage(): async def backup_accounted_llmusage():
"""Backup yesterday's accounted records to history table and remove from llmusage.""" """Backup accounted records with use_date before today to history table."""
env = ServerEnv() env = ServerEnv()
yesterday = (datetime.now() - timedelta(days=1)).strftime('%Y-%m-%d') today = datetime.now().strftime('%Y-%m-%d')
ts = env.timestampstr() ts = env.timestampstr()
batched = 0 batched = 0
async with get_sor_context(env, 'llmage') as sor: async with get_sor_context(env, 'llmage') as sor:
# Select yesterday's accounted records # Select records with use_date < today (i.e. yesterday and earlier)
sql = """select * from llmusage sql = """select * from llmusage
where accounting_status='accounted' where accounting_status='accounted'
and use_date < ${yesterday}$""" and use_date < ${today}$"""
recs = await sor.sqlExe(sql, {'yesterday': yesterday}) recs = await sor.sqlExe(sql, {'today': today})
if not recs: if not recs:
debug(f'backup_accounted_llmusage: no records to backup for use_date < {yesterday}') debug(f'backup_accounted_llmusage: no records to backup for use_date < {today}')
return 0 return 0
debug(f'backup_accounted_llmusage: {len(recs)} records to backup') debug(f'backup_accounted_llmusage: {len(recs)} records to backup')
for r in recs: for r in recs:
@ -356,9 +356,9 @@ async def backend_accounting():
exception(f'{e}, {lu.id=}') exception(f'{e}, {lu.id=}')
await llm_accoung_failed(lu.id, reason=str(e)) await llm_accoung_failed(lu.id, reason=str(e))
# Run backup every 100 iterations (roughly every ~1000 seconds) # Run backup every 30 iterations (~5 minutes)
backup_counter += 1 backup_counter += 1
if backup_counter >= 100: if backup_counter >= 30:
backup_counter = 0 backup_counter = 0
try: try:
await backup_accounted_llmusage() await backup_accounted_llmusage()