mirror of
https://github.com/onyx-dot-app/onyx.git
synced 2026-10-23 03:42:59 -05:00
hagen-danswer
8d443ada5b
* initial commit * almost done * finished 3 tests * minor refactor * built out initial permisison tests * reworked test_deletion * removed logging * all original tests have been converted * renamed user_groups to user_group * mypy * added test for doc set permissions * unified naming for manager methods * Refactored models and added new deletion test * minor additions * better logging+fixed input variables * commented out failed tests * Added readme * readme update * Added auth to IT set auth_type to basic and require_email_verification to false * Update run-it.yml * used verify and added to readme * added api key manager
117 lines
4.0 KiB
Python
117 lines
4.0 KiB
Python
from fastapi import APIRouter
|
|
from fastapi import Depends
|
|
from fastapi import HTTPException
|
|
from fastapi import Query
|
|
from sqlalchemy.orm import Session
|
|
|
|
from danswer.auth.users import current_curator_or_admin_user
|
|
from danswer.auth.users import current_user
|
|
from danswer.db.document_set import check_document_sets_are_public
|
|
from danswer.db.document_set import fetch_all_document_sets_for_user
|
|
from danswer.db.document_set import insert_document_set
|
|
from danswer.db.document_set import mark_document_set_as_to_be_deleted
|
|
from danswer.db.document_set import update_document_set
|
|
from danswer.db.engine import get_session
|
|
from danswer.db.models import User
|
|
from danswer.server.features.document_set.models import CheckDocSetPublicRequest
|
|
from danswer.server.features.document_set.models import CheckDocSetPublicResponse
|
|
from danswer.server.features.document_set.models import DocumentSet
|
|
from danswer.server.features.document_set.models import DocumentSetCreationRequest
|
|
from danswer.server.features.document_set.models import DocumentSetUpdateRequest
|
|
from ee.danswer.db.user_group import validate_user_creation_permissions
|
|
|
|
|
|
router = APIRouter(prefix="/manage")
|
|
|
|
|
|
@router.post("/admin/document-set")
|
|
def create_document_set(
|
|
document_set_creation_request: DocumentSetCreationRequest,
|
|
user: User = Depends(current_curator_or_admin_user),
|
|
db_session: Session = Depends(get_session),
|
|
) -> int:
|
|
validate_user_creation_permissions(
|
|
db_session=db_session,
|
|
user=user,
|
|
target_group_ids=document_set_creation_request.groups,
|
|
object_is_public=document_set_creation_request.is_public,
|
|
)
|
|
try:
|
|
document_set_db_model, _ = insert_document_set(
|
|
document_set_creation_request=document_set_creation_request,
|
|
user_id=user.id if user else None,
|
|
db_session=db_session,
|
|
)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
return document_set_db_model.id
|
|
|
|
|
|
@router.patch("/admin/document-set")
|
|
def patch_document_set(
|
|
document_set_update_request: DocumentSetUpdateRequest,
|
|
user: User = Depends(current_curator_or_admin_user),
|
|
db_session: Session = Depends(get_session),
|
|
) -> None:
|
|
validate_user_creation_permissions(
|
|
db_session=db_session,
|
|
user=user,
|
|
target_group_ids=document_set_update_request.groups,
|
|
object_is_public=document_set_update_request.is_public,
|
|
)
|
|
try:
|
|
update_document_set(
|
|
document_set_update_request=document_set_update_request,
|
|
db_session=db_session,
|
|
user=user,
|
|
)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
|
|
@router.delete("/admin/document-set/{document_set_id}")
|
|
def delete_document_set(
|
|
document_set_id: int,
|
|
user: User = Depends(current_curator_or_admin_user),
|
|
db_session: Session = Depends(get_session),
|
|
) -> None:
|
|
try:
|
|
mark_document_set_as_to_be_deleted(
|
|
db_session=db_session,
|
|
document_set_id=document_set_id,
|
|
user=user,
|
|
)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=400, detail=str(e))
|
|
|
|
|
|
"""Endpoints for non-admins"""
|
|
|
|
|
|
@router.get("/document-set")
|
|
def list_document_sets_for_user(
|
|
user: User | None = Depends(current_user),
|
|
db_session: Session = Depends(get_session),
|
|
get_editable: bool = Query(
|
|
False, description="If true, return editable document sets"
|
|
),
|
|
) -> list[DocumentSet]:
|
|
return [
|
|
DocumentSet.from_model(ds)
|
|
for ds in fetch_all_document_sets_for_user(
|
|
db_session=db_session, user=user, get_editable=get_editable
|
|
)
|
|
]
|
|
|
|
|
|
@router.get("/document-set-public")
|
|
def document_set_public(
|
|
check_public_request: CheckDocSetPublicRequest,
|
|
_: User = Depends(current_user),
|
|
db_session: Session = Depends(get_session),
|
|
) -> CheckDocSetPublicResponse:
|
|
is_public = check_document_sets_are_public(
|
|
document_set_ids=check_public_request.document_set_ids, db_session=db_session
|
|
)
|
|
return CheckDocSetPublicResponse(is_public=is_public)
|