| from typing import Any, Dict, List, Optional |
|
|
| from fastapi import APIRouter, Depends, HTTPException |
| from pydantic import BaseModel |
|
|
| from app.core.auth import get_current_user_id |
| from app.services.agent_service import agent_service |
|
|
|
|
| router = APIRouter(prefix="/agent", tags=["Agent"]) |
|
|
|
|
| class AgentChatMessage(BaseModel): |
| role: str |
| content: str |
|
|
|
|
| class AgentMessageRequest(BaseModel): |
| message: str |
| history: Optional[List[AgentChatMessage]] = [] |
| topic: Optional[str] = None |
| learner_type: str = "Textual" |
| top_k: int = 6 |
| question_count: int = 3 |
| wrong_answer: Optional[str] = None |
| correct_answer: Optional[str] = None |
|
|
|
|
| class AgentAction(BaseModel): |
| tool: str |
| input: Any |
| output: Any |
|
|
|
|
| class AgentSource(BaseModel): |
| text: str |
| score: float |
| topic: Optional[str] = None |
| source: Optional[str] = None |
|
|
|
|
| class AgentMessageResponse(BaseModel): |
| intent: str |
| response: str |
| blocks: List[Dict[str, Any]] = [] |
| actions: List[AgentAction] |
| sources: List[AgentSource] |
| quiz: Optional[List[Dict[str, Any]]] = None |
| feedback: Optional[str] = None |
|
|
|
|
| @router.post("/message", response_model=AgentMessageResponse) |
| def agent_message( |
| body: AgentMessageRequest, |
| user_id: str = Depends(get_current_user_id), |
| ): |
| if not body.message.strip(): |
| raise HTTPException(status_code=400, detail="Message must not be empty.") |
|
|
| history = [ |
| {"role": message.role, "content": message.content} |
| for message in (body.history or []) |
| ] |
|
|
| try: |
| return agent_service.run( |
| message=body.message, |
| user_id=user_id, |
| history=history, |
| topic=body.topic, |
| learner_type=body.learner_type, |
| top_k=body.top_k, |
| question_count=body.question_count, |
| wrong_answer=body.wrong_answer, |
| correct_answer=body.correct_answer, |
| ) |
| except HTTPException: |
| raise |
| except Exception as exc: |
| raise HTTPException(status_code=500, detail=f"Agent error: {exc}") from exc |
|
|
