bugfix

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}')

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)
+				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}')

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}')

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', d)
+		await sor.C('llmusage', llmusage)