"""AI 报告生成服务 — 季度/年度报告自动生成。"""

from app.services.llm_client import LLMClient


async def generate_quarterly_report(company_data: str) -> str:
    """AI 生成季度报告。"""
    llm = LLMClient()
    prompt = f"""请基于以下数据生成季度投后管理报告：

{company_data[:6000]}

报告应包含：企业概览、关键指标变化、风险事项、干预措施及效果、下季度建议。"""
    result = await llm.chat(prompt, temperature=0.4)
    return result if isinstance(result, str) else str(result)


async def generate_annual_report(company_data: str) -> str:
    """AI 生成年度报告。"""
    llm = LLMClient()
    prompt = f"""请基于以下数据生成年度投后管理报告：

{company_data[:6000]}

报告应包含：年度概览、关键里程碑、财务表现、风险回顾、Alpha 归因、下年度战略建议。"""
    result = await llm.chat(prompt, temperature=0.4)
    return result if isinstance(result, str) else str(result)