From fa1a107b5271540b53478777f59ba7ee7abfed5c Mon Sep 17 00:00:00 2001
From: yumoqing <yumoqing@gmail.com>
Date: Tue, 31 Mar 2026 18:19:15 +0800
Subject: [PATCH] bugfix

---
 wwwroot/vidu_inference.dspy | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/wwwroot/vidu_inference.dspy b/wwwroot/vidu_inference.dspy
index e8db6b0..2d3d265 100644
--- a/wwwroot/vidu_inference.dspy
+++ b/wwwroot/vidu_inference.dspy
@@ -11,5 +11,12 @@ userorgid = await get_userorgid()
 if userid is None:
 	return UiError(title='llm inference', message='Please login first')
 f = await checkCustomerBalance(params_kw.llmid, userorgid)
+if not f:
+	return {
+		"status": "error",
+		"data": {
+			"message": "账户余额不够"
+		}
+	}
 env = DictObject(**globals())
 return await inference(request, env=env)