refac
This commit is contained in:
@@ -2345,8 +2345,8 @@ def query_collection_handler(
|
|||||||
k=form_data.k if form_data.k else request.app.state.config.TOP_K,
|
k=form_data.k if form_data.k else request.app.state.config.TOP_K,
|
||||||
reranking_function=(
|
reranking_function=(
|
||||||
(
|
(
|
||||||
lambda sentences: request.app.state.RERANKING_FUNCTION(
|
lambda query, documents: request.app.state.RERANKING_FUNCTION(
|
||||||
sentences, user=user
|
query, documents, user=user
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
if request.app.state.RERANKING_FUNCTION
|
if request.app.state.RERANKING_FUNCTION
|
||||||
|
|||||||
@@ -984,8 +984,8 @@ async def chat_completion_files_handler(
|
|||||||
k=request.app.state.config.TOP_K,
|
k=request.app.state.config.TOP_K,
|
||||||
reranking_function=(
|
reranking_function=(
|
||||||
(
|
(
|
||||||
lambda sentences: request.app.state.RERANKING_FUNCTION(
|
lambda query, documents: request.app.state.RERANKING_FUNCTION(
|
||||||
sentences, user=user
|
query, documents, user=user
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
if request.app.state.RERANKING_FUNCTION
|
if request.app.state.RERANKING_FUNCTION
|
||||||
|
|||||||
Reference in New Issue
Block a user