diff --git a/init.py b/init.py index ad27d6b..f34cc66 100644 --- a/init.py +++ b/init.py @@ -29,10 +29,11 @@ async def kb_list_handler(request, params_kw, *args, **kwargs): userorgid = await env.get_userorgid() async with get_sor_context(env, 'rag') as sor: recs = await sor.R("knowledge_bases", {"org_id": userorgid}) - return json.dumps({ - "status": "SUCCEEDED", - "knowledge_bases": [dict(r) for r in recs] - }, ensure_ascii=False, default=str) + 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" + return json.dumps(result, ensure_ascii=False, default=str) except Exception as e: exception(f"kb_list: {e}, {format_exc()}") return json.dumps({"error": str(e)})