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 2025-10-10 15:00:08 +08:00
parent 4efae00ea3
commit b79378b04f
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
wwwroot/llminference.dspy
View File

@ -4,12 +4,17 @@ if userid is None:
return UiError(title='llm inference', message='Please login first') return UiError(title='llm inference', message='Please login first')
kdbids = params_kw.kdbids kdbids = params_kw.kdbids
if kdbids: if kdbids:
ret = await fused_search(params_kw.prompt, f = RegisterFunction().get('fusedsearch')
if f:
ret = await fusedsearch(params_kw.prompt,
userid=userid, userid=userid,
knowledge_base_ids = kdbids, knowledge_base_ids = kdbids,
limit=params_kw.recall_cnt or 5) limit=params_kw.recall_cnt or 5)
ret['prompt'] = params_kw.prompt ret['prompt'] = params_kw.prompt
params_kw.prompt = str_tmpl_render(params_kw.prompt_tmpl, ret) params_kw.prompt = str_tmpl_render(params_kw.prompt_tmpl, ret)
debug(f'rag return {ret}, {params_kw.prompt=}') debug(f'rag return {ret}, {params_kw.prompt=}')
else:
debug(f'"fusedsearch" function not registed')
env = DictObject(**globals()) env = DictObject(**globals())
return await inference(request, env=env) return await inference(request, env=env)