"""知识图谱路由。"""

from fastapi import APIRouter, Depends
from app.core.dependencies import get_current_user
from app.models.user import User
from app.schemas.common import ApiResponse, success
from app.services.knowledge_graph_builder import build_knowledge_graph, match_best_strategy

router = APIRouter(prefix="/knowledge-graph", tags=["knowledge-graph"])


@router.post("/build", response_model=ApiResponse[dict])
async def build_graph(req: dict, user: User = Depends(get_current_user)):
    """构建知识图谱。"""
    result = await build_knowledge_graph(req.get("management_experiences", ""))
    return success(data=result)


@router.post("/match-strategy", response_model=ApiResponse[dict])
async def match_strategy(req: dict, user: User = Depends(get_current_user)):
    """为新企业匹配最佳管理策略。"""
    result = await match_best_strategy(req.get("new_company_profile", ""), req.get("knowledge_graph", {}))
    return success(data=result)