bugfix
This commit is contained in:
parent
12563e8218
commit
ea8ae11bbd
@ -60,17 +60,22 @@ async def docs(request, params_kw, *params, **kw):
|
|||||||
return helptext
|
return helptext
|
||||||
|
|
||||||
async def rerank(request, params_kw, *params, **kw):
|
async def rerank(request, params_kw, *params, **kw):
|
||||||
debug(f'{params_kw.input=}')
|
debug(f'{params_kw.query=}, {params_kw.documents=}, {params_kw.top_n=}')
|
||||||
se = ServerEnv()
|
se = ServerEnv()
|
||||||
engine = se.engine
|
engine = se.engine
|
||||||
f = awaitify(engine.rerank)
|
f = awaitify(engine.rerank)
|
||||||
query = params_kw.query
|
query = params_kw.query
|
||||||
if query is None:
|
if query is None:
|
||||||
e = exception(f'query is None')
|
e = Exception(f'query is None')
|
||||||
raise e
|
raise e
|
||||||
if isinstance(query, str):
|
documents = params_kw.documents
|
||||||
input = [input]
|
if documents is None:
|
||||||
arr = await f(input)
|
e = Exception(f'documents is None')
|
||||||
|
raise e
|
||||||
|
if isinstance(documents, str):
|
||||||
|
documents = [documents]
|
||||||
|
top_n = parmas_kw.top_n
|
||||||
|
arr = await f(query, params_kw.documents, top_n = top_n)
|
||||||
debug(f'{arr=}, type(arr)')
|
debug(f'{arr=}, type(arr)')
|
||||||
return arr
|
return arr
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user