feat:增加考试接口
This commit is contained in:
31
routers/agent.py
Normal file
31
routers/agent.py
Normal file
@@ -0,0 +1,31 @@
|
||||
from fastapi import APIRouter
|
||||
from starlette.responses import StreamingResponse
|
||||
|
||||
from schemas.agent import QueryQuestionRequest, QueryKnowledgeRequest, QueryExamRequest
|
||||
from service import agent_service
|
||||
|
||||
router = APIRouter(prefix="/agent", tags=["Agent"])
|
||||
|
||||
|
||||
@router.post("/question")
|
||||
def generate_question(query: QueryQuestionRequest):
|
||||
return StreamingResponse(
|
||||
agent_service.generate_question_agent(query),
|
||||
media_type="text/event-stream",
|
||||
)
|
||||
|
||||
|
||||
@router.post("/knowledge")
|
||||
def generate_knowledge(query: QueryKnowledgeRequest):
|
||||
return StreamingResponse(
|
||||
agent_service.generate_knowledge_agent(query),
|
||||
media_type="text/event-stream",
|
||||
)
|
||||
|
||||
|
||||
@router.post("/exam")
|
||||
def generate_exam(query: QueryExamRequest):
|
||||
return StreamingResponse(
|
||||
agent_service.generate_exam_agent(query),
|
||||
media_type="text/event-stream",
|
||||
)
|
||||
Reference in New Issue
Block a user