72 lines
2.8 KiB
Python
72 lines
2.8 KiB
Python
from typing import Annotated
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
|
|
|
from src.app.auth.dependancies import get_current_user
|
|
|
|
from src.app.models.knowledge import Knowledge, KnowledgeCreate
|
|
from src.app.models.question import Question
|
|
|
|
from src.app.data.knowledge import create_knowledge, get_knowledges_by_user, get_knowledge_by_id
|
|
#from src.app.data.knowledge update_knowledge, delete_knowledge
|
|
from src.app.data.question import get_questions, create_question
|
|
|
|
from src.app.services.language_generation import questions_generation
|
|
|
|
#Added in __ini__
|
|
router = APIRouter(tags=["knowledges"])
|
|
|
|
@router.get("/knowledges/")
|
|
def read(current_user: Annotated[str, Depends(get_current_user)]):
|
|
knowledges = get_knowledges_by_user(current_user)
|
|
return knowledges
|
|
|
|
@router.get("/knowledges/{id}")
|
|
def read(id: int, current_user: Annotated[str, Depends(get_current_user)]):
|
|
knowledge = get_knowledge_by_id(id, current_user)
|
|
return knowledge
|
|
|
|
@router.post("/knowledges/")
|
|
def create(knowledge_data: KnowledgeCreate, current_user: Annotated[str, Depends(get_current_user)]):
|
|
knowledge = Knowledge(content=knowledge_data.content, uri=knowledge_data.uri, user=current_user)
|
|
created_knowledge = create_knowledge(knowledge)
|
|
# if created_knowledge is None:
|
|
# raise NotFoundException("Failed to create knowledge")
|
|
return created_knowledge
|
|
|
|
@router.post("/knowledges/{id}/questions")
|
|
def generate_questions(id: int, current_user: Annotated[str, Depends(get_current_user)]):
|
|
knowledge: Knowledge = get_knowledge_by_id(id, current_user)
|
|
if not knowledge:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail="Forbidden. The requested knowledge is not available for the provided ID.",
|
|
headers={"WWW-Authenticate": "Bearer"},
|
|
)
|
|
questions_raw = questions_generation(knowledge)
|
|
for q in questions_raw:
|
|
question = Question(question = q, knowledge=knowledge, user=current_user)
|
|
create_question(question)
|
|
return questions_raw
|
|
|
|
@router.get("/knowledges/{id}/questions")
|
|
def read_questions(id: int, current_user: Annotated[str, Depends(get_current_user)]):
|
|
knowledge: Knowledge = get_knowledge_by_id(id, current_user)
|
|
if not knowledge:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail="Forbidden. The requested knowledge is not available for the provided ID.",
|
|
headers={"WWW-Authenticate": "Bearer"},
|
|
)
|
|
questions = get_questions(knowledge)
|
|
return questions
|
|
|
|
#TODO: adapt with correct pattern
|
|
# @router.post("/knowledges/{id}")
|
|
# def update(id: int, content: str, uri: str):
|
|
# knowledge = update_knowledge(id, content, uri)
|
|
# return knowledge
|
|
|
|
# @router.delete("/knowledges/{id}")
|
|
# def delete(id: int):
|
|
# knowledge = delete_knowledge(id)
|
|
# return knowledge |