diff --git a/llmage/init.py b/llmage/init.py index d14e0b3..09bb256 100644 --- a/llmage/init.py +++ b/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 diff --git a/llmage/utils.py b/llmage/utils.py index 238ecfa..401bfdf 100644 --- a/llmage/utils.py +++ b/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: