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

feat: collection rag integration

Browse Source
This commit is contained in:
Timothy J. Baek
2024-02-03 15:57:06 -08:00
parent 7d2f788a3b
commit 683650ec00
5 changed files with 112 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -97,15 +97,15 @@ async def get_status():
return {"status": True}
class QueryCollectionForm(BaseModel):
class QueryDocForm(BaseModel):
collection_name: str
query: str
k: Optional[int] = 4
@app.post("/query/collection")
def query_collection(
form_data: QueryCollectionForm,
@app.post("/query/doc")
def query_doc(
form_data: QueryDocForm,
user=Depends(get_current_user),
):
try:
@@ -173,8 +173,8 @@ def merge_and_sort_query_results(query_results, k):
return merged_query_results
@app.post("/query/collections")
def query_collections(
@app.post("/query/collection")
def query_collection(
form_data: QueryCollectionsForm,
user=Depends(get_current_user),
):