This commit is contained in:
yumoqing 2025-08-07 18:33:39 +08:00
parent 7346d8cdb3
commit 42f6633813

22
llmage/llmclient.py Normal file
View File

@ -0,0 +1,22 @@
from traceback import format_exc
from appPublic.log import debug, exception
from uapi.appapi import UAPI
from ahserver.serverenv get_serverenv
async def inference(request, env):
uapi = UAPI(request, env)
params = env.params_kw
llmid = params.id
prompt = params.prompt
stream = prompt.stream or True
dbname = env.get_module_dbname('llmage')
db = env.DBPools()
async with db.sqlorContext(dbname) as sor:
llms = await sor.R('llm', {'id':llmid})
if len(llms) == 0:
e = Exception(f'{llmid=} not found')
exception(f'{e}\n{format_exc()}')
raise e
uapi = UAPI(request, env=env, sor=sor)
return env.stream_response(request,
uapi.stream_linify(llms[0].