"""AI 退出预测 Agent。"""

from app.services.llm_client import LLMClient


async def predict_exit(company_data: str) -> dict:
    """AI 计算退出路径 + 时机窗口 + 期望收益对比。"""
    llm = LLMClient()
    prompt = f"""请基于以下企业数据进行退出时机预测：

{company_data[:5000]}

以 JSON 格式返回：
{{"exit_path": "ipo/acquisition/secondary/merger", "timing_window": {{"start": "2025-06", "end": "2026-12"}}, "expected_return": 3.5, "hold_return": 2.8, "confidence": 0.7, "signals": ["退出信号1", "信号2"], "recommendation": "退出建议"}}"""
    result = await llm.chat(prompt, temperature=0.4)
    return result if isinstance(result, dict) else {}