From 44309ce0d2ed740bb132b789bef33437dd458dc0 Mon Sep 17 00:00:00 2001
From: yumoqing <yumoqing@gmail.com>
Date: Fri, 10 Apr 2026 17:19:19 +0800
Subject: [PATCH] bugfix

---
 llmage/accounting.py     | 1 +
 llmage/asyncinference.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/llmage/accounting.py b/llmage/accounting.py
index d230873..b8cdddb 100644
--- a/llmage/accounting.py
+++ b/llmage/accounting.py
@@ -9,6 +9,7 @@ from sqlor.dbpools import get_sor_context
 from ahserver.serverenv import ServerEnv
 from accounting.consume import consume_accounting
 from accounting.getaccount import getCustomerBalance
+from .utils import *
 
 async def llm_charging(ppid, llmusage):
 	env = ServerEnv()
diff --git a/llmage/asyncinference.py b/llmage/asyncinference.py
index 660e989..72f7162 100644
--- a/llmage/asyncinference.py
+++ b/llmage/asyncinference.py
@@ -183,7 +183,7 @@ async def query_task_status(request, luid, onetime=False):
 				await append_new_llmoutput(llmusage.ioinfo, new_output)
 				await modify_llmusage_status(llmusage)
 			if  llmusage.status in ['FAILED', 'SUCCEEDED']:
-				dcritical(f'finished .. {llmusage.status=}')
+				critical(f'finished .. {llmusage.status=}')
 				return
 
 			if onetime: