UniqAI
/
open-webui
mirror of https://github.com/open-webui/open-webui
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refac

This commit is contained in:
Timothy J. Baek 2024-08-27 15:53:29 +02:00
parent 09cba5b87a
commit 69c4687a53
1 changed files with 0 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
backend/apps/rag/main.py
View File

@ -583,19 +583,9 @@ async def get_query_settings(user=Depends(get_admin_user)):
}
@app.get("/file/limit/settings")
async def get_query_settings(user=Depends(get_verified_user)):
return {
"FILE_MAX_SIZE": app.state.config.FILE_MAX_SIZE,
"FILE_MAX_COUNT": app.state.config.FILE_MAX_COUNT,
}
class QuerySettingsForm(BaseModel):
k: Optional[int] = None
r: Optional[float] = None
FILE_MAX_SIZE: Optional[int] = None
FILE_MAX_COUNT: Optional[int] = None
template: Optional[str] = None
hybrid: Optional[bool] = None
@ -612,20 +602,12 @@ async def update_query_settings(
app.state.config.ENABLE_RAG_HYBRID_SEARCH = (
form_data.hybrid if form_data.hybrid else False
)
app.state.config.FILE_MAX_SIZE = (
form_data.FILE_MAX_SIZE if form_data.FILE_MAX_SIZE else 10
)
app.state.config.FILE_MAX_COUNT = (
form_data.FILE_MAX_COUNT if form_data.FILE_MAX_COUNT else 5
)
return {
"status": True,
"template": app.state.config.RAG_TEMPLATE,
"k": app.state.config.TOP_K,
"r": app.state.config.RELEVANCE_THRESHOLD,
"FILE_MAX_SIZE": app.state.config.FILE_MAX_SIZE,
"FILE_MAX_COUNT": app.state.config.FILE_MAX_COUNT,
"hybrid": app.state.config.ENABLE_RAG_HYBRID_SEARCH,
}