123 lines
3.8 KiB
Python
123 lines
3.8 KiB
Python
|
"""
|
|||
|
|
Questions API Router
|
|||
|
|
"""
|
|||
|
|
from typing import List, Optional
|
|||
|
|
from uuid import UUID
|
|||
|
|
from pydantic import BaseModel
|
|||
|
|
from fastapi import APIRouter, Depends, HTTPException, Query
|
|||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|||
|
|
from sqlalchemy import select
|
|||
|
|
from app.core.database import get_db
|
|||
|
|
from app.models.models import Question, Chunk
|
|||
|
|
from app.schemas.base import QuestionCreate, QuestionResponse
|
|||
|
|
|
|||
|
|
router = APIRouter()
|
|||
|
|
|
|||
|
|
|
|||
|
|
class GenerateRequest(BaseModel):
|
|||
|
|
"""Request model for generating questions"""
|
|||
|
|
chunk_ids: List[UUID] = []
|
|||
|
|
count: int = 5
|
|||
|
|
question_types: List[str] = ["fact", "summary"]
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post("/generate", response_model=dict)
|
|||
|
|
async def generate_questions(
|
|||
|
|
project_id: UUID,
|
|||
|
|
request: GenerateRequest,
|
|||
|
|
db: AsyncSession = Depends(get_db)
|
|||
|
|
):
|
|||
|
|
"""Generate questions from chunks using LLM"""
|
|||
|
|
# TODO: Implement LLM-based question generation
|
|||
|
|
# This is a placeholder that creates sample questions
|
|||
|
|
|
|||
|
|
if not request.chunk_ids:
|
|||
|
|
raise HTTPException(status_code=400, detail="chunk_ids is required")
|
|||
|
|
|
|||
|
|
# Get chunks
|
|||
|
|
result = await db.execute(
|
|||
|
|
select(Chunk).where(Chunk.id.in_(request.chunk_ids), Chunk.project_id == project_id)
|
|||
|
|
)
|
|||
|
|
chunks = result.scalars().all()
|
|||
|
|
|
|||
|
|
if not chunks:
|
|||
|
|
raise HTTPException(status_code=404, detail="No chunks found")
|
|||
|
|
|
|||
|
|
# Create sample questions (placeholder)
|
|||
|
|
created_questions = []
|
|||
|
|
for chunk in chunks:
|
|||
|
|
for i in range(request.count):
|
|||
|
|
question = Question(
|
|||
|
|
project_id=project_id,
|
|||
|
|
chunk_id=chunk.id,
|
|||
|
|
content=f"这是关于「{chunk.name}」的问题 {i+1}？",
|
|||
|
|
answer=f"这是问题 {i+1} 的答案。",
|
|||
|
|
question_type=request.question_types[0] if request.question_types else "fact",
|
|||
|
|
source="generated"
|
|||
|
|
)
|
|||
|
|
db.add(question)
|
|||
|
|
created_questions.append(question)
|
|||
|
|
|
|||
|
|
await db.commit()
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
"questions": len(created_questions),
|
|||
|
|
"message": f"Successfully generated {len(created_questions)} questions"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.get("/", response_model=dict)
|
|||
|
|
async def list_questions(
|
|||
|
|
project_id: UUID,
|
|||
|
|
chunk_id: Optional[UUID] = Query(None),
|
|||
|
|
db: AsyncSession = Depends(get_db)
|
|||
|
|
):
|
|||
|
|
"""List questions for a project"""
|
|||
|
|
query = select(Question).where(Question.project_id == project_id)
|
|||
|
|
|
|||
|
|
if chunk_id:
|
|||
|
|
query = query.where(Question.chunk_id == chunk_id)
|
|||
|
|
|
|||
|
|
result = await db.execute(query)
|
|||
|
|
questions = result.scalars().all()
|
|||
|
|
|
|||
|
|
return {"questions": [QuestionResponse.model_validate(q) for q in questions]}
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.put("/{question_id}", response_model=dict)
|
|||
|
|
async def update_question(
|
|||
|
|
project_id: UUID,
|
|||
|
|
question_id: UUID,
|
|||
|
|
question: QuestionCreate,
|
|||
|
|
db: AsyncSession = Depends(get_db)
|
|||
|
|
):
|
|||
|
|
"""Update question"""
|
|||
|
|
result = await db.execute(
|
|||
|
|
select(Question).where(Question.id == question_id, Question.project_id == project_id)
|
|||
|
|
)
|
|||
|
|
db_question = result.scalar_one_or_none()
|
|||
|
|
if not db_question:
|
|||
|
|
raise HTTPException(status_code=404, detail="Question not found")
|
|||
|
|
|
|||
|
|
for key, value in question.model_dump(exclude_unset=True).items():
|
|||
|
|
setattr(db_question, key, value)
|
|||
|
|
|
|||
|
|
await db.commit()
|
|||
|
|
await db.refresh(db_question)
|
|||
|
|
return QuestionResponse.model_validate(db_question)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.delete("/{question_id}", response_model=dict)
|
|||
|
|
async def delete_question(project_id: UUID, question_id: UUID, db: AsyncSession = Depends(get_db)):
|
|||
|
|
"""Delete question"""
|
|||
|
|
result = await db.execute(
|
|||
|
|
select(Question).where(Question.id == question_id, Question.project_id == project_id)
|
|||
|
|
)
|
|||
|
|
question = result.scalar_one_or_none()
|
|||
|
|
if not question:
|
|||
|
|
raise HTTPException(status_code=404, detail="Question not found")
|
|||
|
|
|
|||
|
|
await db.delete(question)
|
|||
|
|
await db.commit()
|
|||
|
|
return {"message": "Question deleted successfully"}
|