yumoqing/llmage
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
yumoqing 2026-04-03 14:22:25 +08:00
parent 2b27bfe1a4
commit e657196ce2
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llmage/init.py
View File

@ -4,6 +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 (
query_order,
get_llm_by_model get_llm_by_model
) )
@ -32,6 +33,7 @@ from .asyncinference import (
def load_llmage(): def load_llmage():
env = ServerEnv() env = ServerEnv()
env.query_order = query_order
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

7
llmage/utils.py
View File

@ -14,6 +14,13 @@ 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 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): async def get_llm_by_model(id, lctype=None):
env = ServerEnv() env = ServerEnv()
async with get_sor_context(env, 'llmage') as sor: async with get_sor_context(env, 'llmage') as sor: