llmage/asyncinference.py

@@ -147,13 +147,10 @@ async def query_task_status(request, upappid, apinames, luid, userid, taskid):
 				if rzt.status == 'SUCCEEDED':
 					if llm.ppid:
 						try:
-							charging = await llm_charging(sor, 
+							chargings = await llm_charging(sor, 
 												llm.ppid, llmusage)
-							if charging:
-								llmusage.amount = charging.amount
-								llmusage.cost = charging.cost
-							else:
-								llmusage.amount = cost = 0.0
+							llmusage.amount = chargings.amount
+							llmusage.cost = chargings.cost
 						except Exception as e:
 							e = Exception(f'{llm.pid} charging error{e}')
 							exception(f'{e}')

llmage/llmclient.py

@@ -97,11 +97,8 @@ 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)
-				if chargings:
-					llmusage.amount = chargings.amount
-					llmusage.cost = chargings.cost
-				else:
-					llmusage.amount = llmusage.cost = 0.00
+				llmusage.amount = chargings.amount
+				llmusage.cost = chargings.cost
 			except Exception as e:
 				e = Exception(f'{llm.pid} charging error{e}')
 				exception(f'{e}')

llmage/syncinference.py

@@ -59,12 +59,9 @@ async def sync_uapi_request(request, llm, sor, callerid, callerorgid, params_kw=
 		llmusage.status = 'SUCCEEDED'
 		if llm.ppid:
 			try:
-				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
+				chargings = await llm_charging(sor, llm.ppid, llmusage)
+				llmusage.amount = chargings.amount
+				llmusage.cost = chargings.cost
 			except Exception as e:
 				e = Exception(f'{llm.pid} charging error{e}')
 				exception(f'{e}')

llmage/utils.py

@@ -149,5 +149,5 @@ where a.upappid=b.id
 async def write_llmusage(llmusage):
 	env = ServerEnv()
 	async with get_sor_context(env, 'llmage') as sor:
-		await sor.C('llmusage', llmusage)
+		await sor.C('llmusage', d)