diff --git a/llmage/asyncinference.py b/llmage/asyncinference.py index 39d6d24..3895544 100644 --- a/llmage/asyncinference.py +++ b/llmage/asyncinference.py @@ -147,10 +147,13 @@ async def query_task_status(request, upappid, apinames, luid, userid, taskid): if rzt.status == 'SUCCEEDED': if llm.ppid: try: - chargings = await llm_charging(sor, + charging = await llm_charging(sor, llm.ppid, llmusage) - llmusage.amount = chargings.amount - llmusage.cost = chargings.cost + if charging: + llmusage.amount = charging.amount + llmusage.cost = charging.cost + else: + llmusage.amount = cost = 0.0 except Exception as e: e = Exception(f'{llm.pid} charging error{e}') exception(f'{e}') diff --git a/llmage/llmclient.py b/llmage/llmclient.py index 73381f7..d388cfd 100644 --- a/llmage/llmclient.py +++ b/llmage/llmclient.py @@ -97,8 +97,11 @@ async def uapi_request(request, llm, sor, callerid, callerorgid, params_kw=None) if llm.ppid and callerorgid: try: chargings = await llm_charging(sor, llm.ppid, llmusage) - llmusage.amount = chargings.amount - llmusage.cost = chargings.cost + if chargings: + llmusage.amount = chargings.amount + llmusage.cost = chargings.cost + else: + llmusage.amount = llmusage.cost = 0.00 except Exception as e: e = Exception(f'{llm.pid} charging error{e}') exception(f'{e}') diff --git a/llmage/syncinference.py b/llmage/syncinference.py index 740077a..5ee92bc 100644 --- a/llmage/syncinference.py +++ b/llmage/syncinference.py @@ -59,9 +59,12 @@ async def sync_uapi_request(request, llm, sor, callerid, callerorgid, params_kw= llmusage.status = 'SUCCEEDED' if llm.ppid: try: - chargings = await llm_charging(sor, llm.ppid, llmusage) - llmusage.amount = chargings.amount - llmusage.cost = chargings.cost + charging = await llm_charging(sor, llm.ppid, llmusage) + if charging: + llmusage.amount = charging.amount + llmusage.cost = charging.cost + else: + llmusage.amount = llmusage.cost = 0.0 except Exception as e: e = Exception(f'{llm.pid} charging error{e}') exception(f'{e}')