mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-17 03:19:59 +03:00
e53bc6807d
* 🗃️ add new tables for multiple brains * 🗑️ remove date input from fetch_user_id_from_credentials * ✨ new /brain endpoints * ♻️ refactor backend utils by splitting it into files * 💡 comments for next actions to update /upload
51 lines
2.5 KiB
Python
51 lines
2.5 KiB
Python
from auth.auth_bearer import AuthBearer, get_current_user
|
|
from fastapi import APIRouter, Depends
|
|
from models.users import User
|
|
from utils.common import CommonsDep
|
|
|
|
explore_router = APIRouter()
|
|
|
|
def get_unique_user_data(commons, user):
|
|
"""
|
|
Retrieve unique user data vectors.
|
|
"""
|
|
response = commons['supabase'].table("vectors").select(
|
|
"name:metadata->>file_name, size:metadata->>file_size", count="exact").filter("user_id", "eq", user.email).execute()
|
|
documents = response.data # Access the data from the response
|
|
# Convert each dictionary to a tuple of items, then to a set to remove duplicates, and then back to a dictionary
|
|
unique_data = [dict(t) for t in set(tuple(d.items()) for d in documents)]
|
|
return unique_data
|
|
|
|
@explore_router.get("/explore", dependencies=[Depends(AuthBearer())], tags=["Explore"])
|
|
async def explore_endpoint(commons: CommonsDep, current_user: User = Depends(get_current_user)):
|
|
"""
|
|
Retrieve and explore unique user data vectors.
|
|
"""
|
|
unique_data = get_unique_user_data(commons, current_user)
|
|
unique_data.sort(key=lambda x: int(x['size']), reverse=True)
|
|
return {"documents": unique_data}
|
|
|
|
|
|
@explore_router.delete("/explore/{file_name}", dependencies=[Depends(AuthBearer())], tags=["Explore"])
|
|
async def delete_endpoint(commons: CommonsDep, file_name: str, credentials: dict = Depends(AuthBearer())):
|
|
"""
|
|
Delete a specific user file by file name.
|
|
"""
|
|
user = User(email=credentials.get('email', 'none'))
|
|
# Cascade delete the summary from the database first, because it has a foreign key constraint
|
|
commons['supabase'].table("summaries").delete().match(
|
|
{"metadata->>file_name": file_name}).execute()
|
|
commons['supabase'].table("vectors").delete().match(
|
|
{"metadata->>file_name": file_name, "user_id": user.email}).execute()
|
|
return {"message": f"{file_name} of user {user.email} has been deleted."}
|
|
|
|
@explore_router.get("/explore/{file_name}", dependencies=[Depends(AuthBearer())], tags=["Explore"])
|
|
async def download_endpoint(commons: CommonsDep, file_name: str, current_user: User = Depends(get_current_user)):
|
|
"""
|
|
Download a specific user file by file name.
|
|
"""
|
|
response = commons['supabase'].table("vectors").select(
|
|
"metadata->>file_name, metadata->>file_size, metadata->>file_extension, metadata->>file_url", "content").match({"metadata->>file_name": file_name, "user_id": current_user.email}).execute()
|
|
documents = response.data
|
|
return {"documents": documents}
|