diff --git a/init.py b/init.py index f34cc66..4aa987d 100644 --- a/init.py +++ b/init.py @@ -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)})