bugfix

llmage/init.py

@@ -4,6 +4,7 @@ from ahserver.serverenv import ServerEnv
 from .keling import keling_token
 from .jimeng import jimeng_auth_headers
 from .utils import (
+	query_order,
 	get_llm_by_model
 )
 
@@ -32,6 +33,7 @@ from .asyncinference import (
 
 def load_llmage():
 	env = ServerEnv()
+	env.query_order = query_order
 	env.get_llm_by_model = get_llm_by_model
 	env.llm_charging = llm_charging
 	env.llm_accounting = llm_accounting

llmage/utils.py

@@ -14,6 +14,13 @@ from uapi.appapi import UAPI, sor_get_callerid, sor_get_uapi
 from ahserver.serverenv import get_serverenv, ServerEnv
 from ahserver.filestorage import FileStorage
 
+async def query_order(userorgid):
+	env = ServerEnv()
+	async with get_sor_context(env, 'llmage') as sor:
+		recs = await sor.R('llmusage', {'userorgid': userorgid})
+		return recs
+	return []
+
 async def get_llm_by_model(id, lctype=None):
 	env = ServerEnv()
 	async with get_sor_context(env, 'llmage') as sor: