from app.tools.java_client import JavaClient
import logging

logger = logging.getLogger(__name__)


class AnalysisAgent:
    """报告解读 Agent: 拉取报告数据 + 问卷数据 + LLM 分析"""

    def __init__(self):
        self.java = JavaClient()

    async def get_report_summary(self, report_id: int) -> dict:
        """获取报告概要 (用于 LLM 上下文)"""
        try:
            client = await self.java._get_client()
            resp = await client.post("/api/health/report/summary", json={
                "reportId": report_id,
            })
            data = resp.json()
            if data.get("code") == 200:
                return data.get("data", {})
        except Exception as e:
            logger.warning("获取报告概要失败: %s", e)
        return {}