4. Làm Chủ MCP Resources & Prompts: Cung Cấp Dữ Liệu Tĩnh & Mẫu Lệnh Động Cho AI

# mcp_resources_and_prompts.py
from mcp.server.fastmcp import FastMCP

mcp = FastMCP("Company Knowledge Server")

# 1. Khai báo Resource tĩnh
@mcp.resource("company://policy/leave")
def get_leave_policy() -> str:
    """Quy định nghỉ phép năm và chế độ thai sản của công ty."""
    return """
    QUY ĐỊNH NGHỈ PHÉP NĂM 2026:
    - Nhân viên chính thức có 12 ngày phép/năm.
    - Nghỉ từ 3 ngày liên tiếp phải báo trước 7 ngày qua hệ thống.
    """

# 2. Khai báo Prompt mẫu có tham số
@mcp.prompt()
def review_code_prompt(language: str, code_snippet: str) -> str:
    """Mẫu prompt chuẩn rà soát bảo mật và hiệu năng mã nguồn."""
    return f"""
    Bạn là Kỹ sư Trưởng chuyên ngành {language}.
    Hãy rà soát đoạn mã sau về 3 khía cạnh:
    1. Lỗ hổng bảo mật (SQL Injection, XSS, Overflow).
    2. Độ phức tạp thuật toán (Big-O).
    3. Đề xuất refactor mã nguồn sạch.

    MÃ NGUỒN:
    ```{language}
    {code_snippet}
    ```
    """