bugfix
This commit is contained in:
parent
e8882b2226
commit
a70419410a
@ -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)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user