From f16e6761f4d19cf0f93f9440a25c295617454467 Mon Sep 17 00:00:00 2001 From: yumoqing Date: Fri, 3 Apr 2026 14:23:18 +0800 Subject: [PATCH] bugfix --- llmage/init.py | 4 ++-- llmage/utils.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/llmage/init.py b/llmage/init.py index 09bb256..0c423fa 100644 --- a/llmage/init.py +++ b/llmage/init.py @@ -4,7 +4,7 @@ from ahserver.serverenv import ServerEnv from .keling import keling_token from .jimeng import jimeng_auth_headers from .utils import ( - query_order, + llm_query_order, get_llm_by_model ) @@ -33,7 +33,7 @@ from .asyncinference import ( def load_llmage(): env = ServerEnv() - env.query_order = query_order + env.llm_query_order = llm_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 401bfdf..3bc819a 100644 --- a/llmage/utils.py +++ b/llmage/utils.py @@ -14,7 +14,7 @@ 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): +async def llm_query_order(userorgid): env = ServerEnv() async with get_sor_context(env, 'llmage') as sor: recs = await sor.R('llmusage', {'userorgid': userorgid})