Fix middleware.py missing prefix on embedding function

This commit is contained in:
Phlogi 2025-03-31 15:19:07 +02:00 committed by GitHub
parent 6ac153c989
commit bc64b2a67f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -596,8 +596,8 @@ async def chat_completion_files_handler(
request=request,
files=files,
queries=queries,
embedding_function=lambda query: request.app.state.EMBEDDING_FUNCTION(
query, user=user
embedding_function=lambda query, prefix: request.app.state.EMBEDDING_FUNCTION(
query, prefix=prefix, user=user
),
k=request.app.state.config.TOP_K,
reranking_function=request.app.state.rf,