fix: bricks Tabular expects {rows,total} format in API response
This commit is contained in:
parent
f814c9bf08
commit
03d4d70edd
9
init.py
9
init.py
@ -29,10 +29,11 @@ async def kb_list_handler(request, params_kw, *args, **kwargs):
|
|||||||
userorgid = await env.get_userorgid()
|
userorgid = await env.get_userorgid()
|
||||||
async with get_sor_context(env, 'rag') as sor:
|
async with get_sor_context(env, 'rag') as sor:
|
||||||
recs = await sor.R("knowledge_bases", {"org_id": userorgid})
|
recs = await sor.R("knowledge_bases", {"org_id": userorgid})
|
||||||
return json.dumps({
|
rows = [dict(r) for r in recs]
|
||||||
"status": "SUCCEEDED",
|
# bricks Tabular expects {rows, total} format
|
||||||
"knowledge_bases": [dict(r) for r in recs]
|
result = {"rows": rows, "total": len(rows)}
|
||||||
}, ensure_ascii=False, default=str)
|
result["status"] = "SUCCEEDED" if rows else "empty"
|
||||||
|
return json.dumps(result, ensure_ascii=False, default=str)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
exception(f"kb_list: {e}, {format_exc()}")
|
exception(f"kb_list: {e}, {format_exc()}")
|
||||||
return json.dumps({"error": str(e)})
|
return json.dumps({"error": str(e)})
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user