106 lines
4.0 KiB
Python
106 lines
4.0 KiB
Python
|
"""
|
|||
|
|
分析管理API
|
|||
|
|
"""
|
|||
|
|
from fastapi import APIRouter, Depends, HTTPException, BackgroundTasks
|
|||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|||
|
|
from sqlalchemy import select
|
|||
|
|
|
|||
|
|
from app.database import get_db
|
|||
|
|
from app.models import Report, AnalysisStatus
|
|||
|
|
from app.schemas import MessageResponse, AnalysisRequest
|
|||
|
|
from app.services.ai_analyzer import ai_analyzer
|
|||
|
|
from app.services.pdf_extractor import pdf_extractor
|
|||
|
|
|
|||
|
|
router = APIRouter(prefix="/api/analysis", tags=["AI分析"])
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def run_analysis_task(report_id: int):
|
|||
|
|
"""后台分析任务"""
|
|||
|
|
async with AsyncSession(get_db) as db: # 注意：这里需要正确的获取session方式，但在后台任务中比较复杂
|
|||
|
|
# 简化处理：在后台任务中重新创建session
|
|||
|
|
from app.database import AsyncSessionLocal
|
|||
|
|
async with AsyncSessionLocal() as session:
|
|||
|
|
stmt = select(Report).where(Report.id == report_id)
|
|||
|
|
result = await session.execute(stmt)
|
|||
|
|
report = result.scalar_one_or_none()
|
|||
|
|
|
|||
|
|
if report:
|
|||
|
|
# 1. 确保已提取
|
|||
|
|
if not report.is_extracted:
|
|||
|
|
await pdf_extractor.extract_and_save(session, report)
|
|||
|
|
|
|||
|
|
# 2. 执行分析
|
|||
|
|
await ai_analyzer.analyze_report(session, report, force=True)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post("/run", response_model=MessageResponse)
|
|||
|
|
async def run_analysis(
|
|||
|
|
request: AnalysisRequest,
|
|||
|
|
background_tasks: BackgroundTasks,
|
|||
|
|
db: AsyncSession = Depends(get_db)
|
|||
|
|
):
|
|||
|
|
"""触发AI分析"""
|
|||
|
|
try:
|
|||
|
|
stmt = select(Report).where(Report.id == request.report_id)
|
|||
|
|
result = await db.execute(stmt)
|
|||
|
|
report = result.scalar_one_or_none()
|
|||
|
|
|
|||
|
|
if not report:
|
|||
|
|
raise HTTPException(status_code=404, detail="报告不存在")
|
|||
|
|
|
|||
|
|
if not report.is_downloaded:
|
|||
|
|
raise HTTPException(status_code=400, detail="报告尚未下载，无法分析")
|
|||
|
|
|
|||
|
|
# 更新状态
|
|||
|
|
report.analysis_status = AnalysisStatus.PENDING.value
|
|||
|
|
await db.commit()
|
|||
|
|
|
|||
|
|
# 启动后台任务
|
|||
|
|
import asyncio
|
|||
|
|
asyncio.create_task(run_analysis_with_new_session(request.report_id, request.force))
|
|||
|
|
|
|||
|
|
return MessageResponse(success=True, message="已开始分析任务")
|
|||
|
|
except HTTPException:
|
|||
|
|
raise
|
|||
|
|
except Exception as e:
|
|||
|
|
import traceback
|
|||
|
|
error_msg = f"启动分析失败: {str(e)}\n{traceback.format_exc()}"
|
|||
|
|
print(error_msg) # 打印到控制台
|
|||
|
|
raise HTTPException(status_code=500, detail=str(e)) # 返回给前端
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def run_analysis_with_new_session(report_id: int, force: bool):
|
|||
|
|
from app.database import AsyncSessionLocal
|
|||
|
|
from app.models import Report, AnalysisStatus # 显式导入以防未加载
|
|||
|
|
|
|||
|
|
async with AsyncSessionLocal() as session:
|
|||
|
|
report = None
|
|||
|
|
try:
|
|||
|
|
stmt = select(Report).where(Report.id == report_id)
|
|||
|
|
result = await session.execute(stmt)
|
|||
|
|
report = result.scalar_one_or_none()
|
|||
|
|
|
|||
|
|
if report:
|
|||
|
|
# 1. 确保已提取
|
|||
|
|
if not report.is_extracted:
|
|||
|
|
await pdf_extractor.extract_and_save(session, report)
|
|||
|
|
|
|||
|
|
# 2. 执行分析
|
|||
|
|
await ai_analyzer.analyze_report(session, report, force=force)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
import traceback
|
|||
|
|
error_trace = traceback.format_exc()
|
|||
|
|
print(f"后台分析任务严重异常 (report_id={report_id}): {e}\n{error_trace}")
|
|||
|
|
|
|||
|
|
# 标记为失败，防止前端永久卡死
|
|||
|
|
if report:
|
|||
|
|
try:
|
|||
|
|
# 重新获取对象以确保 session 仍然有效或在新的 commit 中
|
|||
|
|
report.analysis_status = AnalysisStatus.FAILED.value
|
|||
|
|
await session.commit()
|
|||
|
|
except Exception as commit_err:
|
|||
|
|
print(f"标记任务失败时发生二次错误: {commit_err}")
|
|||
|
|
finally:
|
|||
|
|
await session.close()
|