Files
dsProject/dsLightRag/Routes/TeacherHelperRoute.py
2025-09-04 16:09:11 +08:00

178 lines
6.6 KiB
Python

import json
import logging
from fastapi import APIRouter, Request, HTTPException
from fastapi.responses import StreamingResponse
from TeacherHelper.Kit.TeacherHelper import (
generate_lesson_plan,
generate_courseware,
generate_homework
)
# 创建路由器
router = APIRouter(prefix="/api", tags=["教师辅助工具"])
# 配置日志
logger = logging.getLogger(__name__)
@router.post("/teacher/daoXueAn")
async def generate_dao_xue_an(request: Request):
"""生成导学案接口"""
try:
# 解析请求体
body = await request.body()
data = json.loads(body) if body else {}
# 获取提示词参数,如果没有提供则使用默认值
system_prompt = data.get("system_prompt")
user_prompt = data.get("user_prompt")
# 生成导学案
content_generator = await generate_lesson_plan(system_prompt, user_prompt)
# 定义SSE响应生成器
async def generate_sse_response():
try:
yield "data: 开始生成导学案...\n\n"
async for chunk in content_generator:
yield f"data: {chunk}\n\n"
yield "data: [DONE]\n\n"
except Exception as e:
logger.error(f"生成导学案时发生异常: {str(e)}")
yield f"data: 生成导学案时发生异常: {str(e)}\n\n"
# 返回SSE响应
return StreamingResponse(
generate_sse_response(),
media_type="text/event-stream",
headers={"Cache-Control": "no-cache", "Connection": "keep-alive"}
)
except json.JSONDecodeError:
raise HTTPException(status_code=400, detail="无效的JSON格式")
except Exception as e:
logger.error(f"导学案接口异常: {str(e)}")
raise HTTPException(status_code=500, detail=f"服务器内部错误: {str(e)}")
@router.post("/teacher/jiaoAn")
async def generate_jiao_an(request: Request):
"""生成教案接口"""
try:
# 解析请求体
body = await request.body()
data = json.loads(body) if body else {}
# 获取提示词参数,如果没有提供则使用默认值
system_prompt = data.get("system_prompt")
user_prompt = data.get("user_prompt")
# 生成教案
content_generator = await generate_lesson_plan(system_prompt, user_prompt)
# 定义SSE响应生成器
async def generate_sse_response():
try:
yield "data: 开始生成教案...\n\n"
async for chunk in content_generator:
yield f"data: {chunk}\n\n"
yield "data: [DONE]\n\n"
except Exception as e:
logger.error(f"生成教案时发生异常: {str(e)}")
yield f"data: 生成教案时发生异常: {str(e)}\n\n"
# 返回SSE响应
return StreamingResponse(
generate_sse_response(),
media_type="text/event-stream",
headers={"Cache-Control": "no-cache", "Connection": "keep-alive"}
)
except json.JSONDecodeError:
raise HTTPException(status_code=400, detail="无效的JSON格式")
except Exception as e:
logger.error(f"教案接口异常: {str(e)}")
raise HTTPException(status_code=500, detail=f"服务器内部错误: {str(e)}")
@router.post("/teacher/keJian")
async def generate_ke_jian(request: Request):
"""生成课件接口"""
try:
# 解析请求体
body = await request.body()
data = json.loads(body) if body else {}
# 获取提示词参数,如果没有提供则使用默认值
system_prompt = data.get("system_prompt")
user_prompt = data.get("user_prompt")
# 生成课件
content_generator = await generate_courseware(system_prompt, user_prompt)
# 定义SSE响应生成器
async def generate_sse_response():
try:
yield "data: 开始生成课件...\n\n"
async for chunk in content_generator:
yield f"data: {chunk}\n\n"
yield "data: [DONE]\n\n"
except Exception as e:
logger.error(f"生成课件时发生异常: {str(e)}")
yield f"data: 生成课件时发生异常: {str(e)}\n\n"
# 返回SSE响应
return StreamingResponse(
generate_sse_response(),
media_type="text/event-stream",
headers={"Cache-Control": "no-cache", "Connection": "keep-alive"}
)
except json.JSONDecodeError:
raise HTTPException(status_code=400, detail="无效的JSON格式")
except Exception as e:
logger.error(f"课件接口异常: {str(e)}")
raise HTTPException(status_code=500, detail=f"服务器内部错误: {str(e)}")
@router.post("/teacher/zuoYe")
async def generate_zuo_ye(request: Request):
"""生成作业接口"""
try:
# 解析请求体
body = await request.body()
data = json.loads(body) if body else {}
# 获取提示词参数,如果没有提供则使用默认值
system_prompt = data.get("system_prompt")
user_prompt = data.get("user_prompt")
material_path = data.get("material_path")
# 生成作业
content_generator = await generate_homework(system_prompt, user_prompt, material_path)
# 定义SSE响应生成器
async def generate_sse_response():
try:
yield "data: 开始生成作业...\n\n"
async for chunk in content_generator:
yield f"data: {chunk}\n\n"
yield "data: [DONE]\n\n"
except Exception as e:
logger.error(f"生成作业时发生异常: {str(e)}")
yield f"data: 生成作业时发生异常: {str(e)}\n\n"
# 返回SSE响应
return StreamingResponse(
generate_sse_response(),
media_type="text/event-stream",
headers={"Cache-Control": "no-cache", "Connection": "keep-alive"}
)
except json.JSONDecodeError:
raise HTTPException(status_code=400, detail="无效的JSON格式")
except Exception as e:
logger.error(f"作业接口异常: {str(e)}")
raise HTTPException(status_code=500, detail=f"服务器内部错误: {str(e)}")