yumoqing
07b4893252
- 新增 llmusage_history 表：定时备份已记账(use_date<today)的历史记录 - 新增 llmusage_accounting_failed 表：记录记账失败详情，支持检索 - 新增 backup_accounted_llmusage() 函数：备份+清理历史数据 - 新增 get_failed_accounting_records() 函数：按条件检索失败记录 - 更新 llm_accoung_failed()：同时写入失败表记录 - 新增 failed_accounting.ui 页面和 failed_accounting_list.dspy API - 新增 llmusage CRUD API (create/update/delete) - 新增表索引优化查询性能 - 更新 setup_llmage_perms.sh 添加新端点权限 - 生成生产迁移SQL: scripts/migrate_llmusage_history.sql
77 lines
2.6 KiB
Python
77 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
|
import json
|
|
import os
|
|
|
|
result = {'success': False, 'rows': [], 'total': 0, 'page': 1, 'page_size': 50}
|
|
|
|
try:
|
|
dbname = get_module_dbname('llmage')
|
|
user_orgid = await get_userorgid()
|
|
|
|
# Extract filter parameters from params_kw
|
|
filters = {}
|
|
if params_kw.get('userorgid'):
|
|
filters['userorgid'] = params_kw.get('userorgid')
|
|
if params_kw.get('llmid'):
|
|
filters['llmid'] = params_kw.get('llmid')
|
|
if params_kw.get('handled') is not None:
|
|
filters['handled'] = params_kw.get('handled')
|
|
if params_kw.get('start_date'):
|
|
filters['start_date'] = params_kw.get('start_date')
|
|
if params_kw.get('end_date'):
|
|
filters['end_date'] = params_kw.get('end_date')
|
|
|
|
page = int(params_kw.get('page', 1))
|
|
page_size = int(params_kw.get('page_size', 50))
|
|
|
|
async with DBPools().sqlorContext(dbname) as sor:
|
|
# Build dynamic SQL
|
|
conditions = []
|
|
ns = {}
|
|
|
|
# Default: show unhandled records
|
|
if 'handled' not in filters:
|
|
conditions.append("handled='0'")
|
|
|
|
if filters.get('userorgid'):
|
|
conditions.append("userorgid=${userorgid}$")
|
|
ns['userorgid'] = filters['userorgid']
|
|
if filters.get('llmid'):
|
|
conditions.append("llmid=${llmid}$")
|
|
ns['llmid'] = filters['llmid']
|
|
if filters.get('handled') is not None:
|
|
conditions.append("handled=${handled}$")
|
|
ns['handled'] = filters['handled']
|
|
if filters.get('start_date'):
|
|
conditions.append("use_date>=${start_date}$")
|
|
ns['start_date'] = filters['start_date']
|
|
if filters.get('end_date'):
|
|
conditions.append("use_date<=${end_date}$")
|
|
ns['end_date'] = filters['end_date']
|
|
|
|
where = ""
|
|
if conditions:
|
|
where = "where " + " and ".join(conditions)
|
|
|
|
# Count total
|
|
count_sql = f"select count(*) as cnt from llmusage_accounting_failed {where}"
|
|
count_recs = await sor.sqlExe(count_sql, ns)
|
|
total = count_recs[0].cnt if count_recs else 0
|
|
|
|
# Query with pagination
|
|
offset = (page - 1) * page_size
|
|
query_sql = f"""select * from llmusage_accounting_failed {where}
|
|
order by failed_time desc limit {page_size} offset {offset}"""
|
|
recs = await sor.sqlExe(query_sql, ns)
|
|
|
|
result['rows'] = [dict(r) for r in (recs or [])]
|
|
result['total'] = total
|
|
result['page'] = page
|
|
result['page_size'] = page_size
|
|
result['success'] = True
|
|
|
|
except Exception as e:
|
|
result['error'] = str(e)
|
|
|
|
return json.dumps(result, ensure_ascii=False, default=str)
|