fix: DataViewer expects {status, data: {rows, total}} format

This commit is contained in:
yumoqing 2026-07-22 16:24:37 +08:00
parent 7989186e3e
commit 65078340fc

10
init.py
View File

@ -30,9 +30,7 @@ async def kb_list_handler(request, params_kw, *args, **kwargs):
async with get_sor_context(env, 'rag') as sor:
recs = await sor.R("knowledge_bases", {"org_id": userorgid})
rows = [dict(r) for r in recs]
# bricks Tabular expects {rows, total} format
result = {"rows": rows, "total": len(rows)}
result["status"] = "SUCCEEDED" if rows else "empty"
result = {"status": "SUCCEEDED", "data": {"rows": rows, "total": len(rows)}}
return json.dumps(result, ensure_ascii=False, default=str)
except Exception as e:
exception(f"kb_list: {e}, {format_exc()}")
@ -47,10 +45,8 @@ async def engines_handler(request, params_kw, *args, **kwargs):
async with get_sor_context(env, 'rag') as sor:
sql = "SELECT * FROM engine_configs WHERE status='active' AND (org_id IS NULL OR org_id=${org_id}$) ORDER BY engine_type, priority DESC"
recs = await sor.sqlExe(sql, {"org_id": userorgid})
return json.dumps({
"status": "SUCCEEDED",
"engines": [dict(r) for r in recs]
}, ensure_ascii=False, default=str)
rows = [dict(r) for r in recs]
return json.dumps({"status": "SUCCEEDED", "data": {"rows": rows, "total": len(rows)}}, ensure_ascii=False, default=str)
except Exception as e:
exception(f"engines: {e}, {format_exc()}")
return json.dumps({"error": str(e)})