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

bugfix

Browse Source
This commit is contained in:
yumoqing 2026-03-30 15:52:56 +08:00
parent 2c500c9dc3
commit 9efd0ecc67
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
llmage/init.py
View File

@ -15,7 +15,12 @@ from .llmclient import (
get_llmcatelogs, get_llmcatelogs,
get_llms_by_catelog get_llms_by_catelog
) )
from .accounting import checkCustomerBalance from .accounting import (
checkCustomerBalance,
llm_charging,
llm_accounting
)
from .asyncinference import ( from .asyncinference import (
get_asynctask_status, get_asynctask_status,
query_task_status, query_task_status,
@ -24,6 +29,8 @@ from .asyncinference import (
def load_llmage(): def load_llmage():
env = ServerEnv() env = ServerEnv()
env.llm_charging = llm_charging
env.llm_accounting = llm_accounting
env.get_today_asynctask_list = get_today_asynctask_list env.get_today_asynctask_list = get_today_asynctask_list
env.get_asynctask_status = get_asynctask_status env.get_asynctask_status = get_asynctask_status
env.query_task_status = query_task_status env.query_task_status = query_task_status