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: