yumoqing/rag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: dynamic storage stats display via storage_stats.dspy

Browse Source
This commit is contained in:
yumoqing 2026-07-30 15:45:07 +08:00
parent f63f742f8e
commit f968f9f578
2 changed files with 43 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
wwwroot/knowledge_bases_list/index.ui
View File

@ -77,61 +77,11 @@
] ]
}, },
{ {
"widgettype": "VBox", "widgettype": "urlwidget",
"options": { "options": {
"width": "100%", "url": "{{entire_url('./storage_stats.dspy')}}"
"bgcolor": "#f8f9fa",
"padding": "16px",
"css": "card",
"spacing": "4px"
},
"subwidgets": [
{
"widgettype": "HBox",
"options": {
"width": "100%",
"justifyContent": "space-between"
},
"subwidgets": [
{
"widgettype": "Text",
"options": {
"text": "💾 存储容量",
"cfontsize": 14,
"fontWeight": "bold",
"color": "#333"
} }
}, },
{
"widgettype": "Text",
"options": {
"text": "已用 0MB / 总计 100MB",
"cfontsize": 12,
"color": "#888"
}
}
]
},
{
"widgettype": "VBox",
"options": {
"width": "100%",
"cheight": 0.5,
"bgcolor": "#e0e0e0"
},
"subwidgets": [
{
"widgettype": "VBox",
"options": {
"cwidth": 0,
"cheight": 0.5,
"bgcolor": "#4a90d9"
}
}
]
}
]
},
{ {
"widgettype": "urlwidget", "widgettype": "urlwidget",
"options": { "options": {

40
wwwroot/knowledge_bases_list/storage_stats.dspy Normal file
View File

@ -0,0 +1,40 @@
ns = params_kw.copy()
env = request._run_ns
async with get_sor_context(env, 'rag') as sor:
kb_id = ns.get('kb_id', '')
if kb_id:
recs = await sor.sqlExe("SELECT doc_count, total_size, chunk_count FROM knowledge_bases WHERE id=${id}$", {"id": kb_id})
else:
recs = await sor.sqlExe("SELECT COALESCE(SUM(doc_count),0) AS doc_count, COALESCE(SUM(total_size),0) AS total_size, COALESCE(SUM(chunk_count),0) AS chunk_count FROM knowledge_bases", {})
if recs:
r = recs[0]
docs = r.doc_count or 0
size = r.total_size or 0
chunks = r.chunk_count or 0
else:
docs = 0
size = 0
chunks = 0
def fmt(sz):
if sz < 1024: return str(sz) + 'B'
elif sz < 1024*1024: return str(round(sz/1024,1)) + 'KB'
else: return str(round(sz/(1024*1024),1)) + 'MB'
pct = min(round(size / (100*1024*1024) * 100, 1), 100) if size > 0 else 0
return json.dumps({
"widgettype": "VBox",
"options": {"width": "100%", "bgcolor": "#f8f9fa", "padding": "16px", "css": "card", "spacing": "4px"},
"subwidgets": [
{"widgettype": "HBox", "options": {"width": "100%", "justifyContent": "space-between"}, "subwidgets": [
{"widgettype": "Text", "options": {"text": "💾 存储容量", "cfontsize": 14, "fontWeight": "bold", "color": "#333"}},
{"widgettype": "Text", "options": {"text": f"已用 {fmt(size)} / 总计 100MB | {docs} 文档 {chunks} 块", "cfontsize": 12, "color": "#888"}}
]},
{"widgettype": "VBox", "options": {"width": "100%", "cheight": 0.5, "bgcolor": "#e0e0e0"}, "subwidgets": [
{"widgettype": "VBox", "options": {"cwidth": pct, "cheight": 0.5, "bgcolor": "#4a90d9"}}
]}
]
}, ensure_ascii=False)