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:23:18 +08:00
parent e657196ce2
commit f16e6761f4
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
llmage/init.py
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 (
query_order, llm_query_order,
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.query_order = query_order env.llm_query_order = llm_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

2
llmage/utils.py
View File

@ -14,7 +14,7 @@ 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): async def llm_query_order(userorgid):
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}) recs = await sor.R('llmusage', {'userorgid': userorgid})