"""TOC 约束点识别。"""

from app.services.llm_client import LLMClient


async def identify_constraints(company_data: str) -> dict:
    """AI 识别约束点 — 敏感度分析 → 约束点 = 敏感度 × 改善空间。"""
    llm = LLMClient()
    prompt = f"""请对以下企业数据进行 TOC 约束点识别：

{company_data[:5000]}

以 JSON 格式返回：
{{"constraints": [{{"name": "约束点", "sensitivity": 0.8, "improvement_space": 0.7, "priority_score": 0.56, "action": "改善建议"}}]}}"""
    result = await llm.chat(prompt, temperature=0.4)
    return result if isinstance(result, dict) else {"constraints": []}