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-01-20 13:49:39 +08:00
parent 1df9f4f52d
commit 3142cc8778
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llmage/llmclient.py
View File

@ -388,6 +388,8 @@ async def inference(request, *args, params_kw=None, **kw):
db = env.DBPools() db = env.DBPools()
async with db.sqlorContext(dbname) as sor: async with db.sqlorContext(dbname) as sor:
llm = await get_llm(llmid) llm = await get_llm(llmid)
if params_kw.nostream and llm.stream == 'stream':
llm.stream = 'sync'
if llm.stream == 'async': if llm.stream == 'async':
f = partial(async_uapi_request, request, llm, sor, params_kw=params_kw) f = partial(async_uapi_request, request, llm, sor, params_kw=params_kw)
return await env.stream_response(request, f) return await env.stream_response(request, f)