SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 20:12:55 +01:00
Code Issues Projects Releases Packages Activity

Added new system prompt

Browse source
This commit is contained in:
Saurab-Shrestha9639*969**9858//852 2024-05-29 10:55:42 +05:45
parent 474698834f
commit 759767dc1b
4 changed files with 31 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

12
private_gpt/users/crud/document_crud.py
View file

@ -28,11 +28,21 @@ class CRUDDocuments(CRUDBase[Document, DocumentCreate, DocumentUpdate]):
def get_documents_by_departments(
self, db: Session, *, department_id: int
) -> List[Document]:
all_department_id = 1 # department ID for "ALL" is 1
return (
db.query(self.model)
.join(document_department_association)
.join(Department)
.filter(document_department_association.c.department_id == department_id)
.filter(
or_(
and_(
document_department_association.c.department_id == department_id,
),
and_(
document_department_association.c.department_id == all_department_id,
),
)
)
.order_by(desc(getattr(Document, 'uploaded_at')))
.all()
)