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

feat: write tenantid to llmusage from API request params

Browse Source 
Populate llmusage.tenantid from params_kw.tenantid in all 6 inference paths:
syncinference, product_interface, llmclient, asyncinference
This commit is contained in:
yumoqing 2026-07-09 11:38:43 +08:00
parent 8f63d20f57
commit fd65b5a59e
4 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llmage/asyncinference.py
View File

@ -107,6 +107,7 @@ async def async_uapi_request(request, llm,
llmusage.finish_seconds = finish_seconds
llmusage.status = d.status
llmusage.userorgid = callerorgid
llmusage.tenantid = params_kw.get('tenantid', params_kw.get('tentantid'))
llmusage.ownerid = llm.ownerid
llmusage.accounting_status = 'created'
b = json.dumps(d, ensure_ascii=False)
@ -258,6 +259,7 @@ async def async_uapi_request_product(llm, api_userid, user_id, user_org_id, para
llmusage.finish_seconds = finish_seconds
llmusage.status = d.status
llmusage.userorgid = user_org_id
llmusage.tenantid = params_kw.get('tenantid', params_kw.get('tentantid'))
llmusage.ownerid = llm.ownerid
llmusage.accounting_status = 'created'
await write_llmusage(llmusage)

1
llmage/llmclient.py
View File

@ -87,6 +87,7 @@ async def uapi_request(request, llm, callerid, callerorgid, params_kw=None):
llmusage.finish_seconds = finish_seconds
llmusage.status = 'SUCCEEDED'
llmusage.userorgid = callerorgid
llmusage.tenantid = params_kw.get('tenantid', params_kw.get('tentantid'))
llmusage.ownerid = llm.ownerid
llmusage.accounting_status = 'created'
await write_llmusage(llmusage)

2
llmage/product_interface.py
View File

@ -299,6 +299,7 @@ async def _collect_stream(llm, api_userid, user_id, user_org_id, params_kw, luid
llmusage.finish_seconds = finish_seconds
llmusage.status = 'SUCCEEDED'
llmusage.userorgid = user_org_id
llmusage.tenantid = params_kw.get('tenantid', params_kw.get('tentantid'))
llmusage.ownerid = llm.ownerid
llmusage.accounting_status = 'created'
await write_llmusage(llmusage)
@ -401,6 +402,7 @@ async def execute_product_service_stream(resource_ref_id, user_id, user_org_id,
llmusage.finish_seconds = finish_seconds
llmusage.status = 'SUCCEEDED'
llmusage.userorgid = user_org_id
llmusage.tenantid = params_kw.get('tenantid', params_kw.get('tentantid'))
llmusage.ownerid = full_llm.ownerid
llmusage.accounting_status = 'created'
await write_llmusage(llmusage)

2
llmage/syncinference.py
View File

@ -78,6 +78,7 @@ async def sync_uapi_request(request, llm, callerid, callerorgid, params_kw=None)
llmusage.cost = 0
"""
llmusage.userorgid = callerorgid
llmusage.tenantid = params_kw.get('tenantid', params_kw.get('tentantid'))
llmusage.ownerid = llm.ownerid
llmusage.accounting_status = 'created'
b = json.dumps(d, ensure_ascii=False)
@ -134,6 +135,7 @@ async def sync_uapi_request_product(llm, api_userid, user_id, user_org_id, param
llmusage.status = 'SUCCEEDED'
llmusage.amount = 0.00
llmusage.userorgid = user_org_id
llmusage.tenantid = params_kw.get('tenantid', params_kw.get('tentantid'))
llmusage.ownerid = llm.ownerid
llmusage.accounting_status = 'created'
await write_llmusage(llmusage)