62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
|
"""Knowledge base endpoints."""
|
||
|
|
from fastapi import APIRouter, Depends, HTTPException, Query
|
||
|
|
from sqlalchemy.orm import Session
|
||
|
|
from typing import List, Optional
|
||
|
|
from app.database import get_db
|
||
|
|
from app.schemas.knowledge import KnowledgeCreate, KnowledgeResponse, KnowledgeSearchResult
|
||
|
|
from app.services.knowledge_service import KnowledgeService
|
||
|
|
from app.api.deps import get_current_user
|
||
|
|
from app.models.user import User
|
||
|
|
|
||
|
|
router = APIRouter(prefix="/knowledge")
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("", response_model=List[KnowledgeResponse])
|
||
|
|
def list_docs(
|
||
|
|
project_id: Optional[int] = None,
|
||
|
|
user: User = Depends(get_current_user),
|
||
|
|
db: Session = Depends(get_db),
|
||
|
|
):
|
||
|
|
return KnowledgeService.list_docs(db, user.tenant_id, project_id)
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("", response_model=KnowledgeResponse, status_code=201)
|
||
|
|
def create_doc(
|
||
|
|
req: KnowledgeCreate,
|
||
|
|
user: User = Depends(get_current_user),
|
||
|
|
db: Session = Depends(get_db),
|
||
|
|
):
|
||
|
|
return KnowledgeService.create_doc(db, user.tenant_id, req)
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/search", response_model=List[KnowledgeSearchResult])
|
||
|
|
def search_docs(
|
||
|
|
q: str = Query(..., min_length=1),
|
||
|
|
project_id: Optional[int] = None,
|
||
|
|
user: User = Depends(get_current_user),
|
||
|
|
db: Session = Depends(get_db),
|
||
|
|
):
|
||
|
|
return KnowledgeService.search(db, user.tenant_id, q, project_id)
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/{doc_id}", response_model=KnowledgeResponse)
|
||
|
|
def get_doc(
|
||
|
|
doc_id: int,
|
||
|
|
user: User = Depends(get_current_user),
|
||
|
|
db: Session = Depends(get_db),
|
||
|
|
):
|
||
|
|
doc = KnowledgeService.get_doc(db, user.tenant_id, doc_id)
|
||
|
|
if not doc:
|
||
|
|
raise HTTPException(status_code=404, detail="Document not found")
|
||
|
|
return doc
|
||
|
|
|
||
|
|
|
||
|
|
@router.delete("/{doc_id}")
|
||
|
|
def delete_doc(
|
||
|
|
doc_id: int,
|
||
|
|
user: User = Depends(get_current_user),
|
||
|
|
db: Session = Depends(get_db),
|
||
|
|
):
|
||
|
|
if KnowledgeService.delete_doc(db, user.tenant_id, doc_id):
|
||
|
|
return {"success": True}
|
||
|
|
raise HTTPException(status_code=404, detail="Document not found")
|