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 .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

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.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()