171 lines
5.9 KiB
Plaintext
171 lines
5.9 KiB
Plaintext
# llm_launch_check_page.dspy
|
||
# 服务端渲染模型上线检查页面，输出bricks widget JSON
|
||
# 所有检查在服务端async执行，结果直接嵌入widget
|
||
|
||
try:
|
||
llmid = params_kw.get('id', '')
|
||
|
||
if not llmid:
|
||
return json.dumps({
|
||
"widgettype": "VBox",
|
||
"options": {"width": "100%", "height": "100%", "spacing": 10},
|
||
"subwidgets": [
|
||
{"widgettype": "Title", "options": {"text": "模型上线检查", "level": 2}},
|
||
{"widgettype": "Text", "options": {"text": "缺少模型ID参数", "i18n": False}}
|
||
]
|
||
}, ensure_ascii=False)
|
||
|
||
# --- 服务端执行所有检查 ---
|
||
checks = []
|
||
all_passed = True
|
||
llm_name = ''
|
||
llm_model = ''
|
||
|
||
async def add_check(name, passed, detail=''):
|
||
global all_passed
|
||
checks.append({'name': name, 'passed': passed, 'detail': detail})
|
||
if not passed:
|
||
all_passed = False
|
||
|
||
async with get_sor_context(request._run_ns, 'llmage') as sor:
|
||
recs = await sor.sqlExe(
|
||
"select * from llm where id=${llmid}$", {'llmid': llmid})
|
||
if not recs:
|
||
await add_check('模型记录', False, f'llm id={llmid} 不存在')
|
||
else:
|
||
llm = recs[0]
|
||
llm_name = llm.name or ''
|
||
llm_model = llm.model or ''
|
||
await add_check('模型记录', True, f'{llm_name} ({llm_model})')
|
||
|
||
date_ok = bool(llm.enabled_date and llm.expired_date)
|
||
status_ok = llm.status == 'published'
|
||
await add_check('日期与状态', date_ok and status_ok,
|
||
f"启用:{llm.enabled_date} 失效:{llm.expired_date} 状态:{llm.status}")
|
||
|
||
recs2 = await sor.sqlExe(
|
||
"select a.* from llm a, upapp b where a.id=${llmid}$ and a.upappid=b.id",
|
||
{'llmid': llmid})
|
||
await add_check('上位系统(upapp)', bool(recs2),
|
||
f'upappid={llm.upappid}' + ('' if recs2 else ' 未找到关联'))
|
||
|
||
recs3 = await sor.sqlExe("""
|
||
select a.*, e.ioid, e.stream
|
||
from llm a
|
||
join llm_api_map m on a.id = m.llmid
|
||
join upapp c on a.upappid = c.id
|
||
join uapi e on c.id = e.upappid and m.apiname = e.name
|
||
where a.id=${llmid}$""", {'llmid': llmid})
|
||
if recs3:
|
||
await add_check('API映射(uapi)', True, f'ioid={recs3[0].ioid}, stream={recs3[0].stream}')
|
||
ioid = recs3[0].ioid
|
||
recs4 = await sor.sqlExe(
|
||
"select * from uapiio where id=${ioid}$", {'ioid': ioid})
|
||
await add_check('IO定义(uapiio)', bool(recs4),
|
||
f'uapiio id={ioid}' if recs4 else f'ioid={ioid} 未找到')
|
||
else:
|
||
await add_check('API映射(uapi)', False, f'apiname={getattr(llm, "apiname", "N/A")} 在upapp中未找到')
|
||
await add_check('IO定义(uapiio)', False, '依赖 uapi 未通过')
|
||
|
||
maps = await sor.sqlExe(
|
||
"select * from llm_api_map where llmid=${llmid}$", {'llmid': llmid})
|
||
ppids = []
|
||
if maps:
|
||
ppids = [m.ppid for m in maps if m.ppid]
|
||
await add_check('能力映射(llm_api_map)', True, f'{len(maps)}条记录, {len(ppids)}个有定价')
|
||
else:
|
||
await add_check('能力映射(llm_api_map)', False, '无映射记录')
|
||
|
||
if ppids:
|
||
ppid = ppids[0]
|
||
async with get_sor_context(request._run_ns, 'pricing') as psor:
|
||
pregs = await psor.sqlExe(
|
||
"select * from pricing_program where id=${ppid}$", {'ppid': ppid})
|
||
if pregs:
|
||
await add_check('定价项目(pricing_program)', True, f'{pregs[0].name} (id={ppid})')
|
||
datas = await psor.sqlExe(
|
||
"select count(*) as cnt from pricingdata where ppid=${ppid}$", {'ppid': ppid})
|
||
cnt = datas[0].cnt if datas else 0
|
||
await add_check('定价数据(pricingdata)', cnt > 0,
|
||
f'{cnt}条记录' if cnt > 0 else '无定价数据')
|
||
else:
|
||
await add_check('定价项目(pricing_program)', False, f'ppid={ppid} 未找到')
|
||
await add_check('定价数据(pricingdata)', False, '依赖定价项目未通过')
|
||
else:
|
||
await add_check('定价项目(pricing_program)', False, 'llm_api_map中无ppid')
|
||
await add_check('定价数据(pricingdata)', False, '无定价项目')
|
||
|
||
# --- 构建widget JSON ---
|
||
api_url = entire_url('./api/llm_launch_check_api.dspy')
|
||
status_text = '全部通过 ✅' if all_passed else '存在问题 ❌ 请检查下方详情'
|
||
|
||
# 检查项列表：每项一行Text widget
|
||
check_widgets = []
|
||
for c in checks:
|
||
icon = '✅' if c['passed'] else '❌'
|
||
text = f"{icon} {c['name']}: {c['detail']}"
|
||
check_widgets.append({
|
||
"widgettype": "Text",
|
||
"options": {"text": text, "i18n": False}
|
||
})
|
||
|
||
# 按钮和结果显示区
|
||
subwidgets = [
|
||
{"widgettype": "Title", "options": {"text": f"模型上线检查 — {llm_name}", "level": 2}},
|
||
{"widgettype": "Text", "options": {"text": status_text, "i18n": False}},
|
||
] + check_widgets
|
||
|
||
if all_passed:
|
||
subwidgets.append({
|
||
"widgettype": "Button",
|
||
"options": {"name": "test_btn", "label": "体验一次", "i18n": False},
|
||
"binds": [{
|
||
"wid": "self",
|
||
"event": "click",
|
||
"actiontype": "urldata",
|
||
"target": "test_result",
|
||
"options": {
|
||
"url": f"{api_url}?llmid={llmid}&action=inference"
|
||
}
|
||
}]
|
||
})
|
||
subwidgets.append({
|
||
"widgettype": "Text",
|
||
"options": {"name": "test_result", "text": "", "i18n": False}
|
||
})
|
||
subwidgets.append({
|
||
"widgettype": "Button",
|
||
"options": {"name": "charge_btn", "label": "检查计费", "i18n": False},
|
||
"binds": [{
|
||
"wid": "self",
|
||
"event": "click",
|
||
"actiontype": "urldata",
|
||
"target": "charge_result",
|
||
"options": {
|
||
"url": f"{api_url}?llmid={llmid}&action=check_charging&usages=" + json.dumps({"prompt_tokens": 1000, "completion_tokens": 500})
|
||
}
|
||
}]
|
||
})
|
||
subwidgets.append({
|
||
"widgettype": "Text",
|
||
"options": {"name": "charge_result", "text": "", "i18n": False}
|
||
})
|
||
|
||
widget = {
|
||
"widgettype": "VBox",
|
||
"options": {"width": "100%", "height": "100%", "spacing": 10},
|
||
"subwidgets": subwidgets
|
||
}
|
||
|
||
return json.dumps(widget, ensure_ascii=False)
|
||
|
||
except Exception as e:
|
||
return json.dumps({
|
||
"widgettype": "VBox",
|
||
"options": {"width": "100%", "height": "100%", "spacing": 10},
|
||
"subwidgets": [
|
||
{"widgettype": "Title", "options": {"text": "模型上线检查", "level": 2}},
|
||
{"widgettype": "Text", "options": {"text": f"执行错误: {format_exc()}", "i18n": False}}
|
||
]
|
||
}, ensure_ascii=False)
|