Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 20:07:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #9068 from df-cgdm/main

Browse Source 
**feat** Add user related headers when calling an external embedding api
This commit is contained in:
Timothy Jaeryang Baek
2025-02-05 00:05:44 -08:00
committed by GitHub
parent 8e831a5433 6ca295ec59
commit f6f8c08cb0
6 changed files with 70 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/open_webui/utils/middleware.py
View File

@@ -634,7 +634,7 @@ async def chat_completion_files_handler(
lambda: get_sources_from_files(
files=files,
queries=queries,
embedding_function=request.app.state.EMBEDDING_FUNCTION,
embedding_function=lambda query: request.app.state.EMBEDDING_FUNCTION(query,user=user),
k=request.app.state.config.TOP_K,
reranking_function=request.app.state.rf,
r=request.app.state.config.RELEVANCE_THRESHOLD,