diff --git a/llmage/llmclient.py b/llmage/llmclient.py index c862f88..04f34d7 100644 --- a/llmage/llmclient.py +++ b/llmage/llmclient.py @@ -14,8 +14,8 @@ from uapi.appapi import UAPI, sor_get_callerid, sor_get_uapi from ahserver.serverenv import get_serverenv, ServerEnv from ahserver.filestorage import FileStorage from .asyncinference import async_uapi_request -from .syncinference import syncinference -from .accounting import llm_accounting +from .syncinference import sync_uapi_request +from .accounting import llm_accounting, llm_charging def erase_apikey(e): e = str(e) @@ -373,7 +373,7 @@ async def query_task_status(request, upappid, apinames, luid, userid, taskid): if rzt.status == 'SUCCEEDED': if llm.ppid: try: - chargings = await llm_charginng(sor, + chargings = await llm_charging(sor, llm.ppid, callerid, usage) llmusage.amount = chargings.amount llmusage.cost = chargings.cost