System Prompt 設計：打造專屬 AI 助手人格

# claude_agent.py
from pathlib import Path

# Prompts 目錄路徑
PROMPTS_DIR = Path("data/prompts")


def get_prompt_content(prompt_name: str) -> str | None:
    """讀取 prompt 檔案內容

    Args:
        prompt_name: prompt 名稱（不含 .md）

    Returns:
        prompt 內容，找不到則回傳 None
    """
    prompt_file = PROMPTS_DIR / f"{prompt_name}.md"

    if not prompt_file.exists():
        return None

    return prompt_file.read_text(encoding="utf-8")


def list_available_prompts() -> list[dict]:
    """列出所有可用的 prompts

    Returns:
        [{"name": "default", "display_name": "預設助手", "description": "..."}]
    """
    prompts = []

    for file in PROMPTS_DIR.glob("*.md"):
        # 跳過內部使用的 prompts
        if file.stem == "summarizer":
            continue

        content = file.read_text(encoding="utf-8")

        # 從第一行取得顯示名稱
        first_line = content.split("\n")[0]
        display_name = first_line.replace("#", "").strip()

        # 取得描述（第一個段落）
        lines = content.split("\n")
        description = ""
        for line in lines[1:]:
            if line.strip() and not line.startswith("#"):
                description = line.strip()
                break

        prompts.append({
            "name": file.stem,
            "display_name": display_name,
            "description": description
        })

    return prompts

# api/ai_router.py
from fastapi import APIRouter
from services.claude_agent import list_available_prompts

router = APIRouter(prefix="/api/ai", tags=["AI"])


@router.get("/prompts")
async def get_prompts():
    """取得所有可用的 prompt 列表"""
    return list_available_prompts()

# claude_agent.py

async def call_claude(
    prompt: str,
    model: str = "sonnet",
    history: list[dict] | None = None,
    system_prompt: str | None = None,  # 可以傳入內容或名稱
    timeout: int = 120,
) -> ClaudeResponse:
    """非同步呼叫 Claude CLI"""

    # 如果傳入的是 prompt 名稱，讀取檔案內容
    if system_prompt and not system_prompt.startswith("#"):
        loaded_prompt = get_prompt_content(system_prompt)
        if loaded_prompt:
            system_prompt = loaded_prompt

    # 建立命令
    cmd = ["claude", "-p", full_prompt, "--model", model]

    if system_prompt:
        cmd.extend(["--system-prompt", system_prompt])

    # ... 其餘程式碼 ...

async def get_dynamic_system_prompt(chat_id: str, base_prompt_name: str) -> str:
    """根據對話上下文動態組合 prompt"""

    # 讀取基礎 prompt
    base_prompt = get_prompt_content(base_prompt_name)

    # 如果有對話摘要，加入上下文
    chat = await get_chat(chat_id)
    if chat and chat.get("summary"):
        base_prompt += f"\n\n## 對話背景\n{chat['summary']}"

    # 如果是特定時段，加入額外提醒
    now = datetime.now()
    if now.hour >= 22 or now.hour < 6:
        base_prompt += "\n\n（提醒：現在是深夜/凌晨，建議使用者注意休息）"

    return base_prompt

# 隨機選擇 prompt 版本（用於 A/B 測試）
import random

def get_prompt_for_ab_test(base_name: str) -> str:
    versions = [
        f"{base_name}",
        f"{base_name}_v2",  # 測試版本
    ]

    selected = random.choice(versions)
    return get_prompt_content(selected) or get_prompt_content(base_name)

"""Prompt 管理模組"""

from pathlib import Path

PROMPTS_DIR = Path("data/prompts")


def get_prompt_content(prompt_name: str) -> str | None:
    """讀取 prompt 檔案內容"""
    prompt_file = PROMPTS_DIR / f"{prompt_name}.md"

    if not prompt_file.exists():
        return None

    return prompt_file.read_text(encoding="utf-8")


def list_available_prompts() -> list[dict]:
    """列出所有可用的 prompts"""
    prompts = []

    for file in PROMPTS_DIR.glob("*.md"):
        # 跳過內部使用的
        if file.stem == "summarizer":
            continue

        content = file.read_text(encoding="utf-8")

        # 解析標題和描述
        lines = content.split("\n")
        display_name = lines[0].replace("#", "").strip() if lines else file.stem

        description = ""
        for line in lines[1:]:
            stripped = line.strip()
            if stripped and not stripped.startswith("#"):
                description = stripped
                break

        prompts.append({
            "name": file.stem,
            "display_name": display_name,
            "description": description
        })

    return prompts


def validate_prompt(content: str) -> bool:
    """驗證 prompt 格式是否正確"""
    required_sections = ["你的個性", "能力範圍", "注意事項"]

    for section in required_sections:
        if section not in content:
            return False

    return True