101 lines
3.9 KiB
Python
101 lines
3.9 KiB
Python
"""
|
||
手动测试单份报告的AI分析（调试版）
|
||
"""
|
||
import asyncio
|
||
import sys
|
||
sys.path.insert(0, ".")
|
||
|
||
from app.database import AsyncSessionLocal, init_db
|
||
from app.models import Report, AnalysisResult
|
||
from app.services.ai_analyzer import ai_analyzer
|
||
from sqlalchemy import select, delete
|
||
from sqlalchemy.orm import selectinload
|
||
|
||
async def test_single_report():
|
||
print("=" * 60)
|
||
print("调试AI分析 - 测试单份已提取的报告")
|
||
print("=" * 60)
|
||
|
||
await init_db()
|
||
|
||
async with AsyncSessionLocal() as db:
|
||
# 找一份有提取内容但AI总结为空的报告
|
||
stmt = select(Report).where(
|
||
Report.is_extracted == True,
|
||
Report.analysis_status == "completed"
|
||
).options(
|
||
selectinload(Report.company),
|
||
selectinload(Report.extracted_contents),
|
||
selectinload(Report.analysis_results)
|
||
).limit(1)
|
||
|
||
result = await db.execute(stmt)
|
||
report = result.scalar_one_or_none()
|
||
|
||
if not report:
|
||
print("❌ 没有找到符合条件的报告")
|
||
return
|
||
|
||
print(f"\n报告: {report.title}")
|
||
print(f"公司: {report.company.short_name}")
|
||
print(f"提取内容数: {len(report.extracted_contents)}")
|
||
print(f"分析结果数: {len(report.analysis_results)}")
|
||
|
||
# 检查是否有summary类型的分析结果
|
||
has_summary = any(r.analysis_type == "summary" for r in report.analysis_results)
|
||
print(f"有AI总结: {has_summary}")
|
||
|
||
if not has_summary and report.extracted_contents:
|
||
print("\n→ 尝试生成AI总结...")
|
||
|
||
# 收集已有的章节分析
|
||
section_analyses = [r for r in report.analysis_results if r.analysis_type == "section"]
|
||
print(f"已有章节分析: {len(section_analyses)}")
|
||
|
||
if section_analyses:
|
||
# 将章节分析转换为summarize_analyses需要的格式
|
||
valid_results = []
|
||
for s in section_analyses:
|
||
valid_results.append({
|
||
"section_name": s.section_name,
|
||
"analysis": s.summary,
|
||
"success": True
|
||
})
|
||
|
||
company_name = report.company.short_name or report.company.company_name
|
||
|
||
print(f"调用AI汇总，共 {len(valid_results)} 个章节...")
|
||
summary = await ai_analyzer.summarize_analyses(valid_results, company_name, report.title)
|
||
|
||
print(f"\n汇总结果:")
|
||
print(f" success: {summary.get('success')}")
|
||
print(f" tokens: {summary.get('tokens', 0)}")
|
||
|
||
if summary.get("success"):
|
||
print(f" summary (前500字): {summary.get('summary', '')[:500]}")
|
||
|
||
# 保存结果
|
||
final_analysis = AnalysisResult(
|
||
report_id=report.id,
|
||
analysis_type="summary",
|
||
section_name="综合分析",
|
||
ai_model=ai_analyzer.model,
|
||
summary=summary["summary"],
|
||
token_count=summary.get("tokens", 0),
|
||
is_final=True
|
||
)
|
||
db.add(final_analysis)
|
||
await db.commit()
|
||
print("\n✅ AI总结已保存！")
|
||
else:
|
||
print(f"\n❌ 汇总失败")
|
||
if "error" in summary:
|
||
print(f" 错误: {summary.get('error')}")
|
||
else:
|
||
print("❌ 没有章节分析，无法生成汇总")
|
||
else:
|
||
print("\n✅ 报告已有AI总结，无需重新生成")
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(test_single_report())
|