From 79adf497093bd1869470e604424cbe7068eb4b83 Mon Sep 17 00:00:00 2001 From: yumoqing Date: Fri, 3 Apr 2026 14:34:58 +0800 Subject: [PATCH] bugfix --- llmage/init.py | 4 ++-- llmage/utils.py | 25 +++++++++++++++++++++---- 2 files changed, 23 insertions(+), 6 deletions(-) diff --git a/llmage/init.py b/llmage/init.py index 0c423fa..04c5af7 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 ( - llm_query_order, + llm_query_orders, get_llm_by_model ) @@ -33,7 +33,7 @@ from .asyncinference import ( def load_llmage(): env = ServerEnv() - env.llm_query_order = llm_query_order + env.llm_query_orders = llm_query_orders 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 3bc819a..edb842d 100644 --- a/llmage/utils.py +++ b/llmage/utils.py @@ -14,12 +14,29 @@ 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 llm_query_order(userorgid): +async def llm_query_orders(userorgid, page, pagerows=80): env = ServerEnv() async with get_sor_context(env, 'llmage') as sor: - recs = await sor.R('llmusage', {'userorgid': userorgid}) - return recs - return [] + sql = """select llmid, +use_date, +use_time, +userid, +usage, +status, +amount, +userorgid, +accounting_status +from llmusage +where userorgid = ${userorgid}$""" + ns = dict( + page=page, + pagerows=pagerows, + order="use_time desc", + userorgid=userorgid) + + data = await sor.sqlExe(sql, ns) + return data + return {'total': 0, 'rows':[]} async def get_llm_by_model(id, lctype=None): env = ServerEnv()