This commit is contained in:
yumoqing 2025-10-10 15:07:06 +08:00
parent e8882b2226
commit a70419410a

View File

@ -4,13 +4,14 @@ 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 rfexe('fusedsearch', params_kw.prompt, data = {
userid=userid, 'prompt': params_kw.prompt
knowledge_base_ids = kdbids, }
limit=params_kw.recall_cnt or 5) tmpl = params_kw.prompt_tmpl
ret['prompt'] = params_kw.prompt ret = await rfexe('fusedsearch', request, params_kw)
params_kw.prompt = str_tmpl_render(params_kw.prompt_tmpl, ret) data.update(ret)
debug(f'rag return {ret}, {params_kw.prompt=}') params_kw.prompt = str_tmpl_render(tmpl, data)
debug(f'rag return {data}, {params_kw.prompt=}')
env = DictObject(**globals()) env = DictObject(**globals())
return await inference(request, env=env) return await inference(request, env=env)