This commit is contained in:
yumoqing 2026-04-03 14:34:58 +08:00
parent f16e6761f4
commit 79adf49709
2 changed files with 23 additions and 6 deletions

View File

@ -4,7 +4,7 @@ from ahserver.serverenv import ServerEnv
from .keling import keling_token from .keling import keling_token
from .jimeng import jimeng_auth_headers from .jimeng import jimeng_auth_headers
from .utils import ( from .utils import (
llm_query_order, llm_query_orders,
get_llm_by_model get_llm_by_model
) )
@ -33,7 +33,7 @@ from .asyncinference import (
def load_llmage(): def load_llmage():
env = ServerEnv() 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.get_llm_by_model = get_llm_by_model
env.llm_charging = llm_charging env.llm_charging = llm_charging
env.llm_accounting = llm_accounting env.llm_accounting = llm_accounting

View File

@ -14,12 +14,29 @@ from uapi.appapi import UAPI, sor_get_callerid, sor_get_uapi
from ahserver.serverenv import get_serverenv, ServerEnv from ahserver.serverenv import get_serverenv, ServerEnv
from ahserver.filestorage import FileStorage from ahserver.filestorage import FileStorage
async def llm_query_order(userorgid): async def llm_query_orders(userorgid, page, pagerows=80):
env = ServerEnv() env = ServerEnv()
async with get_sor_context(env, 'llmage') as sor: async with get_sor_context(env, 'llmage') as sor:
recs = await sor.R('llmusage', {'userorgid': userorgid}) sql = """select llmid,
return recs use_date,
return [] 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): async def get_llm_by_model(id, lctype=None):
env = ServerEnv() env = ServerEnv()